Go操作mysql Memcache Mongodb

• 操作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)