Golang对MongoDB数据库的操作简单封装教程
前言
golang 对mongodb的操作简单封装
使用mongodb的go驱动库 ,对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, 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() }
()
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对的支持。
上一篇: 详解golang consul-grpc 服务注册与发现
下一篇: 卧槽。。。有你这样当爹的嘛