Go操作mysql Memcache Mongodb

date: 2014-12-12

操作mysql
操作mongodb
操作memcache

发现go提供的操作msyql、memcache、mongodb的文档没有lua-ngx的好读，即实例不是很明确，此文列出自己操作时的实例，可让快速入手，但是优化以及错误处理等还有很多工作。

操作mysql

go-sql-driver/mysql Example 上提供了用go操作mysql的两个例子。例子中关于查询，一个是一次只取一个结果，一个是一次去多个结果，而且使用了prepare的方法，防止sql注入攻击。

此文中是参考上面例子，做的测试。

只取一个数据的例子

import (

"database/sql"

_ "github.com/go-sql-driver/mysql"

)

func main() {

db, err := sql.Open("mysql", "user:passwd@tcp(host:port)/db_name")

if err != nil {

panic(err.Error())

}

defer db.Close()

stmtOut, err := db.Prepare("SELECT uid FROM mj_table limit 1")

if err != nil {

panic(err.Error())

}

var uid string

err = stmtOut.QueryRow().Scan(&uid)

if err != nil {

Log.Info("err to Scan: %s", err)

return

}

Log.Info("uid: %s", uid)

}

取多个数据的例子

...

rows, err := db.Query("select appid from mj_table limit 1")

if err != nil {

Log.Info("No Err in select msyql: %s", err)

return

}

// Get column names

columns, err := rows.Columns()

if err != nil {

panic(err.Error())

}

// Make a slice for the values

values := make([]sql.RawBytes, len(columns))

scanArgs := make([]interface{}, len(values))   // 必须要有此类型

for i := range values {

scanArgs[i] = &values[i]

}

// Fetch rows

for rows.Next() {

// get RawBytes from data

err = rows.Scan(scanArgs...)

if err != nil {

panic(err.Error()) 

}

var value string

for i, col := range values {

if col == nil {

value = "NULL"

} else {

value = string(col)

}

Log.Info("name: %s, value: %s ", columns[i], value)

}

Log.Info("-----------------------------------")

}

...

操作mongodb

mgo是go语言操作MongoDb的驱动，因为接口比较丰富，官方文档看着也就比较长。

下面代码提供了建立连接、插入数据、查询、删除的操作

import (

"time"

"gopkg.in/mgo.v2"   // mgo的包

"gopkg.in/mgo.v2/bson"

)

func main() {

var (

err error

)

//建立连接

mongoURL := "192.168.1.140,192.168.1.141:20001,192.168.1.141:20002"  //可在配置文件中设置

maxWait := time.Duration(10 * time.Second)

sess, err = mgo.DialWithTimeout(mongoURL, maxWait)

if err != nil {

Log.Error("mgo.Dial err:%s", err)

panic(err)

}

sess.SetMode(mgo.Monotonic, true)

defer sess.Close()

//插入数据  

//...构造要插入数据

db_name := "db"

table_name := "table"  // 即mongodb中的Collection

dataBytes := []byte("{\"name\":\"jgj\"}")

var intf map[string]interface{}

err = json.Unmarshal(dataBytes, &intf)

//...执行入库操作

// new_sess := sess.Copy()  // 可以复制一份回话，用新回话做操作

col := sess.DB(appid).C(classname)

err = col.Insert(intf)

if err != nil {

 Log.Error("db POST, db insert error, appid:%s, class:%s, data:%s, err:%s", appid, classname, string(dataBytes), err)

 return

}

// 获取数据

var objectid interface{}

objectid = string("5486b4e08bb3c2659c000001")   

err = col.FindId(objectid).One(&intf)

this.Data["json"] = intf

// this.ServeJson() // 显示

// 更新某个对象

err = col.Update(bson.M{"_id": objectid, "updatedAt": oldUpdatedAt}, intf)

// 删除某个对象

err = col.RemoveId(objectid)

// 查询

selectParam := "{\"name\":true, \"age\":true}"

var selectIntf map[string]bool

err = json.Unmarshal([]byte(selectParam), &selectIntf)

whereParam := "{\"or\":[{\"name\":\"jgj\"},{\"age\":\"29\"}]}"

var whereIntf := map[string]interface{}{}

err = json.Unmarshal([]byte(whereParam), &whereIntf)

//...json to bson. 代码未提供

ConvertBSON(whereIntf)

sortParam = "name,-age"

offset = 1

limit = 10

//...TODO 此处要有错误的判断

query := col.Find(whereIntf)

query = query.Sort(sortParam)

query = query.Select(selectIntf)

query = query.Skip(offset)

query = query.Limit(limit)

iter := query.Iter()

var result []bson.M = make([]bson.M, 0)

intf := bson.M{}

for iter.Next(&intf) {

result = append(result, intf)

intf = bson.M{}

}

iter.Close()

this.Data["json"] = result

this.ServeJson()

}

操作memcache

GoDoc中memcache的说明提供了操作的一些函数，没有给出实际的例子。下面是测试的小例子

con := memcache.New("ip:host")

if con == nil {

Log.Info("Failed to connect Memcache")

}

item := &memcache.Item{Key:"jgj",Value:[]byte("test_value"),Expiration:0 }

err := con.Set(item)

if err != nil {

Log.Info("failed to set item: %s", err)

}

item1, err := con.Get("jgj")

if err != nil {

Log.Info("Failed to get item %s ", err)

return

}

Log.Info("get value: %s", item1.Value)