使用 MongoDB 解码将字符串从 MongoDB 转换为 int

Question

当我尝试将我的集合中的字符串类型的属性解码为我的结构中的 int 类型的属性时，我在使用 Go 的 mongodb 驱动程序时遇到了一些问题。我有一个集合，该集合的属性应该是 int 类型，该属性被编辑单个文档的人转换为 string 类型。由于所有文档都具有该属性作为 int 除了那个，我收到一个错误“无法将字符串解码为整数类型”。我想出了一种方法来做到这一点，但这不是“应该”做的事情。我在 Go 的 mongodb 驱动程序中修改了 default_value_decoders.go 中的 IntDecodeValue 函数。下面是我添加的。

    case bsontype.String:

    s, err := vr.ReadString()

    if err != nil {

        return err

    }

    i32, err := strconv.Atoi(s)

    if err != nil {

        return err

    }

    i64 = int64(i32)

我知道当我更新驱动程序时这将被覆盖，但我一直在努力思考如何处理这种情况。我知道最好的解决方案是不允许直接编辑文档，但要考虑这种情况。

Answer 1

您可以通过自定义 BSON 解码器执行此操作。文档中的一个示例是https://pkg.go.dev/go.mongodb.org/mongo-driver/bson/bsoncodec?tab=doc#example-Registry-CustomDecoder。对于您的特定用例，以下应该有效：func intDecodeValue(dc bsoncodec.DecodeContext, vr bsonrw.ValueReader, val reflect.Value) {// Same logic as DefaultValueDecoders.IntDecodeValue// In the switch statement for vr.Type, you can add a case for bsontype.String}要注册此解码器以便在执行 CRUD 操作时使用它，您可以使用SetRegistry客户端选项：decoder := bsoncodec.ValueDecoderFunc(intDecodeValue)registry := bson.NewRegistryBuilder().RegisterDefaultDecoder(reflect.Int, decoder).Build()clientOpts := options.Client().SetRegistry(registry)client, err := mongo.Connect(context.TODO(), clientOpts)请注意，由于 Go 区分不同的整数类型（例如 int8/int16/int32/int64），因此您需要调用RegisterDefaultDecoder为您可能在结构中看到的每种整数类型注册自定义解码器，类似于在RegisterDefaultDecoders中的函数default_value_decoders.go。

Answer 2

这是在处理字符串和 int 字段的结构中将该字符串字段解码为 int 的可能解决方案。我在这个例子中使用了以下结构：type Obj struct {&nbsp; &nbsp; Field1&nbsp; &nbsp;string `bson:"field1"`&nbsp; &nbsp; IntField int&nbsp; &nbsp; `bson:"intField"`}并插入以下文档（注意第二个文档的字符串字段为"intField": "2"）：db.s2i.insertOne({"field1": "value", "intField": 3})db.s2i.insertOne({"field1": "value2", "intField": "2"})使用聚合框架中的$toInt运算符，如下所示：pipeline := []bson.M{bson.M{"$match": bson.M{}}, bson.M{"$project": bson.M{"field1": 1, "intField": bson.M{"$toInt": "$intField"}}}}cur, err := client.Database("stack").Collection("s2i").Aggregate(context.TODO(), pipeline)for cur.Next(context.TODO()) {&nbsp; &nbsp; var res *Obj&nbsp; &nbsp; err = cur.Decode(&res)&nbsp; &nbsp; fmt.Println(res, err)&nbsp; &nbsp; fmt.Println("Value of res.IntField:")&nbsp; &nbsp; fmt.Println(res.IntField)&nbsp; &nbsp; fmt.Println("Type of res.IntField:")&nbsp; &nbsp; fmt.Println(reflect.TypeOf(res.IntField))}返回以下文档，其中“2”解码为 int：&{value 3} <nil>Value of res.IntField:3Type of res.IntField:int&{value2 2} <nil>Value of res.IntField:2Type of res.IntField:int