Golang对MongoDB数据库的操作简单封装教程
前言
Golang对MongoDB的操作简单封装
使用MongoDB的Go驱动库mgo,对MongoDB的操作做一下简单封装
mgo（音mango）是MongoDB的Go语言驱动，它用基于Go语法的简单API实现了丰富的特性，并经过良好测试。
初始化
操作没有用户权限的MongoDB
varglobalS*mgo.Session
funcinit(){
s,err:=mgo.Dial(dialInfo)
iferr!=nil{
log.Fatalf("CreateSession:%s\n",err)
}
globalS=s
}
如果MongoDB设置了用户权限需要使用下面的方法操作
funcinit(){
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)
iferr!=nil{
log.Fatalf("CreateSession:%s\n",err)
}
globalS=s
}
连接具体的数据和文档
每一次操作都copy一份Session,避免每次创建Session,导致连接数量超过设置的最大值
获取文档对象c:=Session.DB(db).C(collection)
funcconnect(db,collectionstring)(*mgo.Session,*mgo.Collection){
ms:=globalS.Copy()
c:=ms.DB(db).C(collection)
ms.SetMode(mgo.Monotonic,true)
returnms,c
}
插入数据
每次操作之后都要主动关闭SessiondeferSession.Close()
db:操作的数据库
collection:操作的文档(表)
doc:要插入的数据
funcInsert(db,collectionstring,docinterface{})error{
ms,c:=connect(db,collection)
deferms.Close()
returnc.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:查询到的结果
funcFindOne(db,collectionstring,query,selector,resultinterface{})error{
ms,c:=connect(db,collection)
deferms.Close()
returnc.Find(query).Select(selector).One(result)
}
funcFindAll(db,collectionstring,query,selector,resultinterface{})error{
ms,c:=connect(db,collection)
deferms.Close()
returnc.Find(query).Select(selector).All(result)
}
//test查询title="标题",并且返回结果中去除`_id`字段
varresultData
err=db.FindOne(database,collection,bson.M{"title":"标题"},bson.M{"_id":0},&result)
更新数据
db:操作的数据库
collection:操作的文档(表)
selector:更新条件
update:更新的操作
funcUpdate(db,collectionstring,selector,updateinterface{})error{
ms,c:=connect(db,collection)
deferms.Close()
returnc.Update(selector,update)
}
//更新，如果不存在就插入一个新的数据`upsert:true`
funcUpsert(db,collectionstring,selector,updateinterface{})error{
ms,c:=connect(db,collection)
deferms.Close()
_,err:=c.Upsert(selector,update)
returnerr
}
//`multi:true`
funcUpdateAll(db,collectionstring,selector,updateinterface{})error{
ms,c:=connect(db,collection)
deferms.Close()
_,err:=c.UpdateAll(selector,update)
returnerr
}
//test
err=db.Update(database,collection,bson.M{"_id":"5b3c30639d5e3e24b8786540"},bson.M{"$set":bson.M{"title":"更新标题"}})
删除数据
db:操作的数据库
collection:操作的文档(表)
selector:删除条件
funcRemove(db,collectionstring,selectorinterface{})error{
ms,c:=connect(db,collection)
deferms.Close()
returnc.Remove(selector)
}
funcRemoveAll(db,collectionstring,selectorinterface{})error{
ms,c:=connect(db,collection)
deferms.Close()
_,err:=c.RemoveAll(selector)
returnerr
}
//test
err=db.Remove(database,collection,bson.M{"_id":"5b3c30639d5e3e24b8786540"})
分页查询
db:操作的数据库
collection:操作的文档(表)
page:当前页面
limit:每页的数量值
query:查询条件
selector:需要过滤的数据(projection)
result:查询到的结果
funcFindPage(db,collectionstring,page,limitint,query,selector,resultinterface{})error{
ms,c:=connect(db,collection)
deferms.Close()
returnc.Find(query).Select(selector).Skip(page*limit).Limit(limit).All(result)
}
其他操作
funcIsEmpty(db,collectionstring)bool{
ms,c:=connect(db,collection)
deferms.Close()
count,err:=c.Count()
iferr!=nil{
log.Fatal(err)
}
returncount==0
}
funcCount(db,collectionstring,queryinterface{})(int,error){
ms,c:=connect(db,collection)
deferms.Close()
returnc.Find(query).Count()
}
完整的代码请参考 （本地下载）
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对毛票票的支持。