Golang 對MongoDB的操作簡單封裝
使用MongoDB的Go驅動庫 mgo,對MongoDB的操作做一下簡單封裝
初始化
- 操作沒有使用者許可權的
MongoDB
var globalS *mgo.Session
func init() {
s, err := mgo.Dial(dialInfo)
if err != nil {
log.Fatalf("Create Session: %s
", 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
", 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, 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)
複製程式碼更新資料
db:操作的資料庫
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)
}
//更新，如果不存在就插入一個新的資料 `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": "更新標題"}})
複製程式碼刪除資料
db:操作的資料庫
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:操作的資料庫
collection:操作的文件(表)
page:當前頁面
limit:每頁的數量值
query:查詢條件
selector:需要過濾的資料(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()
}
複製程式碼