Golang對MongoDB數(shù)據(jù)庫的操作簡單封裝教程
更新時間：2018年07月05日 10:53:01 作者：CodeMiner
mongodb官方?jīng)]有關于go的mongodb的驅動，因此只能使用第三方驅動，mgo就是使用最多的一種。下面這篇文章主要給大家介紹了關于利用Golang對MongoDB數(shù)據(jù)庫的操作簡單封裝的相關資料，需要的朋友可以參考下
前言
Golang 對MongoDB的操作簡單封裝
使用MongoDB的Go驅動庫 mgo,對MongoDB的操作做一下簡單封裝
mgo（音mango）是MongoDB的Go語言驅動，它用基于Go語法的簡單API實現(xiàn)了豐富的特性，并經(jīng)過良好測試。
初始化
操作沒有用戶權限的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設置了用戶權限需要使用下面的方法操作
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ù)據(jù)庫 authdb: admin
Username: authuser, // 設置的用戶名 authuser: user
Password: authpass, // 設置的密碼 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,導致連接數(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ù)
每次操作之后都要主動關閉 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: "標題",
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:查詢到的結果
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 Data
err = db.FindOne(database, collection, bson.M{"title": "標題"}, 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": "更新標題"}})
刪除數(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:當前頁面
limit:每頁的數(shù)量值
query:查詢條件
selector:需要過濾的數(shù)據(jù)(projection)
result:查詢到的結果
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()
}
總結
以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。
相關文章
詳解Go語言中如何通過Goroutine實現(xiàn)高并發(fā)
在Go語言中,并發(fā)編程是一個核心且強大的特性,Go語言通過goroutine和channel等機制,使得并發(fā)編程變得更加簡單和直觀,本文給大家介紹了Go語言中如何通過Goroutine快速實現(xiàn)高并發(fā),感興趣的小伙伴跟著小編一起來看看吧2024-10-10
golang簡易實現(xiàn)?k8s?的yaml上傳并應用示例方案
這篇文章主要為大家介紹了golang簡易實現(xiàn)?k8s?的yaml上傳并應用示例方案,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-07-07
Golang自定義開發(fā)Prometheus?exporter詳解
Exporter是基于Prometheus實施的監(jiān)控系統(tǒng)中重要的組成部分，承擔數(shù)據(jù)指標的采集工作，這篇文章主要為大家介紹了如何自定義編寫開發(fā)?Prometheus?exporter，感興趣的可以了解一下2023-06-06
使用channel來控制子協(xié)程的優(yōu)點是實現(xiàn)簡單，缺點是當需要大量創(chuàng)建協(xié)程時就需要有相同數(shù)量的channel，而且對于子協(xié)程繼續(xù)派生出來的協(xié)程不方便控制2021-07-07
自旋鎖是指當一個線程在獲取鎖的時候，如果鎖已經(jīng)被其他線程獲取，那么該線程將循環(huán)等待，然后不斷地判斷是否能夠被成功獲取，知直到獲取到鎖才會退出循環(huán)2022-10-10
Go?語言有一些非常實用的語法糖（syntactic?sugar）,它們使得代碼更加簡潔和易讀,本文為大家整理了15個常見的語法糖,有需要的可以了解下2025-01-01
這篇文章主要為大家介紹了Go語言快速入門指針Map示例教程，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-08-08
基于Go語言實現(xiàn)的簡易api網(wǎng)關的示例代碼
本文主要介紹了基于Go語言實現(xiàn)的簡易api網(wǎng)關，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-12-12
這篇文章主要介紹了Go?語言入門之Go?計時器，文章基于GO語言的相關資料展開對其中計時器的詳細內(nèi)容。具有一定的參考價值，需要的小伙伴可以參考一下2022-05-05
Golang實現(xiàn)四種負載均衡的算法(隨機,輪詢等)
本文介紹了示例介紹了Golang 負載均衡的四種實現(xiàn)，主要包括了隨機，輪詢，加權輪詢負載，一致性hash，感興趣的小伙伴們可以參考一下2021-06-06