欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-04-29 14:11:16
前言 golang 对mongodb的操作简单封装 使用mongodb的go驱动库 ,对mongodb的操作做一下简单封装 mgo(音mango)是mongodb的g...

前言

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

 ()

总结

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