Wie implementiert man benutzerdefiniertes BSON-Marshalling in Go?

Benutzerdefiniertes BSON-Marshalling

Benutzerdefiniertes BSON-Marshalling ähnelt dem benutzerdefinierten JSON-Marshalling, Sie müssen jedoch bson.Getter und bson implementieren Stattdessen .Setter-Schnittstellen. Mit diesen Schnittstellen können Sie definieren, wie Ihr benutzerdefinierter Typ in und aus BSON-Daten konvertiert werden soll.

Angenommen, Sie haben einen Währungstyp, der benutzerdefiniertes Marshalling erfordert:

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
}

Mit Mit dieser Implementierung können Sie jetzt Ihren benutzerdefinierten Währungstyp beim MGO BSON-Encoder registrieren:

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")

Jetzt, wenn Sie ein Dokument mit einem Währungsfeld speichern MongoDB, der MGO-Encoder, verwendet automatisch Ihre benutzerdefinierte Marshalling-Funktion, um die Währung in BSON umzuwandeln.

Beachten Sie, dass dies nur ein Beispiel dafür ist, wie Sie benutzerdefiniertes Marshalling für BSON-Typen hinzufügen können. Die spezifische Implementierung hängt von den Anforderungen Ihrer speziellen Anwendung ab.

Das obige ist der detaillierte Inhalt vonWie implementiert man benutzerdefiniertes BSON-Marshalling in Go?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!