Go NOSQL數(shù)據(jù)庫(kù)操作

NoSQL(Not Only SQL)，指的是非關(guān)系型的數(shù)據(jù)庫(kù)。隨著Web2.0的興起，傳統(tǒng)的關(guān)系數(shù)據(jù)庫(kù)在應(yīng)付Web2.0網(wǎng)站，特別是超大規(guī)模和高并發(fā)的SNS類型的Web2.0純動(dòng)態(tài)網(wǎng)站已經(jīng)顯得力不從心，暴露了很多難以克服的問(wèn)題，而非關(guān)系型的數(shù)據(jù)庫(kù)則由于其本身的特點(diǎn)得到了非常迅速的發(fā)展。

而Go語(yǔ)言作為21世紀(jì)的C語(yǔ)言，對(duì)NOSQL的支持也是很好，目前流行的NOSQL主要有redis、mongoDB、Cassandra和Membase等。這些數(shù)據(jù)庫(kù)都有高性能、高并發(fā)讀寫(xiě)等特點(diǎn)，目前已經(jīng)廣泛應(yīng)用于各種應(yīng)用中。我接下來(lái)主要講解一下redis和mongoDB的操作。

redis

redis是一個(gè)key-value存儲(chǔ)系統(tǒng)。和Memcached類似，它支持存儲(chǔ)的value類型相對(duì)更多，包括string(字符串)、list(鏈表)、set(集合)和zset(有序集合)。

目前應(yīng)用redis最廣泛的應(yīng)該是新浪微博平臺(tái)，其次還有Facebook收購(gòu)的圖片社交網(wǎng)站instagram。以及其他一些有名的互聯(lián)網(wǎng)企業(yè)

Go目前支持redis的驅(qū)動(dòng)有如下

• https://github.com/garyburd/redigo (推薦)
• https://github.com/go-redis/redis
• https://github.com/hoisie/redis
• https://github.com/alphazero/Go-Redis
• https://github.com/simonz05/godis

我以redigo驅(qū)動(dòng)為例來(lái)演示如何進(jìn)行數(shù)據(jù)的操作:

package main

import (
	"fmt"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/garyburd/redigo/redis"
)

var (
	Pool *redis.Pool
)

func init() {
	redisHost := ":6379"
	Pool = newPool(redisHost)
	close()
}

func newPool(server string) *redis.Pool {

	return &redis.Pool{

		MaxIdle:     3,
		IdleTimeout: 240 * time.Second,

		Dial: func() (redis.Conn, error) {
			c, err := redis.Dial("tcp", server)
			if err != nil {
				return nil, err
			}
			return c, err
		},

		TestOnBorrow: func(c redis.Conn, t time.Time) error {
			_, err := c.Do("PING")
			return err
		}
	}
}

func close() {
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt)
	signal.Notify(c, syscall.SIGTERM)
	signal.Notify(c, syscall.SIGKILL)
	go func() {
		<-c
		Pool.Close()
		os.Exit(0)
	}()
}

func Get(key string) ([]byte, error) {

	conn := Pool.Get()
	defer conn.Close()

	var data []byte
	data, err := redis.Bytes(conn.Do("GET", key))
	if err != nil {
		return data, fmt.Errorf("error get key %s: %v", key, err)
	}
	return data, err
}

func main() {
	test, err := Get("test")
	fmt.Println(test, err)
}

另外以前我fork了最后一個(gè)驅(qū)動(dòng)，修復(fù)了一些bug，目前應(yīng)用在我自己的短域名服務(wù)項(xiàng)目中(每天200W左右的PV值)

https://github.com/astaxie/goredis

接下來(lái)的以我自己fork的這個(gè)redis驅(qū)動(dòng)為例來(lái)演示如何進(jìn)行數(shù)據(jù)的操作

package main

import (
	"fmt"

	"github.com/astaxie/goredis"
)

func main() {
	var client goredis.Client
	// 設(shè)置端口為redis默認(rèn)端口
	client.Addr = "127.0.0.1:6379"
	
	//字符串操作
	client.Set("a", []byte("hello"))
	val, _ := client.Get("a")
	fmt.Println(string(val))
	client.Del("a")

	//list操作
	vals := []string{"a", "b", "c", "d", "e"}
	for _, v := range vals {
		client.Rpush("l", []byte(v))
	}
	dbvals,_ := client.Lrange("l", 0, 4)
	for i, v := range dbvals {
		println(i,":",string(v))
	}
	client.Del("l")
}

我們可以看到操作redis非常的方便，而且我實(shí)際項(xiàng)目中應(yīng)用下來(lái)性能也很高。client的命令和redis的命令基本保持一致。所以和原生態(tài)操作redis非常類似。

mongoDB

MongoDB是一個(gè)高性能，開(kāi)源，無(wú)模式的文檔型數(shù)據(jù)庫(kù)，是一個(gè)介于關(guān)系數(shù)據(jù)庫(kù)和非關(guān)系數(shù)據(jù)庫(kù)之間的產(chǎn)品，是非關(guān)系數(shù)據(jù)庫(kù)當(dāng)中功能最豐富，最像關(guān)系數(shù)據(jù)庫(kù)的。他支持的數(shù)據(jù)結(jié)構(gòu)非常松散，采用的是類似json的bjson格式來(lái)存儲(chǔ)數(shù)據(jù)，因此可以存儲(chǔ)比較復(fù)雜的數(shù)據(jù)類型。Mongo最大的特點(diǎn)是他支持的查詢語(yǔ)言非常強(qiáng)大，其語(yǔ)法有點(diǎn)類似于面向?qū)ο蟮牟樵冋Z(yǔ)言，幾乎可以實(shí)現(xiàn)類似關(guān)系數(shù)據(jù)庫(kù)單表查詢的絕大部分功能，而且還支持對(duì)數(shù)據(jù)建立索引。

下圖展示了mysql和mongoDB之間的對(duì)應(yīng)關(guān)系，我們可以看出來(lái)非常的方便，但是mongoDB的性能非常好。

目前Go支持mongoDB最好的驅(qū)動(dòng)就是mgo，這個(gè)驅(qū)動(dòng)目前最有可能成為官方的pkg。

安裝mgo:

go get gopkg.in/mgo.v2

下面我將演示如何通過(guò)Go來(lái)操作mongoDB：

package main

import (
	"fmt"
	"log"

	"gopkg.in/mgo.v2"
	"gopkg.in/mgo.v2/bson"
)

type Person struct {
	Name  string
	Phone string
}

func main() {
	session, err := mgo.Dial("server1.example.com,server2.example.com")
	if err != nil {
		panic(err)
	}
	defer session.Close()

	// Optional. Switch the session to a monotonic behavior.
	session.SetMode(mgo.Monotonic, true)

	c := session.DB("test").C("people")
	err = c.Insert(&Person{"Ale", "+55 53 8116 9639"},
		&Person{"Cla", "+55 53 8402 8510"})
	if err != nil {
		log.Fatal(err)
	}

	result := Person{}
	err = c.Find(bson.M{"name": "Ale"}).One(&result)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("Phone:", result.Phone)
}

我們可以看出來(lái)mgo的操作方式和beedb的操作方式幾乎類似，都是基于struct的操作方式，這個(gè)就是Go Style。