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)

插入数据

每次操作之后都要主动关闭 Session defer Session.Close()

db:操作的数据库

collection:操作的文档(表)

doc:要插入的数据

查询数据

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

完整的代码请参考（本地下载）

总结

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