Go MongoDB项目自定义_id类型解析报错，如何实现包隔离？

问题核心在于MongoDB的bson解码器仅原生支持primitive.ObjectID，自定义的Id类型（无论是类型别名还是嵌入结构体）对解码器来说是全新类型，无法自动完成编解码。以下是两种可行的解决方案，均能保持models包与原生驱动的隔离：

方案一：让自定义Id类型实现bson编解码接口 #

针对type Id primitive.ObjectID的定义，实现bson.Marshaler和bson.Unmarshaler接口，手动完成与原生类型的转换：

package my_mongo_driver

import (
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/primitive"
)

type Id primitive.ObjectID

// MarshalBSON 将自定义Id序列化为bson格式
func (id Id) MarshalBSON() ([]byte, error) {
    return bson.Marshal(primitive.ObjectID(id))
}

// UnmarshalBSON 将bson数据反序列化为自定义Id
func (id *Id) UnmarshalBSON(data []byte) error {
    var objID primitive.ObjectID
    if err := bson.Unmarshal(data, &objID); err != nil {
        return err
    }
    *id = Id(objID)
    return nil
}

实现这两个接口后，bson解码器就能正确识别并处理自定义Id类型，models包依然只需依赖my_mongo_driver，无需引入原生驱动。

方案二：注册自定义编解码器到Mongo客户端 #

如果需要批量处理多个自定义类型，可以在初始化Mongo客户端时，向bson注册表注册自定义编解码器，让驱动知晓类型转换规则：

package my_mongo_driver

import (
    "context"
    "reflect"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/bsoncodec"
    "go.mongodb.org/mongo-driver/bson/bsonrw"
    "go.mongodb.org/mongo-driver/bson/primitive"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

func NewMongoClient(uri string) (*mongo.Client, error) {
    // 构建自定义bson注册表
    registryBuilder := bson.NewRegistryBuilder()

    // 注册Id类型的编码器：将自定义Id转为primitive.ObjectID
    registryBuilder.RegisterTypeEncoder(
        reflect.TypeOf(Id{}),
        bsoncodec.ValueEncoderFunc(func(ec bsoncodec.EncodeContext, vw bsonrw.ValueWriter, val reflect.Value) error {
            objID := primitive.ObjectID(val.Interface().(Id))
            return bsoncodec.DefaultValueEncoders.ObjectIDEncodeValue(ec, vw, reflect.ValueOf(objID))
        }),
    )

    // 注册Id类型的解码器：将primitive.ObjectID转为自定义Id
    registryBuilder.RegisterTypeDecoder(
        reflect.TypeOf(Id{}),
        bsoncodec.ValueDecoderFunc(func(dc bsoncodec.DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
            var objID primitive.ObjectID
            err := bsoncodec.DefaultValueDecoders.ObjectIDDecodeValue(dc, vr, reflect.ValueOf(&objID).Elem())
            if err != nil {
                return err
            }
            val.Set(reflect.ValueOf(Id(objID)))
            return nil
        }),
    )

    // 使用自定义注册表创建客户端
    clientOpts := options.Client().ApplyURI(uri).SetRegistry(registryBuilder.Build())
    return mongo.Connect(context.Background(), clientOpts)
}

这种方式适合有多个自定义类型需要适配的场景，所有编解码逻辑集中在my_mongo_driver包中，进一步强化了隔离性。

为什么之前的两种定义方式失败？ #

• type Id primitive.ObjectID：Go的类型系统将其视为全新类型，而非primitive.ObjectID的别名，bson解码器不会自动关联两者。
• type Id struct {primitive.ObjectID}：嵌入结构体同样创建了新类型，解码器无法识别其内部的原生ObjectID字段，除非明确告知转换规则。

内容的提问来源于stack exchange，提问作者raphael.oester