Golang对MongoDB数据库的操作简单封装教程

2020-01-28 13:22:19王旭

更新数据

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

完整的代码请参考 (本地下载)

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对易采站长站的支持。