mongodb - 从 bson 结构转换 protoc 生成的结构的最佳方法是什么？

Question

我正在用 Golang 编写一个 RESTful API，它也有一个 gRPC api。 API 连接到 MongoDB 数据库，并使用结构来映射实体。我还有一个 .proto 定义，它与我用于 MongoDB 的结构相匹配。

我只是想知道是否有一种方法可以共享或重新使用 .proto 定义的代码来进行 MongoDB 调用。我注意到 strucs protoc 生成的每个字段都有 json 标签，但显然没有 bson 标签等。

我有类似的东西......

// Menu -
type Menu struct {
    ID          bson.ObjectId      `json:"id" bson"_id"`
    Name        string             `json:"name" bson:"name"`
    Description string             `json:"description" bson:"description"`
    Mixers      []mixers.Mixer     `json:"mixers" bson:"mixers"`
    Sections    []sections.Section `json:"sections" bson:"sections"`
}

但是我也有 protoc 生成的代码...

type Menu struct {
    Id          string     `protobuf:"bytes,1,opt,name=id" json:"id,omitempty"`
    Name        string     `protobuf:"bytes,2,opt,name=name" json:"name,omitempty"`
    Description string     `protobuf:"bytes,3,opt,name=description" json:"description,omitempty"`
    Mixers      []*Mixer   `protobuf:"bytes,4,rep,name=mixers" json:"mixers,omitempty"`
    Sections    []*Section `protobuf:"bytes,5,rep,name=sections" json:"sections,omitempty"`
}

目前我必须根据我正在做的事情在两个结构之间进行转换。这很乏味，我可能会受到相当大的性能影响。那么有没有更好的方法在两者之间进行转换，或者将其中一个重新用于两项任务？

Answer 1

遇到过同样的问题，有几种解决方法。它们分为两种通用方法:

1. 使用相同的数据类型
2. 使用两种不同的结构类型并在它们之间进行映射

如果要使用相同的数据类型，则必须修改代码生成

您可以使用 gogoprotobuf 之类的内容它具有添加标签的扩展名。这应该在您的结构中为您提供 bson 标记。

您还可以使用正则表达式或涉及 go 抽象语法树的更复杂的方法对生成的文件进行后处理。

如果您选择在它们之间进行映射:

1. 使用 反射。您可以编写一个包含两个结构的包，并尝试从一个结构中获取值并将其应用于另一个结构。您必须处理边缘情况(轻微的命名差异、哪些类型是等效的等)，但如果出现边缘情况，您将能够更好地控制它们。

2. 使用 JSON 作为中介。只要生成的 json 标签匹配，这将是一个快速的编码练习，并且如果这不是在代码中的紧密循环中，序列化和反序列化的性能损失可能是可以接受的。

3. 手写或代码生成映射功能。根据你有多少结构，你可以写出一堆在两者之间转换的函数。

在我的工作场所，我们最终做了所有这些工作:派生 protoc 生成器来做一些自定义标签，一个基于反射的结构覆盖包，用于在任意结构之间进行映射，以及一些对性能更敏感的手写结构或自动化程度较低的映射。