使用 mongo-driver 自定义 BSON 编组和解组

Question

我有一个像下面这样的结构字段。我还将相同结构的原始 protobuf 存储在 db 中。现在每次都将数据获取或保存到 mongo。ReallyBigRaw当我想保存到数据库时，我必须从 proto更新，而当我必须解组ReallyBigRaw以ReallyBigObj给出响应时，我必须解组。有没有办法我可以实现一些接口或提供一些回调函数，以便 mongo 驱动程序在保存或从 DB 获取数据之前自动执行此操作。另外，我没有使用官方的 golang mongo 驱动程序mgo，我已经阅读了一些可以在mgogolang 库中完成的答案。import (    "github.com/golang/protobuf/jsonpb"    "go.mongodb.org/mongo-driver/bson"    "go.mongodb.org/mongo-driver/bson/primitive"    proto "github.com/dinesh/api/go")type ReallyBig struct {    ID           string                 `bson:"_id,omitempty"`    DraftID      string                 `bson:"draft_id,omitempty"`// Marshaled ReallyBigObj proto to map[string]interface{} stored in DB    ReallyBigRaw map[string]interface{} `bson:"raw,omitempty"`    ReallyBigObj *proto.ReallyBig       `bson:"-"`    CreatedAt    primitive.DateTime     `bson:"created_at,omitempty"`    UpdatedAt    primitive.DateTime     `bson:"updated_at,omitempty"`}func (r *ReallyBig) GetProto() (*proto.ReallyBig, error) {    if r.ReallyBigObj != nil {        return r.ReallyBigObj, nil    }    Obj, err := getProto(r.ReallyBigRaw)    if err != nil {        return nil, err    }    r.ReallyBigObj = Obj    return r.ReallyBigObj, nil}func getRaw(r *proto.ReallyBig) (map[string]interface{}, error) {    m := jsonpb.Marshaler{}    b := bytes.NewBuffer([]byte{})    // marshals proto to json format    err := m.Marshal(b, r)    if err != nil {        return nil, err    }    var raw map[string]interface{}    // unmarshal the raw data to an interface    err = json.Unmarshal(b.Bytes(), &raw)    if err != nil {        return nil, err    }    return raw, nil}func getProto(raw map[string]interface{}) (*proto.ReallyBig, error) {    b, err := json.Marshal(raw)    if err != nil {        return nil, err    }    u := jsonpb.Unmarshaler{}    var reallyBigProto proto.ReallyBig    err = u.Unmarshal(bytes.NewReader(b), &recipeProto)    if err != nil {        return nil, err    }    return &reallyBigProto, nil}

Answer 1

我实现了MarshalerandUnmarshaler接口。由于 mongo 驱动程序调用MarshalBSON并且UnmarshalBSON如果类型实现Marshaler并且Unmarshaler我们也最终陷入无限循环。为了避免这种情况，我们创建了type. 别名Golang仅继承字段而不是方法，因此我们最终调用正常bson.Marshal和bson.Unmarshal

func (r *ReallyBig) MarshalBSON() ([]byte, error) {

    type ReallyBigAlias ReallyBig

    reallyBigRaw, err := getRaw(r.ReallyBigObj)

    if err != nil {

        return nil, err

    }

    r.ReallyBigRaw = reallyBigRaw

    return bson.Marshal((*ReallyBigAlias)(r))

}

func (r *ReallyBig) UnmarshalBSON(data []byte) error {

    type ReallyBigAlias ReallyBig

    err := bson.Unmarshal(data, (*ReallyBigAlias)(r))

    if err != nil {

        return err

    }

    reallyBigProto, err := getProto(r.ReallyBigRaw)

    if err != nil {

        return err

    }

    r.ReallyBigObj = reallyBigProto

    return nil

}