Golang 對MongoDB的操作簡單封裝

Golang 對MongoDB的操作簡單封裝

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

初始化

• 操作沒有使用者權限的MongoDB

var globalS *mgo.Sessionfunc 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)}// 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:查詢到的結果

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:更新的操作

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:刪除條件

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:查詢到的結果

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()}

完整的代碼請參考