Golang 对MongoDB的操作简单封装
MongoDBMongoDB
初始化
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 }
连接具体的数据和文档
SessionSessionc := 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 }
插入数据
Sessiondefer Session.Close()
func Insert(db, collection string, doc interface{}) error { ms, c := connect(db, collection) defer ms.Close() return c.Insert(doc) }// testdata := &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 }//testerr = 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 }//testerr = 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() }