Mgo库的常见坑总结

ChrRuffin 8年前

来自: http://ipfans.github.io/2016/01/something-about-mgo-driver/

mgo库 是一个很好用的 MongoDB 驱动。对我们来说,主力数据库是 MongoDB ,因此这个驱动对我们来说也是非常重要的。但是,mgo库有些问题算是一些坑,这里我做了一些简单的整理。

一些关于bson.ObjectId的问题

ObjectId为空的判断

如果你看 bson.ObjectId 定义的话,它是一个 string 类型的数据。但是如果你直接定义一个结构,并且生成对象时,这个对象并不是这样的。

我们首先定义一个结构体:

type Home struct {      ID   bson.ObjectId `bson:"_id,omitempty"`      Name string        `bson:"name"`  }

然后看看一个生成的内容

h := Home{Name: "123"}  fmt.Println(h.ID)

结果是 ObjectIdHex("") 。换句话说,如果你是想判断一个结构体的 ObjectId 是否为空,使用 h.ID == "" 是一定会结果为 false 的。如果你想判断是否为空,正确的方式应该为:

h := Home{Name: "123"}  fmt.Println(h.ID)  fmt.Println(h.ID.Hex() == "")

正确的生成ObjectId

首先值得注意的是 NewObjectIdWithTime(t time.Time) 这个方法生成的 ObjectId 并不是唯一的,结果可能导致的是插入失败。最有效的方式是设定 ObjectId 对象支持 omitempty 属性,就像我上面生成的结构体一样,由数据库统一调配生成ObjectId。如果真的确实需要,可以选择 NewObjctId() 。

时间问题

之前看到有人问,为什么保存的时间进入到数据库中慢了8个小时呢?原因是在保存进入 MongoDB 时,数据是按照 UTC 时间(不懂什么是UTC? 看这里 )进行的保存,但是取出是按照当前时区来取出。那么问题来了,我的客户如果不都是国人,我怎么保存时间呢?目前我们采用了两种方式来确定数据库的保存时间。一种是 Unix时间戳 ,这个是不受到时区的影响的,由前端格式化为对应的时区时间;另外一种则是需要在额外的对从 MongoDB 数据库中取出的数据进行额外的时区校准,简单来说可以这样:

type Home struct {      ID         bson.ObjectId `bson:"_id,omitempty"`      Name       string        `bson:"name"`      InsertTime time.Time     `bson:"insert_time"`  }    func main() {      sess, _ := mgo.Dial("127.0.0.1")      c := sess.DB("test").C("home")      h := Home{Name: "123", InsertTime: time.Now()}      c.Upsert(bson.M{"name": "123"}, h)      c.Find(bson.M{"name": "123"}).One(&h)      fmt.Println(h.InsertTime.Format("2006-01-02 15:04:05"))      tz, _ := time.LoadLocation("America/New_York")      fmt.Println(h.InsertTime.In(tz).Format("2006-01-02 15:04:05"))  }

更高效的使用Session

在 MongoDB 中合理使用 Session 可以更高效的操作数据库,做法是在之前进行一次 Copy 操作:

sessionCopy := mongoSession.Copy()  defer sessionCopy.Close()  collection := sessionCopy.DB(TestDatabase).C("buoy_stations")    log.Printf("RunQuery : %d : Executing\n", query)    // Retrieve the list of stations.  var buoyStations []BuoyStation  err := collection.Find(nil).All(&buoyStations)  if err != nil {      log.Printf("RunQuery : ERROR : %s\n", err)      return  }    log.Printf("RunQuery : %d : Count[%d]\n", query, len(buoyStations))

另外值得一提的是MongoDB本身,目前我们已经提升到了 MongoDB 3.0+ 版本,优势是相对之前版本的 WiredTiger 引擎比较令人印象深刻(如果有条件可以选择最新的3.2版本)。大家对之前 MongoDB 中索引建立的痛苦有印象,这个现象在 WiredTiger 引擎中也有所改进。更多的改进可以在 WiredTiger 的 官网 和MongoDB的 性能白皮书 中了解更多。最后,当然,我仍然不建议在非严重必要情况下创建多余索引,这对 MongoDB 的性能和资源消耗都有较大影响。