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