MongoDB 处理自定义 BSON Marshaling

在本文中，我们将介绍如何在 MongoDB 中处理自定义 BSON 编组。BSON 是 MongoDB 中使用的一种二进制表示格式，用于存储和交换数据。

阅读更多：MongoDB 教程

什么是 BSON？

BSON（Binary JSON）是一种可序列化数据的二进制表示形式。它类似于 JSON，但有更多的数据类型和更高效的存储格式。BSON 主要用于 MongoDB 中的文档存储和网络传输。

BSON 编组

当我们将数据存储到 MongoDB 中时，数据会被自动编组为 BSON 格式。MongoDB 使用 BSON 格式来实现数据持久化和查询。

在默认情况下，MongoDB 提供了将常见数据类型（如字符串、整数、数组等）转换为 BSON 的编组规则。但是，对于自定义类型，我们需要显式地实现 BSON 编组。

实现自定义 BSON 编组

要实现自定义 BSON 编组，我们需要创建一个实现 BSON.Marshaler 接口的结构体。该接口具有一个方法 MarshalBSONValue()，用于将结构体编组为 BSON。

下面是一个示例，演示如何实现一个自定义类型 Person，并将其编组为 BSON。

type Person struct {
    Name string
    Age  int
}

func (p Person) MarshalBSONValue() (bsontype.Type, []byte, error) {
    doc := bsonx.Doc{
        {"name", bsonx.String(p.Name)},
        {"age", bsonx.Int32(int32(p.Age))},
    }
    return bsonx.Document(doc).MarshalBSON()
}

在上面的示例中，我们创建了一个名为 Person 的结构体，并定义了 MarshalBSONValue() 方法。该方法将 Person 的字段编组为一个包含 name 和 age 键的 BSON 文档。

使用自定义类型进行插入操作时，MongoDB 将调用 MarshalBSONValue() 方法以将结构体编组为 BSON。

自定义 BSON 编组的示例

让我们看一个完整的示例，演示如何使用自定义 BSON 编组。

package main

import (
    "context"
    "fmt"
    "log"

    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/bsoncodec"
    "go.mongodb.org/mongo-driver/bson/primitive"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

type Person struct {
    ID   primitive.ObjectID `bson:"_id,omitempty"`
    Name string             `bson:"name"`
    Age  int                `bson:"age"`
}

func (p Person) MarshalBSONValue() (bsontype.Type, []byte, error) {
    doc := bsonx.Doc{
        {"_id", bsonx.ObjectID(p.ID)},
        {"name", bsonx.String(p.Name)},
        {"age", bsonx.Int32(int32(p.Age))},
    }
    return bsonx.Document(doc).MarshalBSON()
}

const (
    uri      = "mongodb://localhost:27017"
    database = "mydb"
    collection = "persons"
)

func main() {
    clientOptions := options.Client().ApplyURI(uri)
    client, err := mongo.Connect(context.Background(), clientOptions)
    if err != nil {
        log.Fatal(err)
    }

    collection := client.Database(database).Collection(collection)

    p := Person{
        ID:   primitive.NewObjectID(),
        Name: "Alice",
        Age:  30,
    }

    _, err = collection.InsertOne(context.Background(), p)
    if err != nil {
        log.Fatal(err)
    }

    filter := bson.M{"name": "Alice"}
    result := Person{}
    err = collection.FindOne(context.Background(), filter).Decode(&result)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println("Inserted and retrieved document:", result)
}

在上面的示例中，我们创建了一个 Person 结构体，实现了 MarshalBSONValue() 方法。然后，我们使用自定义类型将数据插入 MongoDB，并进行检索和解码。