2017年05月07日 01:39:26 极客Geek 阅读数:2780
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/netdxy/article/details/71307383
这篇博客主要记录下 Golang 中引入 Mongo 数据库对 Mongo 的修改操作的方法。
使用gopkg.in/mgo.v2库操作,修改操作主要使用mongodb中Collection对象的Update、UpdateAll、UpdateId、Upsert、UpsertId方法。
统一封装下getDB方法
package main
import (
"fmt"
"gopkg.in/mgo.v2"
"gopkg.in/mgo.v2/bson" ) // get mongodb db func getDB() *mgo.Database { session, err := mgo.Dial("127.0.0.1:27017") if err != nil { panic(err) } session.SetMode(mgo.Monotonic, true) db := session.DB("test") return db }
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
Update修改单条记录
func (c *Collection) Update(selector interface{}, update interface{}) error
- 1
和mysql不同,Update函数只能用来修改单条记录,即使条件能匹配多条记录,也只会修改第一条匹配的记录。
selector := bson.M{"_id": bson.ObjectIdHex("56fdce98189df8759fd61e5b")}
data := bson.M{"age": 21} err := getDB().C("user").Update(selector, data) if err != nil { panic(err) }
- 1
- 2
- 3
- 4
- 5
- 6
类似于mongodb语法:
db.getCollection('user').update(
{ "_id": ObjectId("56fdce98189df8759fd61e5b") },
{ "age": 21} )
- 1
- 2
- 3
- 4
这边有一个小坑,就是这种写法的修改数据会更新整个文档为 {“age”: 21},而不是只更新age的值。
更新前数据:
{
"_id" : ObjectId("56fdce98189df8759fd61e5b"),
"name" : "Tom",
"age" : 20 } 更新后数据: { "_id" : ObjectId("56fdce98189df8759fd61e5b"), "age" : 21 }
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
如果需要只更新age的值,则需要使用$set关键词:
selector := bson.M{"_id": bson.ObjectIdHex("571de968a99cff2c68264807")}
data := bson.M{"$set": bson.M{"age": 21}} err := getDB().C("user").Update(selector, data) if err != nil { panic(err) }
- 1
- 2
- 3
- 4
- 5
- 6
- 7
如果更新的数据不存在,则会报一个not found的错误:
// 更新不存在的数据
selector := bson.M{"_id": bson.ObjectIdHex("16fdce98189df8759fd61e5b")}
data := bson.M{"age": 21} err := getDB().C("user").Update(selector, data) if err != nil { fmt.Println(err == mgo.ErrNotFound) // output: true }
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
由于Update只能更新一条数据,如果需要批量更新则需要使用UpdateAll函数
UpdateAll批量更新
func (c *Collection) UpdateAll(selector interface{}, update interface{}) (info *ChangeInfo, err error)
- 1
- 2
selector := bson.M{"name": "Tom"}
data := bson.M{"$set": bson.M{"age": 22}} changeInfo, err := getDB().C("user").UpdateAll(selector, data) if err != nil { panic(err) } fmt.Printf("%+v\n", changeInfo) // output: &{Updated:2 Removed:0 UpsertedId:<nil>}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
UpdateId根据Id更新
和Update函数类似,就是条件直接换成了objectid,这个使用也挺频繁的。
func (c *Collection) UpdateId(id interface{}, update interface{}) error
类似
err := collection.Update(bson.M{"_id": id}, update)
- 1
- 2
- 3
id := bson.ObjectIdHex("571de968a99cff2c68264807")
data := bson.M{"$set": bson.M{"age": 30}} err := getDB().C("user").UpdateId(id, data) if err != nil { panic(err) }
- 1
- 2
- 3
- 4
- 5
- 6
Upsert更新或者插入数据
这个函数就是如果数据存在就更新,否则就新增一条记录。这个函数也挺常用的。
func (c *Collection) Upsert(selector interface{}, update interface{}) (info *ChangeInfo, err error)
- 1
selector := bson.M{"key": "max"}
data := bson.M{"$set": bson.M{"value": 30}} changeInfo, err := getDB().C("config").Upsert(selector, data) if err != nil { panic(err) } fmt.Printf("%+v\n", changeInfo) // 首次执行output: &{Updated:0 Removed:0 UpsertedId:ObjectIdHex("571df02ea99cff2c6826480a")} // 再次执行output: &{Updated:1 Removed:0 UpsertedId:<nil>}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
UpsertId按照更新或者插入数据
func (c *Collection) UpsertId(id interface{}, update interface{}) (info *ChangeInfo, err error)
类似
info, err := collection.Upsert(bson.M{"_id": id}, update)
- 1
- 2
- 3
id := bson.ObjectIdHex("571df02ea99cff2c6826480b")
data := bson.M{"$set": bson.M{"key": "max", "value": 30}} changeInfo, err := getDB().C("config").UpsertId(id, data) if err != nil { panic(err) } fmt.Printf("%+v\n", changeInfo) // 首次执行output: &{Updated:0 Removed:0 UpsertedId:ObjectIdHex("571df02ea99cff2c6826480b")} // 再次执行output: &{Updated:1 Removed:0 UpsertedId:<nil>}