Go语言下与MongoDB通信-使用mgo

mgo简介 mgo（音mango）是MongoDB的Go语言驱动，它用基于Go语法的简单API实现了丰富的特性，并经过良好测试。 官方网站：http://labix.org/mgo。 golang.tc（golangtc.com）网站的数据存储就是是用的MongoDB+mgo。近一年使用下来表现良好。 API文档 下面是mgo、mgo/bson、mgo/txn的在线文档。 •mgo GoDoc GoWalker •mgo/bson GoDoc GoWalker •mgo/txn GoDoc GoWalker 安装 安装mgo之前请先安装Golang和MongoDB，安装过程不再赘述。 安装bzr版本工具（mgo使用Bazaar作为版本控制系统，因安装的时候需要去Bazaar拉取代码）。 安装命令 go get labix.org/v2/mgo 示例 下面的代码是个示例。 package main import ( "fmt" "labix.org/v2/mgo" "labix.org/v2/mgo/bson" ) type Person struct { Name string Phone string } func main() { session, err := mgo.Dial("") if err != nil { panic(err) } defer session.Close() // Optional. Switch the session to a monotonic behavior. session.SetMode(mgo.Monotonic, true) c := session.DB("test").C("people") err = c.Insert(&Person{"Ale", "+55 53 8116 9639"}, &Person{"Cla", "+55 53 8402 8510"}) if err != nil { panic(err) } result := Person{} err = c.Find(bson.M{"name": "Ale"}).One(&result) if err != nil { panic(err) } fmt.Println("Phone:", result.Phone) } 启动MongoDB，把上面代码复制了跑一下，如果输出下面内容，说明安装成功。 Phone: +55 53 8116 9639 具体代码什么意思先不用管，后面讲解每个方法的用法。 如何使用 下面介绍如何使用mgo，主要介绍集合的操作。对数据库，用户等操作，请自行查看文档。 第一步当然是先导入mgo包 import ( "labix.org/v2/mgo" "labix.org/v2/mgo/bson" ) 连接服务器 通过方法Dial()来和MongoDB服务器建立连接。Dial()定义如下： func Dial(url string) (*Session, error) 具体使用： session, err := mgo.Dial(url) 如果是本机，并且MongoDB是默认端口27017启动的话，下面几种方式都可以。 session, err := mgo.Dial("") session, err := mgo.Dial("localhost") session, err := mgo.Dial("127.0.0.1") session, err := mgo.Dial("localhost:27017") session, err := mgo.Dial("127.0.0.1:27017") 如果不在本机或端口不同，传入相应的地址即可。如： mongodb://myuser:mypass@localhost:40001,otherhost:40001/mydb 切换数据库 通过Session.DB()来切换相应的数据库。 func (s *Session) DB(name string) *Database 如切换到test数据库。 db := session.DB("test") 切换集合 通过Database.C()方法切换集合（Collection），这样我们就可以通过对集合进行增删查改操作了。 func (db *Database) C(name string) *Collection 如切换到`users`集合。 c := db.C("users") 对集合进行操作 介绍插入、查询、修改、删除操作。 先提一下ObjectId，MongoDB每个集合都会一个名为_id的主键，这是一个24位的16进制字符串。对应到mgo中就是bson.ObjectId。 这里我们定义一个struct，用来和集合对应。 type User struct { Id_ bson.ObjectId `bson:"_id"` Name string `bson:"name"` Age int `bson:"age"` JoinedAt time.Time `bson:"joined_at"` Interests []string `bson:"interests"` } 注解 注意User的字段首字母大写，不然不可见。通过bson:”name”这种方式可以定义MongoDB中集合的字段名，如果不定义，mgo自动把struct的字段名首字母小写作为集合的字段名。如果不需要获得id_，Id_可以不定义，在插入的时候会自动生成。 插入 插入方法定义如下： func (c *Collection) Insert(docs ...interface{}) error 下面代码插入两条集合数据。 err = c.Insert(&User{ Id_: bson.NewObjectId(), Name: "Jimmy Kuu", Age: 33, JoinedAt: time.Now(), Interests: []string{"Develop", "Movie"}, }) if err != nil { panic(err) } err = c.Insert(&User{ Id_: bson.NewObjectId(), Name: "Tracy Yu", Age: 31, JoinedAt: time.Now(), Interests: []string{"Shoping", "TV"}, }) if err != nil { panic(err) } 这里通过bson.NewObjectId()来创建新的ObjectId，如果创建完需要用到的话，放在一个变量中即可，一般在Web开发中可以作为参数跳转到其他页面。 通过MongoDB客户端可以发现，两条即可已经插入。 { "_id" : ObjectId( "5204af979955496907000001" ), "name" : "Jimmy Kuu", "age" : 33, "joned_at" : Date( 1376038807950 ), "interests" : [ "Develop", "Movie" ] } { "_id" : ObjectId( "5204af979955496907000002" ), "name" : "Tracy Yu", "age" : 31, "joned_at" : Date( 1376038807971 ), "interests" : [ "Shoping", "TV" ] } 查询 通过func (c *Collection) Find(query interface{}) *Query来进行查询，返回的Query struct可以有附加各种条件来进行过滤。 通过Query.All()可以获得所有结果，通过Query.One()可以获得一个结果，注意如果没有数据或者数量超过一个，One()会报错。 条件用bson.M{key: value}，注意key必须用MongoDB中的字段名，而不是struct的字段名。 无条件查询 查询所有 var users []User c.Find(nil).All(&users) fmt.Println(users) 上面代码可以把所有Users都查出来： [{ObjectIdHex("5204af979955496907000001") Jimmy Kuu 33 2013-08-09 17:00:07.95 +0800 CST [Develop Movie]} {ObjectIdHex("5204af979955496907000002") Tracy Yu 31 2013-08-09 17:00:07.971 +0800 CST [Shoping TV]}] 根据ObjectId查询 id := "5204af979955496907000001" objectId := bson.ObjectIdHex(id) user := new(User) c.Find(bson.M{"_id": objectId}).One(&user) fmt.Println(user) 结果如下： &{ObjectIdHex("5204af979955496907000001") Jimmy Kuu 33 2013-08-09 17:00:07.95 +0800 CST [Develop Movie]} 更简单的方式是直接用FindId()方法： c.FindId(objectId).One(&user) 注解 注意这里没有处理err。当找不到的时候用One()方法会出错。 单条件查询 =($eq) c.Find(bson.M{"name": "Jimmy Kuu"}).All(&users) !=($ne) c.Find(bson.M{"name": bson.M{"$ne": "Jimmy Kuu"}}).All(&users) >($gt) c.Find(bson.M{"age": bson.M{"$gt": 32}}).All(&users) <($lt) c.Find(bson.M{"age": bson.M{"$lt": 32}}).All(&users) >=($gte) c.Find(bson.M{"age": bson.M{"$gte": 33}}).All(&users) <=($lte) c.Find(bson.M{"age": bson.M{"$lte": 31}}).All(&users) in($in) c.Find(bson.M{"name": bson.M{"$in": []string{"Jimmy Kuu", "Tracy Yu"}}}).All(&users) 多条件查询 and($and) c.Find(bson.M{"name": "Jimmy Kuu", "age": 33}).All(&users) or($or) c.Find(bson.M{"$or": []bson.M{bson.M{"name": "Jimmy Kuu"}, bson.M{"age": 31}}}).All(&users) 修改 通过func (*Collection) Update来进行修改操作。 func (c *Collection) Update(selector interface{}, change interface{}) error 注意修改单个或多个字段需要通过$set操作符号，否则集合会被替换。 修改字段的值($set) c.Update(bson.M{"_id": bson.ObjectIdHex("5204af979955496907000001")}, bson.M{"$set": bson.M{ "name": "Jimmy Gu", "age": 34, }}) inc($inc) 字段增加值 c.Update(bson.M{"_id": bson.ObjectIdHex("5204af979955496907000001")}, bson.M{"$inc": bson.M{ "age": -1, }}) push($push) 从数组中增加一个元素 c.Update(bson.M{"_id": bson.ObjectIdHex("5204af979955496907000001")}, bson.M{"$push": bson.M{ "interests": "Golang", }}) pull($pull) 从数组中删除一个元素 c.Update(bson.M{"_id": bson.ObjectIdHex("5204af979955496907000001")}, bson.M{"$pull": bson.M{ "interests": "Golang", }}) 删除 c.Remove(bson.M{"name": "Jimmy Kuu"}) 注解 这里也支持多条件，参考多条件查询。