问题是目前的bson编解码器不支持将string编码/解码成/从null。
解决这个问题的一种方法是为string 类型创建一个自定义解码器,我们在其中处理null 值:我们只使用空字符串(更重要的是不报告错误)。
自定义解码器由bsoncodec.ValueDecoder 类型描述。他们可以在bsoncodec.Registry 注册,例如使用bsoncodec.RegistryBuilder。
注册可以在多个级别设置/应用,甚至可以在整个mongo.Client、mongo.Database 或mongo.Collection 上设置/应用,在获取它们时,作为其选项的一部分,例如options.ClientOptions.SetRegistry().
首先让我们看看我们如何为string 做到这一点,接下来我们将看看如何改进/将解决方案推广到任何类型。
1。处理null 字符串
首先,让我们创建一个自定义字符串解码器,它可以将null 转换为(n 个空)字符串:
import (
"go.mongodb.org/mongo-driver/bson/bsoncodec"
"go.mongodb.org/mongo-driver/bson/bsonrw"
"go.mongodb.org/mongo-driver/bson/bsontype"
)
type nullawareStrDecoder struct{}
func (nullawareStrDecoder) DecodeValue(dctx bsoncodec.DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
if !val.CanSet() || val.Kind() != reflect.String {
return errors.New("bad type or not settable")
}
var str string
var err error
switch vr.Type() {
case bsontype.String:
if str, err = vr.ReadString(); err != nil {
return err
}
case bsontype.Null: // THIS IS THE MISSING PIECE TO HANDLE NULL!
if err = vr.ReadNull(); err != nil {
return err
}
default:
return fmt.Errorf("cannot decode %v into a string type", vr.Type())
}
val.SetString(str)
return nil
}
好的,现在让我们看看如何将这个自定义字符串解码器用于mongo.Client:
clientOpts := options.Client().
ApplyURI("mongodb://localhost:27017/").
SetRegistry(
bson.NewRegistryBuilder().
RegisterDecoder(reflect.TypeOf(""), nullawareStrDecoder{}).
Build(),
)
client, err := mongo.Connect(ctx, clientOpts)
从现在开始,使用这个client,每当你将结果解码为string值时,这个注册的nullawareStrDecoder解码器将被调用来处理转换,它接受bson null值并设置Go空字符串"".
但我们可以做得更好...继续阅读...
2。处理任何类型的null 值:“类型中性”空值感知解码器
一种方法是创建一个单独的自定义解码器,并为我们希望处理的每种类型注册它。这似乎是很多工作。
我们可以(并且应该)做的是创建一个单独的“类型中性”自定义解码器,它只处理 nulls,如果 BSON 值不是 null,则应该调用默认解码器来处理非null 值。
这非常简单:
type nullawareDecoder struct {
defDecoder bsoncodec.ValueDecoder
zeroValue reflect.Value
}
func (d *nullawareDecoder) DecodeValue(dctx bsoncodec.DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
if vr.Type() != bsontype.Null {
return d.defDecoder.DecodeValue(dctx, vr, val)
}
if !val.CanSet() {
return errors.New("value not settable")
}
if err := vr.ReadNull(); err != nil {
return err
}
// Set the zero value of val's type:
val.Set(d.zeroValue)
return nil
}
我们只需要弄清楚nullawareDecoder.defDecoder 的用途。为此,我们可以使用默认注册表:bson.DefaultRegistry,我们可以查找各个类型的默认解码器。很酷。
所以我们现在要做的是为我们想要处理nulls 的所有类型注册一个nullawareDecoder 的值。这并不难。我们只列出我们想要的类型(或这些类型的值),我们可以通过一个简单的循环来处理所有问题:
customValues := []interface{}{
"", // string
int(0), // int
int32(0), // int32
}
rb := bson.NewRegistryBuilder()
for _, v := range customValues {
t := reflect.TypeOf(v)
defDecoder, err := bson.DefaultRegistry.LookupDecoder(t)
if err != nil {
panic(err)
}
rb.RegisterDecoder(t, &nullawareDecoder{defDecoder, reflect.Zero(t)})
}
clientOpts := options.Client().
ApplyURI("mongodb://localhost:27017/").
SetRegistry(rb.Build())
client, err := mongo.Connect(ctx, clientOpts)
在上面的示例中,我为 string、int 和 int32 注册了可识别空值的解码器,但您可以根据自己的喜好扩展此列表,只需将所需类型的值添加到上面的 customValues 切片中.