用于插入或获取 mongo-driver 文档的自定义编码器/解码器

Question

我已经阅读了这个friendly article，关于使用官方 go mongo 驱动程序对自定义对象进行编码和解码。

有一个很好的例子，如何将它们编组为扩展的 json 格式 (bson.MarshalExtJSONWithRegistry)。但我想知道如何使用InserOne() 将此文档放入集合中（然后从中获取）。看看这个伪代码：

// myReg - variable created according to linked article in question.

// WithRegistry do **not** exist in mongo-driver lib is part of pseudocode
mongoCollection := client.Database("db").Collection("coll").WithRegistry(myReg)
// Now InserOne() honor myReg (type *bsoncodec.Registry) when serialize `val` and puting it into mongodb
mongoCollection.InsertOne(context.TODO(), val)

我浏览了 API 文档，发现有 Marshaler 和 Unmarshaler 接口，但是通过注册表方式，我可以在不同的集合上以不同的方式（反）序列化相同的类型（例如，当从旧格式迁移到新格式）。

所以问题是如何将*bsoncodec.Registry 与集合函数（如InserOne、UpdateOne、FindOne 等）一起使用，如果不是实现我的目标的最惯用方式（自定义（反）序列化)。

Answer 1

Database.Collection() 方法具有“可选”options.CollectionOptions 参数，该参数具有设置bsoncodec.Registry 的选项。如果您使用配置了注册表的选项获取您的集合，则该注册表将用于对该集合执行的所有操作。

像这样使用它：

opts := options.Collection().SetRegistry(myReg)
c := client.Database("db").Collection("coll", opts)

引用我的相关回答：How to ignore nulls while unmarshalling a MongoDB document?

注册可以在多个级别上设置/应用，甚至可以设置到整个mongo.Client、mongo.Database 或只是mongo.Collection，在获取它们时，作为它们选项的一部分，例如options.ClientOptions.SetRegistry().

因此，当您不从旧格式迁移到新格式时，您可以将注册表设置为“客户端”级别并“完成”。只要驱动程序处理您注册的自定义类型的值，就会应用您的注册表和自定义编码器/解码器。