• 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方法

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    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
    }

    查找单条记录

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    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]
     
    }

    查找多条记录

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    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
    
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    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}...]
     
    }

    查询嵌套格式数据

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    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}]}
    }

    查找数据总数

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    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
    
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    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
    
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    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} ...]
        // ...
    }
  • 相关阅读:
    [Python]爬虫v0.1
    [Python]同是新手的我,分享一些经验
    [python]闭包到底是什么鬼?
    测试Flask应用_学习笔记
    Flask模板_学习笔记
    SQL Server Alwayson概念总结
    JDBC数据库编程:ResultSet接口
    JDBC操作,执行数据库更新操作
    接口怎么实例化?
    java数据库编程:JDBC操作及数据库
  • 原文地址:https://www.cnblogs.com/williamjie/p/9598748.html
Copyright © 2020-2023  润新知