Heim > Backend-Entwicklung > Golang > Wie implementiert man benutzerdefiniertes BSON-Marshalling in Go?

Wie implementiert man benutzerdefiniertes BSON-Marshalling in Go?

Linda Hamilton
Freigeben: 2024-11-28 03:02:14
Original
341 Leute haben es durchsucht

How to Implement Custom BSON Marshaling 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
}
Nach dem Login kopieren

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")
Nach dem Login kopieren

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!

Quelle:php.cn
Erklärung dieser Website
Der Inhalt dieses Artikels wird freiwillig von Internetnutzern beigesteuert und das Urheberrecht liegt beim ursprünglichen Autor. Diese Website übernimmt keine entsprechende rechtliche Verantwortung. Wenn Sie Inhalte finden, bei denen der Verdacht eines Plagiats oder einer Rechtsverletzung besteht, wenden Sie sich bitte an admin@php.cn
Neueste Artikel des Autors
Beliebte Tutorials
Mehr>
Neueste Downloads
Mehr>
Web-Effekte
Quellcode der Website
Website-Materialien
Frontend-Vorlage