Bagaimana untuk Mengabaikan Nilai Null Apabila Membongkar Dokumen MongoDB dalam Go?

Mengabaikan Nilai Null semasa Menyahmarshalling Dokumen MongoDB

Apabila menyahmarshall dokumen MongoDB ke dalam struct Go yang mengandungi medan rentetan yang tidak boleh dibatalkan, menemui null nilai dalam dokumen boleh membawa kepada ralat. Untuk menangani isu ini, adalah perlu untuk mencari cara untuk mengabaikan nilai nol ini semasa unmarshalling.

Menggunakan Penyahkod Tersuai

Satu pendekatan untuk mengendalikan nilai nol ialah dengan mencipta penyahkod tersuai untuk jenis rentetan. Penyahkod tersuai ini akan mengenali nilai nol dan mengendalikannya dengan menetapkan medan sepadan kepada rentetan kosong, dengan berkesan mengabaikan nilai nol:

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:
        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
}

Penyahkod tersuai ini kemudiannya boleh didaftarkan dengan bsoncodec.Registry dan digunakan untuk objek mongo.Client:

clientOpts := options.Client().
    ApplyURI("mongodb://localhost:27017/").
    SetRegistry(
        bson.NewRegistryBuilder().
            RegisterDecoder(reflect.TypeOf(""), nullawareStrDecoder{}).
            Build(),
    )
client, err := mongo.Connect(ctx, clientOpts)

Mencipta Null-Aware Type-Neutral Penyahkod

Untuk mengendalikan nilai nol untuk berbilang jenis, adalah mungkin untuk mencipta penyahkod neutral jenis tunggal yang menyemak nilai nol dan, jika ditemui, menetapkan medan yang sepadan dengan nilai sifarnya. jenis:

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
    }

    val.Set(d.zeroValue)
    return nil
}

Penyahkod ini boleh didaftarkan dengan bsoncodec.Registry untuk jenis tertentu atau untuk semua jenis:

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)})
}

Atas ialah kandungan terperinci Bagaimana untuk Mengabaikan Nilai Null Apabila Membongkar Dokumen MongoDB dalam Go?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!