如何用Mongo-Driver有效地将BSON转换成JSON?

qyyhg6bp  于 2022-09-18  发布在  Go
关注(0)|答案(1)|浏览(281)

我想有效地将mongo-go-driver中的bson转换为json。

我应该小心处理NaN,因为如果数据中存在NaN,则json.Marshal会失败。

例如,我想将下面的bson数据转换为json。

b, _ := bson.Marshal(bson.M{"a": []interface{}{math.NaN(), 0, 1}})
// How to convert b to json?

以下是失败的。

// decode
var decodedBson bson.M
bson.Unmarshal(b, &decodedBson)
_, err := json.Marshal(decodedBson)
if err != nil {
    panic(err) // it will be invoked
    // panic: json: unsupported value: NaN
}
qij5mzcb

qij5mzcb1#

如果您了解BSON的结构,则可以创建实现json.Marshalerjson.Unmarshaler接口的自定义类型,并随心所欲地处理NaN。例如:

type maybeNaN struct{
    isNan  bool
    number float64
}

func (n maybeNaN) MarshalJSON() ([]byte, error) {
    if n.isNan {
        return []byte("null"), nil // Or whatever you want here
    }
    return json.Marshal(n.number)
}

func (n *maybeNan) UnmarshalJSON(p []byte) error {
    if string(p) == "NaN" {
        n.isNan = true
        return nil
    }
    return json.Unmarshal(p, &n.number)
}

type myStruct struct {
    someNumber maybeNaN `json:"someNumber" bson:"someNumber"`
    /* ... */
}

如果您的BSON具有任意结构,您唯一的选择是使用反射遍历该结构,并将出现的任何NaN转换为类型(可能是如上所述的自定义类型)

相关问题