解析 JSON 日期字符串为 BSON 日期的方法在 Golang 中

大家好，我们又见面了啊~本文《解析 JSON 日期字符串为 BSON 日期的方法在 Golang 中》的内容中将会涉及到等等。如果你正在学习Golang相关知识，欢迎关注我，以后会给大家带来更多Golang相关文章，希望我们能一起进步！下面就开始本文的正式内容~

这应该很容易，但我没有取得太大进展。假设我有带有 utc 日期的 json，如下所示：

{
  "name": "lex",
  "dob": "2022-11-01t06:30:30.639326208z"
}

我想将它插入到 mongodb 集合中。在 go 中，我这样做：

import (
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo"
)

...

var doc any
// by the way what does the 2nd param do?  docs don't mention it
_ = bson.UnmarshalExtJSON(jsonBytes, false, &doc)
mongoCollection.InsertOne(context.TODO(), doc)

插入有效，但 dob 字段只是一个字符串。将其插入为日期的正确方法是什么？我的限制是：

1. 我无法更改输入 json
2. 我无法手动编写 mongo 文档，因为我可能会处理大型不可预测的对象，并且不知道哪些字段/嵌套字段具有日期字符串

这可行吗？看来如何做到这一点必须有一条铺好的路。

正确答案

可以定义一个自定义解码器bsoncodec.ValueDecoder，将日期字符串解码为time.time，然后通过RegisterTypeDecoder注册

func datetimedecodevalue(dc bsoncodec.decodecontext, vr bsonrw.valuereader, val reflect.value) error {
    ts, _ := vr.readstring()
    t, err := time.parse(time.rfc3339nano, ts)
    if err != nil {
        return err
    } 

    val.set(reflect.valueof(t))

    return nil
}

var tdatetime = reflect.typeof(time.time{})

type testdoc struct {
    name string    `bson:"name"`
    dob  time.time `bson:"dob"`
}

func createcustomregistry() *bsoncodec.registrybuilder {
    var primitivecodecs bson.primitivecodecs

    rb := bsoncodec.newregistrybuilder()
    bsoncodec.defaultvalueencoders{}.registerdefaultencoders(rb)
    bsoncodec.defaultvaluedecoders{}.registerdefaultdecoders(rb)
    rb.registertypedecoder(tdatetime, bsoncodec.valuedecoderfunc(datetimedecodevalue))
    primitivecodecs.registerprimitivecodecs(rb)
    return rb
}

或者将日期字符串解析为 bson 日期示例代码

func datetimedecodevalue(dc bsoncodec.decodecontext, vr bsonrw.valuereader, val reflect.value) error {
    ts, _ := vr.readstring()
    t, err := time.parse(time.rfc3339nano, ts)
    if err != nil {
        val.setstring(ts)
    } else {
        val.set(reflect.valueof(primitive.datetime(t.unixmilli())))
    }

    return nil
}

var tdatetime = reflect.typeof(primitive.datetime(0))

type testdoc struct {
    name string             `bson:"name"`
    dob  primitive.datetime `bson:"dob"`
}

func createcustomregistry() *bsoncodec.registrybuilder {
    var primitivecodecs bson.primitivecodecs

    rb := bsoncodec.newregistrybuilder()
    bsoncodec.defaultvalueencoders{}.registerdefaultencoders(rb)
    bsoncodec.defaultvaluedecoders{}.registerdefaultdecoders(rb)
    rb.registertypedecoder(tdatetime, bsoncodec.valuedecoderfunc(datetimedecodevalue))
    primitivecodecs.registerprimitivecodecs(rb)
    return rb
}

然后通过UnmarshalExtJSONWithRegistry使用这个寄存器

func main() {
    jsonBytes := []byte(`{
       "name": "ben",
       "dob": "2022-11-01T06:30:30.639326208Z"
    }`)

    var customRegistry = createCustomRegistry().Build()

    var doc TestDoc
    _ = bson.UnmarshalExtJSONWithRegistry(customRegistry, jsonBytes, false, &doc)
    fmt.Println(doc)
}

完整代码 PLAYGROUND

关于canonical参数请参考doc

理论要掌握，实操不能落！以上关于《解析 JSON 日期字符串为 BSON 日期的方法在 Golang 中》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！