golang使用mongoDB

下面是我的模块代码

你可以go get github.com/yihubaikai/gopublic/mongo

里面下载

package mongo

import (
    "fmt"
    "github.com/yihubaikai/gopublic"
    "gopkg.in/mgo.v2"
    "gopkg.in/mgo.v2/bson"
    //"log"
)

//单条数据节点
type DataNode struct {
    Nick      string `json:"nick"`
    Class     string `json:"class"`
    Flag      int    `json:"flag"`
    Starttime string `json:"starttime"`
}

type Person struct {
    NAME  string
    PHONE string
}
type Men struct {
    Persons []Person
}

var session *mgo.Session
var database *mgo.Database = nil

// get mongodb
//db: data
//host： 127.0.0.1
//
func GetDB() *mgo.Database {
    if database == nil {
        session, err := mgo.Dial("127.0.0.1:27017")
        if err != nil {
            panic(err)
        }
        //defer session.close()
        session.SetMode(mgo.Monotonic, true)
        database = session.DB("data")
        return database
    } else {
        return database
    }

    /*
        var err error
            dialInfo := &mgo.DialInfo{
                Addrs:     []string{config.Hosts},
                Direct:    false,
                Timeout:   time.Second * 1,
                PoolLimit: 4096, // Session.SetPoolLimit    }
            //创建一个维护套接字池的session
            session, err = mgo.DialWithInfo(dialInfo)

            if err != nil {
                log.Println(err.Error())
            }
            session.SetMode(mgo.Monotonic, true)
            //使用指定数据库
            database = session.DB(config.Database)
    */
}

//插入记录
func Insert(db *mgo.Database, item, nick, class string) bool {
    c := db.C(item)
    //defer c.Close()
    //result := DataNode{}
    err, _ := c.Upsert(bson.M{"nick": nick}, &DataNode{Nick: nick, Class: class, Flag: 0, Starttime: hPub.Gettime()})
    if err != nil {
        fmt.Println("添加记录成功:", item, nick, class, hPub.Gettime())
        return true
    } else {
        fmt.Println("更新记录成功:", item, nick, class, hPub.Gettime())
        return true
    }
    //fmt.Println(rs)
    return true
}

//获取记录
func GetOne(db *mgo.Database, item string, find, update int) (nick, class string) {
    nick = ""
    class = ""
    c := db.C(item)
    result := DataNode{}
    err := c.Find(bson.M{"flag": find}).One(&result)
    if err == nil {
        fmt.Println("nick", result.Nick, "class", result.Class)
        nick = result.Nick
        class = result.Class

        //获取记录了之后更新一下记录
        //err = c.Update(bson.M{"nick": nick}, bson.M{"$set": bson.M{"flag": 2, "starttime": hPub.Gettime()}})
        changeInfo, err := c.UpdateAll(bson.M{"nick": nick}, bson.M{"$set": bson.M{"flag": update}})
        if err == nil {
            fmt.Println("获取记录：更新记录成功:", changeInfo)
        } else {
            fmt.Println("获取记录：更新记录失败:", changeInfo)
        }
    } else {
        fmt.Println("获取记录失败")
    }
    return nick, class
}

//--------------------------------------------------------------------
//插入单条数据
func Insert2(db *mgo.Database) {
    //db := GetDB()

    c := db.C("user")
    type User struct {
        Name string "bson:`name`"
        Age  int    "bson:`age`"
    }

    err := c.Insert(&User{Name: "Tom", Age: 20})
    if err != nil {
        panic(err)
    }
    fmt.Println(err)
}
func Insert3(db *mgo.Database, item, nick, class string) bool {
    c := db.C(item)
    result := DataNode{}
    err := c.Find(bson.M{"nick": nick}).One(&result)
    if err != nil {
        //err := c.Upsert(&DataNode{Nick: nick, Class: class, Flag: 0, Starttime: hPub.Gettime()})
        err, _ := c.Upsert(bson.M{"nick": nick}, &DataNode{Nick: nick, Class: class, Flag: 0, Starttime: hPub.Gettime()})
        if err != nil {
            fmt.Println("添加记录成功:", item, nick, class, hPub.Gettime())
            return true
        } else {
            fmt.Println("更新记录成功:", item, nick, class, hPub.Gettime())
            return true
        }
    } else {
        fmt.Println("可能存在")
        return true
    }
}

//一次插入多条记录
func insertMuti() {
    db := GetDB()

    c := db.C("user")
    type User struct {
        Name string "bson:`name`"
        Age  int    "bson:`age`"
    }

    err := c.Insert(&User{Name: "Tom", Age: 20}, &User{Name: "Anny", Age: 28})
    if err != nil {
        panic(err)
    }
    fmt.Println(err)
}

//插入数组格式数据
func insertArray() {
    db := GetDB()
    c := db.C("user")

    type User struct {
        Name   string   "bson:`name`"
        Age    int      "bson:`age`"
        Groups []string "bson:`groups`"
    }

    err := c.Insert(&User{
        Name:   "Tom",
        Age:    20,
        Groups: []string{"news", "sports"},
    })
    if err != nil {
        panic(err)
    }
    fmt.Println(err)
}

//插入嵌套数据
func insertNesting() {
    db := GetDB()

    c := db.C("user")

    type Toy struct {
        Name string "bson:`name`"
    }
    type User struct {
        Name string "bson:`name`"
        Age  int    "bson:`age`"
        Toys []Toy
    }

    err := c.Insert(&User{
        Name: "Tom",
        Age:  20,
        Toys: []Toy{{Name: "dog"}},
    })
    if err != nil {
        panic(err)
    }
    fmt.Println(err)
}

//插入map格式的数据
func insertMap() {
    db := GetDB()
    c := db.C("user")

    user := map[string]interface{}{
        "name":   "Tom",
        "age":    20,
        "groups": []string{"news", "sports"},
        "toys": []map[string]interface{}{
            {
                "name": "dog",
            },
        },
    }

    err := c.Insert(&user)
    if err != nil {
        panic(err)
    }
    fmt.Println(err)
}

//插入关联其它集合ObjectId的数据
//要使用bson.ObjectIdHex函数对字符串进行转化，bson.ObjectIdHex函数原型
func insertObjectId() {
    db := GetDB()
    c := db.C("user")

    user := map[string]interface{}{
        "name":     "Tom",
        "age":      20,
        "group_id": bson.ObjectIdHex("540046baae59489413bd7759"),
    }

    err := c.Insert(&user)
    if err != nil {
        panic(err)
    }
    fmt.Println(err)
}

func QueryOne() {

    db := GetDB()
    c := db.C("user")
    //*****查询单条数据*******
    result := Person{}
    err := c.Find(bson.M{"NAME": "456"}).One(&result)
    if err != nil {
        panic(err)
    }
    fmt.Println("Phone:", result.NAME, result.PHONE)

}

func QueryMutil() {
    db := GetDB()
    c := db.C("user")
    //*****查询多条数据*******
    result := Person{}
    var personAll Men //存放结果
    iter := c.Find(nil).Iter()
    for iter.Next(&result) {
        fmt.Printf("Result: %v ", result.NAME)
        personAll.Persons = append(personAll.Persons, result)
    }
}