快捷導(dǎo)航

Golang對MongoDB數(shù)據(jù)庫的操作簡單封裝教程

更新時間：2018年07月05日 10:53:01 作者：CodeMiner

mongodb官方?jīng)]有關(guān)于go的mongodb的驅(qū)動，因此只能使用第三方驅(qū)動，mgo就是使用最多的一種。下面這篇文章主要給大家介紹了關(guān)于利用Golang對MongoDB數(shù)據(jù)庫的操作簡單封裝的相關(guān)資料，需要的朋友可以參考下

前言

Golang 對MongoDB的操作簡單封裝

使用MongoDB的Go驅(qū)動庫 mgo,對MongoDB的操作做一下簡單封裝

mgo（音mango）是MongoDB的Go語言驅(qū)動，它用基于Go語法的簡單API實現(xiàn)了豐富的特性，并經(jīng)過良好測試。

初始化

操作沒有用戶權(quán)限的MongoDB

var globalS *mgo.Session

func init() {
 s, err := mgo.Dial(dialInfo)
 if err != nil {
 log.Fatalf("Create Session: %s\n", err)
 }
 globalS = s
}

如果MongoDB設(shè)置了用戶權(quán)限需要使用下面的方法操作

func init() {
 dialInfo := &mgo.DialInfo{
 Addrs: []string{dbhost}, //數(shù)據(jù)庫地址 dbhost: mongodb://user@123456:127.0.0.1:27017
 Timeout: timeout,  // 連接超時時間 timeout: 60 * time.Second
 Source: authdb,  // 設(shè)置權(quán)限的數(shù)據(jù)庫 authdb: admin
 Username: authuser,  // 設(shè)置的用戶名 authuser: user
 Password: authpass, // 設(shè)置的密碼 authpass: 123456
 PoolLimit: poollimit, // 連接池的數(shù)量 poollimit: 100
 }

 s, err := mgo.DialWithInfo(dialInfo)
 if err != nil {
 log.Fatalf("Create Session: %s\n", err)
 }
 globalS = s
}

連接具體的數(shù)據(jù)和文檔

每一次操作都copy一份 Session,避免每次創(chuàng)建Session,導(dǎo)致連接數(shù)量超過設(shè)置的最大值

獲取文檔對象 c := Session.DB(db).C(collection)

func connect(db, collection string) (*mgo.Session, *mgo.Collection) {
 ms := globalS.Copy()
 c := ms.DB(db).C(collection)
 ms.SetMode(mgo.Monotonic, true)
 return ms, c
}

插入數(shù)據(jù)

每次操作之后都要主動關(guān)閉 Session defer Session.Close()

db:操作的數(shù)據(jù)庫

collection:操作的文檔(表)

doc:要插入的數(shù)據(jù)

func Insert(db, collection string, doc interface{}) error {
 ms, c := connect(db, collection)
 defer ms.Close()

 return c.Insert(doc)
}

// test
data := &Data{
 Id: bson.NewObjectId().Hex(),
 Title: "標(biāo)題",
 Des: "博客描述信息",
 Content: "博客的內(nèi)容信息",
 Img: "https://upload-images.jianshu.io/upload_images/8679037-67456031925afca6.png?imageMogr2/auto-orient/strip%7CimageView2/2/w/700",
 Date: time.Now(),
}

err := db.Insert("Test", "TestModel", data)

查詢數(shù)據(jù)

db:操作的數(shù)據(jù)庫

collection:操作的文檔(表)

query:查詢條件

selector:需要過濾的數(shù)據(jù)(projection)

result:查詢到的結(jié)果

func FindOne(db, collection string, query, selector, result interface{}) error {
 ms, c := connect(db, collection)
 defer ms.Close()

 return c.Find(query).Select(selector).One(result)
}

func FindAll(db, collection string, query, selector, result interface{}) error {
 ms, c := connect(db, collection)
 defer ms.Close()

 return c.Find(query).Select(selector).All(result)
}

//test 查詢title="標(biāo)題",并且返回結(jié)果中去除`_id`字段
var result Data
err = db.FindOne(database, collection, bson.M{"title": "標(biāo)題"}, bson.M{"_id":0}, &result)

更新數(shù)據(jù)

db:操作的數(shù)據(jù)庫

collection:操作的文檔(表)

selector:更新條件

update:更新的操作

func Update(db, collection string, selector, update interface{}) error {
 ms, c := connect(db, collection)
 defer ms.Close()

 return c.Update(selector, update)
}

//更新，如果不存在就插入一個新的數(shù)據(jù) `upsert:true`
func Upsert(db, collection string, selector, update interface{}) error {
 ms, c := connect(db, collection)
 defer ms.Close()

 _, err := c.Upsert(selector, update)
 return err
}

// `multi:true`
func UpdateAll(db, collection string, selector, update interface{}) error {
 ms, c := connect(db, collection)
 defer ms.Close()

 _, err := c.UpdateAll(selector, update)
 return err
}

//test
err = db.Update(database, collection, bson.M{"_id": "5b3c30639d5e3e24b8786540"}, bson.M{"$set": bson.M{"title": "更新標(biāo)題"}})

刪除數(shù)據(jù)

db:操作的數(shù)據(jù)庫

collection:操作的文檔(表)

selector:刪除條件

func Remove(db, collection string, selector interface{}) error {
 ms, c := connect(db, collection)
 defer ms.Close()

 return c.Remove(selector)
}

func RemoveAll(db, collection string, selector interface{}) error {
 ms, c := connect(db, collection)
 defer ms.Close()

 _, err := c.RemoveAll(selector)
 return err
}

//test
err = db.Remove(database,collection,bson.M{"_id":"5b3c30639d5e3e24b8786540"})

分頁查詢

db:操作的數(shù)據(jù)庫

collection:操作的文檔(表)

page:當(dāng)前頁面

limit:每頁的數(shù)量值

query:查詢條件

selector:需要過濾的數(shù)據(jù)(projection)

result:查詢到的結(jié)果

func FindPage(db, collection string, page, limit int, query, selector, result interface{}) error {
 ms, c := connect(db, collection)
 defer ms.Close()

 return c.Find(query).Select(selector).Skip(page * limit).Limit(limit).All(result)
}

其他操作

func IsEmpty(db, collection string) bool {
 ms, c := connect(db, collection)
 defer ms.Close()
 count, err := c.Count()
 if err != nil {
 log.Fatal(err)
 }
 return count == 0
}

func Count(db, collection string, query interface{}) (int, error) {
 ms, c := connect(db, collection)
 defer ms.Close()
 return c.Find(query).Count()
}

完整的代碼請參考（本地下載）

總結(jié)

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。

您可能感興趣的文章: