Go 在 MongoDB 中常用查询与修改的操作

青衣

以下所有例子中结构定义如下：

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"`
  Num []int `bson:"num"`
}

1、查询
　　通过func (c *Collection) Find(query interface{}) *Query来进行查询，返回的Query struct可以有附加各种条件来进行过滤。
　　通过Query.All()可以获得所有结果，通过Query.One()可以获得一个结果，注意如果没有数据或者数量超过一个，One()会报错。
　　条件用bson.M{key: value}，注意key必须用MongoDB中的字段名，而不是struct的字段名。
1.1、查询所有

var users []User
c.Find(nil).All(&users)

　　上面代码可以把所有Users都查出来：
1.2、根据ObjectId查询

id := "5204af979955496907000001"
objectId := bson.ObjectIdHex(id)
user := new(User)
c.Find(bson.M{"_id": objectId}).One(&user)

　　更简单的方式是直接用FindId()方法：

c.FindId(objectId).One(&user)

　　注意这里没有处理err。当找不到的时候用One()方法会出错。
1.3、单条件查询

=($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)

1.4、多条件查询

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)

2、修改
　　通过func (*Collection) Update来进行修改操作。

func (c *Collection) Update(selector interface{}, change interface{}) error

　　注意修改单个或多个字段需要通过$set操作符号，否则集合会被替换。
2.1、($set)

//修改字段的值
c.Update(
  bson.M{"_id": bson.ObjectIdHex("5204af979955496907000001")},
  bson.M{"$set": bson.M{ "name": "Jimmy Gu", "age": 34 }}
)

2.2、inc($inc)

//字段增加值
c.Update(
  bson.M{"_id": bson.ObjectIdHex("5204af979955496907000001")},
  bson.M{"$inc": bson.M{ "age": -1 }}
)
//字段Num数组第三个数增加值
c.Update(
  bson.M{"_id": bson.ObjectIdHex("5204af979955496907000001")},
  bson.M{"$inc": bson.M{ "Num." + strconv.Itoa(2): 1 }})

2.3、push($push)

//从数组中增加一个元素
c.Update(
  bson.M{"_id": bson.ObjectIdHex("5204af979955496907000001")},
  bson.M{"$push": bson.M{ "interests": "Golang" }}
)

2.4、pull($pull)

//从数组中删除一个元素
c.Update(
  bson.M{"_id": bson.ObjectIdHex("5204af979955496907000001")},
  bson.M{"$pull": bson.M{ "interests": "Golang" }}
)

2.5、删除

c.Remove(bson.M{"name": "Jimmy Kuu"})

　　补充：golang mongodb查找find demo
　　使用gopkg.in/mgo.v2库操作，插入操作主要使用mongodb中Collection对象的Find方法，函数原型：

func (c *Collection) Find(query interface{}) *Query

　　查找的时候Find的参数都会用bson.M类型

type M map[string]interface{}

　　例如：bson.M{"name": "Tom"}相当直接mongodb的查询条件{"name": "Tom"}
统一封装下getDB方法

package main
 
import (
   "fmt"
 
   "gopkg.in/mgo.v2"
   "gopkg.in/mgo.v2/bson"
)
 
// get mongodb db
func getDB() *mgo.Database {
   session, err := mgo.Dial( "172.16.27.134:10001" )
   if err != nil {
     panic(err)
   }
 
   session.SetMode(mgo.Monotonic, true)
   db := session.DB( "test" )
   return db
}

查找单条记录

func findOne() {
   db := getDB()
 
   c := db.C( "user" )
 
   // 用struct接收,一般情况下都会这样处理
   type User struct {
     Name string  "bson:`name`"
     Age  int   "bson:`age`"
   }
   user := User{}
   err := c.Find(bson.M{ "name" :  "Tom" }).One(&user)
   if err != nil {
     panic(err)
   }
   fmt.Println(user)
   // output: {Tom 20}
 
   // 用bson.M结构接收，当你不了解返回的数据结构格式时，可以用这个先查看，然后再定义struct格式
   // 在处理mongodb组合查询时，经常这么干
   result := bson.M{}
   err = c.Find(nil).One(&result)
   if err != nil {
     panic(err)
   }
   fmt.Println(result)
   // output: map[_id:ObjectIdHex("56fdce98189df8759fd61e5b") name:Tom age:20]
 
}

查找多条记录

