维持mgo会话的最佳做法

Question

我目前正在使用带有mgo lib的mongodb进行Web应用程序,但我不确定我使用它的方式是不是很好..

package db

import (
    "gopkg.in/mgo.v2"
)

const (
    MongoServerAddr = "192.168.0.104"
    RedisServerAddr = "192.168.0.104"
)

var (
    MongoSession, err = mgo.Dial(MongoServerAddr)

    MDB  = MongoSession.DB("message")
    MCol = MDB.C("new")
    MSav = MDB.C("save")

    UDB  = MongoSession.DB("account")
    UCol = UDB.C("user")
)

我初始化db会话并创建获取集合和文档值的变量,因此当我需要查询集合时,我使用变量来创建它.

像那样 :

func UserExist(username string) bool {
    user := Users{}
    err := db.UCol.Find(bson.M{"username": username}).One(&user)
    if err != nil {
        return false
    } else {
        return true
    }
}

那么有最好的做法还是这个很好......？谢谢

Answer 1

我建议不要使用这样的全球会话.相反,您可以创建一个负责所有数据库交互的类型.例如:

type DataStore struct {
    session *mgo.Session
}

func (ds *DataStore) ucol() *mgo.Collection { ... }

func (ds *DataStore) UserExist(user string) bool { ... }

该设计有许多好处.一个重要的是,它允许您同时在飞行中拥有多个会话,因此,例如,如果您有一个http处理程序,则可以创建一个仅由该一个请求由独立会话支持的本地会话:

func (s *WebSite) dataStore() *DataStore {
    return &DataStore{s.session.Copy()}
}    

func (s *WebSite) HandleRequest(...) {
    ds := s.dataStore()
    defer ds.Close()
    ...
}

在这种情况下,mgo驱动程序表现良好,因为会话在内部进行缓存和重用/维护.每个会话在使用时也将由独立套接字支持,并且可能配置了独立设置,并且还将具有独立的错误处理.如果您使用单个全局会话,这些是您最终必须处理的问题.