前言
使用MongoDB的Go驅動庫 mgo,對MongoDB的操作做一下簡單封裝
mgo（音mango）是MongoDB的Go語言驅動，它用基于Go語法的簡單API實現了豐富的特性，并經過良好測試。
初始化
操作沒有用戶權限的MongoDB
|
1
2
3
4
5
6
7
8
9
|
var globalS *mgo.Sessionfunc init() { s, err := mgo.Dial(dialInfo) if err != nil { log.Fatalf("Create Session: %s\n", err) } globalS = s} |
如果MongoDB設置了用戶權限需要使用下面的方法操作
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
func init() { dialInfo := &mgo.DialInfo{ Addrs: []string{dbhost}, //數據庫地址 dbhost: mongodb://user@123456:127.0.0.1:27017 Timeout: timeout, // 連接超時時間 timeout: 60 * time.Second Source: authdb, // 設置權限的數據庫 authdb: admin Username: authuser, // 設置的用戶名 authuser: user Password: authpass, // 設置的密碼 authpass: 123456 PoolLimit: poollimit, // 連接池的數量 poollimit: 100 } s, err := mgo.DialWithInfo(dialInfo) if err != nil { log.Fatalf("Create Session: %s\n", err) } globalS = s} |
連接具體的數據和文檔
每一次操作都copy一份 Session,避免每次創建Session,導致連接數量超過設置的最大值
獲取文檔對象 c := Session.DB(db).C(collection)
|
1
2
3
4
5
6
|
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} |
插入數據
每次操作之后都要主動關閉 Session defer Session.Close()
db:操作的數據庫
collection:操作的文檔(表)
doc:要插入的數據
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
func Insert(db, collection string, doc interface{}) error { ms, c := connect(db, collection) defer ms.Close() return c.Insert(doc)}// testdata := &Data{ Id: bson.NewObjectId().Hex(), Title: "標題", Des: "博客描述信息", Content: "博客的內容信息", 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) |
查詢數據
db:操作的數據庫
collection:操作的文檔(表)
query:查詢條件
selector:需要過濾的數據(projection)
result:查詢到的結果
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
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="標題",并且返回結果中去除`_id`字段var result Dataerr = db.FindOne(database, collection, bson.M{"title": "標題"}, bson.M{"_id":0}, &result) |
更新數據
db:操作的數據庫
collection:操作的文檔(表)
selector:更新條件
update:更新的操作
|
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
|
func Update(db, collection string, selector, update interface{}) error { ms, c := connect(db, collection) defer ms.Close() return c.Update(selector, update)}//更新，如果不存在就插入一個新的數據 `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}//testerr = db.Update(database, collection, bson.M{"_id": "5b3c30639d5e3e24b8786540"}, bson.M{"$set": bson.M{"title": "更新標題"}}) |
刪除數據
db:操作的數據庫
collection:操作的文檔(表)
selector:刪除條件
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
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}//testerr = db.Remove(database,collection,bson.M{"_id":"5b3c30639d5e3e24b8786540"}) |
分頁查詢
db:操作的數據庫
collection:操作的文檔(表)
page:當前頁面
limit:每頁的數量值
query:查詢條件
selector:需要過濾的數據(projection)
result:查詢到的結果
|
1
2
3
4
5
6
|
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)} |
其他操作
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
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()} |
總結
以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對服務器之家的支持。
原文鏈接：https://www.jianshu.com/p/c2aaebf11725
