golang mongoldb

Steven_25bb · · 594 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

使用gopkg.in/mgo.v2库操作,插入操作主要使用mongodb中Collection对象函数原型

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

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

     type M map[string]interface{}

     例如:bson.M{"name": "Tom"}相当直接mongodb的查询条件{"name": "Tom"}

连接数据库

     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 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{"$and": []bson.M{bson.M{"name": bson.M{"$ne":"zhangsan"}},
        bson.M{"status": bson.M{"$ne": "lisi"}}}}).All(&users)
    if err != nil {
        panic(err)
    }
    fmt.Println(users)
    // output: {Tom 20 [{dog}]}
}

查询年龄的总合

      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`"
        }
    }

//返回数据
type Resp struct{
   ID string `json:"_id" bson:"_id"`
  AgeCount int64 `json:"agecount" bson:"agecount"`
}
    var resp Resp
    // 只查询toys字段存在的
    err := c.Pipe([]bson.M{
        {
            "$match": bson.M{
                "name":     "zhangsan",
        },
        {
            "$group": bson.M{
                "_id": "null",
                "amount": bson.M{
                    "$sum": "$age",
                },
            },
        },
    }).One(&resp)
    if err != nil {
        panic(err)
    }
    fmt.Println(resp)
    // 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} ...]
    // ...
}

注意:这里其他的插入操作略


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:Steven_25bb

查看原文:golang mongoldb

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

594 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传