自定义 BSON 封送处理
自定义 BSON 封送处理类似于自定义 JSON 封送处理，但您需要实现 bson.Getter 和 bson .Setter 接口代替。这些接口允许您定义自定义类型如何与 BSON 数据相互转换。
例如，假设您有一个需要自定义封送处理的货币类型：
package yourpackage
import (
"context"
"errors"
"github.com/globalsign/mgo/bson"
"github.com/shopspring/decimal"
)
type Currency struct {
Value decimal.Decimal
CurrencyCode string
}
func (c Currency) GetBSON() (interface{}, error) {
if c.CurrencyCode == "" {
return nil, errors.New("CurrencyCode cannot be empty")
}
return bson.M{
"value": c.Value.String(),
"currencyCode": c.CurrencyCode,
}, nil
}
func (c *Currency) SetBSON(raw bson.Raw) error {
var m bson.M
if err := raw.Unmarshal(&m); err != nil {
return err
}
val, ok := m["value"]
if !ok {
return errors.New("missing 'value' field in BSON data")
}
c.Value, _ = decimal.NewFromString(val.(string))
code, ok := m["currencyCode"]
if !ok {
return errors.New("missing 'currencyCode' field in BSON data")
}
c.CurrencyCode, _ = code.(string)
return nil
}与通过此实现，您现在可以使用 MGO BSON 编码器注册自定义货币类型：
session := mgo.Dial("mongodb://localhost:27017")
defer session.Close()
session.SetCollectionValidator("products",{
"validator":{
"$jsonSchema":{
"bsonType":"object",
"required":["name","code","price"],
"properties":{
"name":{"bsonType":"string"},
"code":{"bsonType":"string"},
"price":{
"bsonType":"object",
"required":["value", "currencyCode"],
"properties":{
"value":{"bsonType":"string", "pattern":"^[0-9,]+(.[0-9,]+)?$"},
"currencyCode":{"bsonType":"string", "enum":["USD", "EUR", "GBP"]},
},
},
}
}
}
})
collection := session.DB("products").C("products")现在，当您使用货币保存文档时字段到 MongoDB，MGO 编码器将自动使用您的自定义封送处理函数将货币转换为 BSON。
请注意，这只是如何为 BSON 类型添加自定义封送处理的一个示例。具体实现将根据您特定应用的要求而有所不同。
以上是如何在 Go 中实现自定义 BSON 封送处理？的详细内容。更多信息请关注PHP中文网其他相关文章！
