使用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} ...]
// ...
}
转载请注明：快乐编程 » golang mongodb查找find demo
