如何在 Go 中实现自定义 BSON 封送处理？

自定义 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中文网其他相关文章！