Golang 中操做 Mongo Update 的方法

Golang 和 MongoDB 中的 ISODate 时间交互问题

2018年02月27日 11:28:43 独一无二的小个性 阅读数：357 标签： Golang MongoDB 时间交互 时间转换  更多

我的分类： MongoDB Golang

 

版权声明：本文为博主原创文章，未经博主容许不得转载。 https://blog.csdn.net/u010649766/article/details/79385948

MongoDB 中有一种时间格式数据 ISODate，参考以下： 

若是在 Golang 中查询这条记录，Golang用什么类型的变量来保存呢？

查找 ISODate 时间字段

在 Golang 中可使用 time.Time 数据类型来保存 MongoDB 中的 ISODate 时间。

type Model struct {
    Id   bson.ObjectId `bson:"_id,omitempty"` Time time.Time `bson:"time"` } m := Model{} err := c.Find(bson.M{"_id": bson.ObjectIdHex("572f3c68e43001d2c1703aa7")}).One(&m) if err != nil { panic(err) } fmt.Printf("%+v\n", m) // output: {Id:ObjectIdHex("572f3c68e43001d2c1703aa7") Time:2015-07-08 17:29:14.002 +0800 CST}

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11

从输出中能够看到 Golang 输出的时间格式是 CST 时区，Golang 在处理的过程当中将 ISO 时间转换成了 CST 时间，从时间面板上来看也比 MongoDB 中的快上 8 个小时，这个是正常的。

那么 Golang 作插入操做和或者时间比较操做的时候须要本身转换时间戳吗？答案是不须要的，来看下插入的例子。

插入时间

从新插入一条记录，记录的Time字段为当前时间，在golang中能够经过time.Now获取当前时间，查看输出能够看到是CST的时间格式

now := time.Now()
fmt.Printf("%+v\n", now) // output: 2016-05-12 14:34:00.998011694 +0800 CST err = c.Insert(Model{Time: now}) if err != nil { panic(err) }

• 1
• 2
• 3
• 4
• 5
• 6
• 7

查看 MongoDB 中的记录

插入当前时间到 MongoDB: 

能够看到存储到 MongoDB 中时间的自动转换为了 ISO 时间，时间少了 8 个小时。小结一下就是 Golang 和 MongoDB 中的时间交互不须要考虑额外的东西，由于驱动都进行了转换。

时间字符串转成 time.Time

有时咱们会将 time.Time 的时间以字符串的形式存储，那么要和 MongoDB 交互的时候就须要转换 time.Time 格式

// 时间字符串转到到time.Time格式 // 使用time.Parse方法进行转换 timeString := "2016-05-12 14:34:00.998011694 +0800 CST" t, err := time.Parse("2006-01-02 15:04:05.999999999 -0700 MST", timeString) if err != nil { panic(err) } fmt.Printf("%+v\n", t)

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8

代码中比较难理解的就是 time.Parse 的第一个参数，这个实际上是 Golang 当中的定义，详细看下 time.Time.String() 的源码就会明白了。

 

 

 

 

这篇博客主要记录下 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>}