1. 程式人生 > 資料庫 >Golang對MongoDB資料庫的操作簡單封裝教程

Golang對MongoDB資料庫的操作簡單封裝教程

前言

Golang 對MongoDB的操作簡單封裝

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

mgo(音mango)是MongoDB的Go語言驅動,它用基於Go語法的簡單API實現了豐富的特性,並經過良好測試。

初始化

操作沒有使用者許可權的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},//資料庫地址 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)

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:要插入的資料

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: "部落格的內容資訊",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:查詢到的結果

func FindOne(db,query,selector,result interface{}) error {
 ms,collection)
 defer ms.Close()

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

func FindAll(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)

更新資料

db:操作的資料庫

collection:操作的文件(表)

selector:更新條件

update:更新的操作

func Update(db,update interface{}) error {
 ms,collection)
 defer ms.Close()

 return c.Update(selector,update)
}

//更新,如果不存在就插入一個新的資料 `upsert:true`
func Upsert(db,collection)
 defer ms.Close()

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

// `multi:true`
func UpdateAll(db,err := c.UpdateAll(selector,update)
 return err
}

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

刪除資料

db:操作的資料庫

collection:操作的文件(表)

selector:刪除條件

func Remove(db,selector interface{}) error {
 ms,collection)
 defer ms.Close()

 return c.Remove(selector)
}

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

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

分頁查詢

db:操作的資料庫

collection:操作的文件(表)

page:當前頁面

limit:每頁的數量值

query:查詢條件

selector:需要過濾的資料(projection)

result:查詢到的結果

func FindPage(db,page,limit int,collection)
 defer ms.Close()

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

其他操作

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

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

完整的程式碼請參考(本地下載)

總結

以上就是這篇文章的全部內容了,希望本文的內容對大家的學習或者工作具有一定的參考學習價值,如果有疑問大家可以留言交流,謝謝大家對我們的支援。