go(四)
来谈谈go语言与mongodb的交互
首先,定义几个连接变量
var (
mgoSession *mgo.Session
logSession *mgo.Session
sequenceC = "sequence"
)
然后定义一些配置变量:
var dbUrl string
var gameDb string
var userDb string
var logDb string
var logDbUrl string
var dbPoolLimit int
给配置变量绑定数据
flag.StringVar(&dbUrl, "db", "127.0.0.1", "db url.")
flag.StringVar(&logDbUrl, "logDBUrl", "", "logdb url.")
flag.StringVar(&gameDb, "gameDB", "poker_game", "gameDB名称")
flag.StringVar(&userDb, "userDB", "poker_user", "userDB名称")
flag.StringVar(&logDb, "logDB", "poker_log", "logDB名称")
flag.IntVar(&dbPoolLimit, "dbPoolLimit", 200, "数据库连接数限制")
然后,得到连接
func getSession() *mgo.Session {
if mgoSession == nil {
var err error
mgoSession, err = mgo.Dial(dbUrl)
if err != nil {
panic(err)
}
mgoSession.SetPoolLimit(dbPoolLimit)
}
return mgoSession.Clone()
}
然后写个操作数据库的方法就可以了
func WithGameCollection(collection string, s func(*mgo.Collection) error) error {
session := getSession()
defer session.Close()
c := session.DB(gameDb).C(collection)
return s(c)
}