func findMuit() {
   db := getDB()
 
   c := db.C( "user" )
 
   // 使用All方法，一次性消耗较多内存，如果数据较多，可以考虑使用迭代器
   type User struct {
     Id   bson.ObjectId `bson: "_id,omitempty" `
     Name string     "bson:`name`"
     Age  int      "bson:`age`"
   }
   var users []User
   err := c.Find(nil).All(&users)
   if err != nil {
     panic(err)
   }
   fmt.Println(users)
   // output: [{ObjectIdHex("56fdce98189df8759fd61e5b") Tom 20}...]
 
   // 使用迭代器获取数据可以避免一次占用较大内存
   var user User
   iter := c.Find(nil).Iter()
   for iter.Next(&user) {
     fmt.Println(user)
   }
   // output:
   // {ObjectIdHex("56fdce98189df8759fd61e5b") Tom 20}
   // {ObjectIdHex("56fdce98189df8759fd61e5c") Tom 20}
   // ...
}

查找指定字段
　　主要使用Select函数：

func (q *Query) Select(selector interface{}) *Query

func findField() {
   db := getDB()
 
   c := db.C( "user" )
 
   // 只读取name字段
   type User struct {
     Name string  "bson:`name`"
   }
   var users []User
   err := c.Find(bson.M{}).Select(bson.M{ "name" :  1 }).All(&users)
   if err != nil {
     panic(err)
   }
   fmt.Println(users)
   // output: [{Tom} {Tom} {Anny}...]
 
   // 只排除_id字段
   type User2 struct {
     Name string  "bson:`name`"
     Age  int   "bson:`age`"
   }
   var users2 []User2
   err = c.Find(bson.M{}).Select(bson.M{ "_id" :  0 }).All(&users2)
   if err != nil {
     panic(err)
   }
   fmt.Println(users2)
   // output: [{Tom 20} {Tom 20} {Anny 28}...]
 
}

查询嵌套格式数据

func findNesting() {
   db := getDB()
 
   c := db.C( "user" )
 
   // 使用嵌套的struct接收数据
   type User struct {
     Name string  "bson:`name`"
     Age  int   "bson:`age`"
     Toys []struct {
       Name string  "bson:`name`"
     }
   }
   var users User
   // 只查询toys字段存在的
   err := c.Find(bson.M{ "toys" : bson.M{ "$exists" : true}}).One(&users)
   if err != nil {
     panic(err)
   }
   fmt.Println(users)
   // output: {Tom 20 [{dog}]}
}

查找数据总数

func count() {
   db := getDB()
 
   c := db.C( "user" )
 
   // 查找表总数
   count, err := c.Count()
   if err != nil {
     panic(err)
   }
   fmt.Println(count)
   // output: 8
 
   // 结合find条件查找
   count, err = c.Find(bson.M{ "name" :  "Tom" }).Count()
   if err != nil {
     panic(err)
   }
   fmt.Println(count)
   // output: 6
 
}

对数据进行排序
　　使用Sort函数

func (q *Query) Sort(fields ...string) *Query

func findSort() {
   db := getDB() 
   c := db.C( "user" ) 
   type User struct {
     Id   bson.ObjectId `bson: "_id,omitempty" `
     Name string     "bson:`name`"
     Age  int      "bson:`age`"
   }
   var users []User
   // 按照age字段降序排列，如果升序去掉横线"-"就可以了
   err := c.Find(nil).Sort( "-age" ).All(&users)
   if err != nil {
     panic(err)
   }
   fmt.Println(users)
   // output:
   // [{ObjectIdHex("56fdce98189df8759fd61e5d") Anny 28} ...]
   // ...
}

分页查询
　　使用Skip函数和Limit函数

func (q *Query) Skip(n int) *Query
func (q *Query) Limit(n int) *Query

func findPage() {
   db := getDB() 
   c := db.C( "user" ) 
   type User struct {
     Id   bson.ObjectId `bson: "_id,omitempty" `
     Name string     "bson:`name`"
     Age  int      "bson:`age`"
   }
   var users []User
   // 表示从偏移位置为2的地方开始取两条记录
   err := c.Find(nil).Sort( "-age" ).Skip( 2 ).Limit( 2 ).All(&users)
   if err != nil {
     panic(err)
   }
   fmt.Println(users)
   // output:
   // [{ObjectIdHex("56fdce98189df8759fd61e5d") Anny 20} ...]
   // ...
}

　　以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。