处理自定义 BSON 编组

编程并不是一个机械性的工作，而是需要有思考，有创新的工作，语法是固定的，但解决问题的思路则是依靠人的思维，这就需要我们坚持学习和更新自己的知识。今天golang学习网就整理分享《处理自定义 BSON 编组》，文章讲解的知识点主要包括golang，如果你对Golang方面的知识点感兴趣，就不要错过golang学习网，在这可以对大家的知识积累有所帮助，助力开发能力的提升。

问题内容

我有许多需要自定义编组的结构。当我测试时，我使用的是 JSON 和标准的 JSON 编组器。由于它不会编组未导出的字段，因此我需要编写一个自定义 MarshalJSON 函数，该函数运行良好。当我在包含需要自定义编组作为字段的父结构上调用 json.Marshal 时，它工作正常。

现在我需要将所有内容编组到 BSON 以进行一些 MongoDB 工作，并且我找不到任何有关如何编写自定义 BSON 编组的文档。谁能告诉我如何为我在下面演示的 BSON/mgo 做同样的事情？

currency.go （重要部分）

type Currency struct {
    value        decimal.Decimal //The actual value of the currency.
    currencyCode string          //The ISO currency code.
}

/*
MarshalJSON implements json.Marshaller.
*/
func (c Currency) MarshalJSON() ([]byte, error) {
    f, _ := c.Value().Float64()
    return json.Marshal(struct {
        Value        float64 `json:"value" bson:"value"`
        CurrencyCode string  `json:"currencyCode" bson:"currencyCode"`
    }{
        Value:        f,
        CurrencyCode: c.CurrencyCode(),
    })
}

/*
UnmarshalJSON implements json.Unmarshaller.
*/
func (c *Currency) UnmarshalJSON(b []byte) error {

    decoded := new(struct {
        Value        float64 `json:"value" bson:"value"`
        CurrencyCode string  `json:"currencyCode" bson:"currencyCode"`
    })

    jsonErr := json.Unmarshal(b, decoded)

    if jsonErr == nil {
        c.value = decimal.NewFromFloat(decoded.Value)
        c.currencyCode = decoded.CurrencyCode
        return nil
    } else {
        return jsonErr
    }
}

product.go （同样，只是相关部分）

type Product struct {
    Name  string
    Code  string
    Price currency.Currency
}

当我调用 json.Marshal(p) 其中 p 是一个产品时，它会产生我想要的输出，而不需要您创建一个结构的模式（不确定名称），该结构只是具有所有导出字段的克隆。

在我看来，使用我使用的内联方法极大地简化了 API，并且不会让你有额外的结构来使事情变得混乱。

正确答案

自定义 bson Marshalling/Unmarshalling 的工作方式几乎相同，您必须分别实现Getter和Setter接口

像这样的东西应该工作：

type Currency struct {
    value        decimal.Decimal //The actual value of the currency.
    currencyCode string          //The ISO currency code.
}

// GetBSON implements bson.Getter.
func (c Currency) GetBSON() (interface{}, error) {
    f := c.Value().Float64()
    return struct {
        Value        float64 `json:"value" bson:"value"`
        CurrencyCode string  `json:"currencyCode" bson:"currencyCode"`
    }{
        Value:        f,
        CurrencyCode: c.currencyCode,
    }, nil
}

// SetBSON implements bson.Setter.
func (c *Currency) SetBSON(raw bson.Raw) error {

    decoded := new(struct {
        Value        float64 `json:"value" bson:"value"`
        CurrencyCode string  `json:"currencyCode" bson:"currencyCode"`
    })

    bsonErr := raw.Unmarshal(decoded)

    if bsonErr == nil {
        c.value = decimal.NewFromFloat(decoded.Value)
        c.currencyCode = decoded.CurrencyCode
        return nil
    } else {
        return bsonErr
    }
}

今天关于《处理自定义 BSON 编组》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于golang的内容请关注golang学习网公众号！