起因

  • 在一个老项目里面直接用的 protobuf 生成的结构体与数据库交互
  • 在 protoc 2 生成的 protobuf 没有 XXX 这些东西生成故能正常运行
  • 因为版本升级到 protoc 3 后编译的 protobuf 就不能正常运行了,会出现数据库找不到 XXX_ 这些字段的情况

解决方案

  • 把每张表单独生成 struct 放在 models 层里面,这样的话老服务改动太大我肯定是不能接受的
  • 因为我们使用的 orm ,在xorm 中如果 struct 某个字段不需要映射的话就可以用 xorm:"-”
  • gorm 也可以这样操作 gorm:"-”
  • 那我们就可以在 XXX_ 这些字段后面的 tag 中添加忽略,就可以实现我们的目表

开始搞事情

获取代码

go get github.com/golang/protobuf/protoc-gen-go

修改代码

  • 在路径 github.com/golang/protobuf/protoc-gen-go/generator/generator.go 下找到 generateInternalStructFields 方法
  • 在XXX_ 后添加 xorm:"-" or gorm:"-"
// generateInternalStructFields just adds the XXX_<something> fields to the message struct.
func (g *Generator) generateInternalStructFields(mc *msgCtx, topLevelFields []topLevelField) {
	g.P("XXX_NoUnkeyedLiteral\tstruct{} `json:\"-\" xorm:\"-\"`") // prevent unkeyed struct literals
	if len(mc.message.ExtensionRange) > 0 {
		messageset := ""
		if opts := mc.message.Options; opts != nil && opts.GetMessageSetWireFormat() {
			messageset = "protobuf_messageset:\"1\" "
		}
		g.P(g.Pkg["proto"], ".XXX_InternalExtensions `", messageset, "json:\"-\" xorm:\"-\"`")
	}
	g.P("XXX_unrecognized\t[]byte `json:\"-\" xorm:\"-\"`")
	g.P("XXX_sizecache\tint32 `json:\"-\" xorm:\"-\"`")

}

编译代码得到 protoc-gen-go 并放到 /go/bin/ 下

  • 编译 proto
protoc  --go_out=. *.proto

成果

type BoolValue struct {
	Value                bool     `protobuf:"varint,1,opt,name=value,proto3" json:"value,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-" xorm:"-"`
	XXX_unrecognized     []byte   `json:"-" xorm:"-"`
	XXX_sizecache        int32    `json:"-" xorm:"-"`
}

结语

  • 通过这样的方法我们就可以实现我们想要的样子了
  • 不过还是建议一个表对应一个 struct 放在 models 里面
  • 欢迎添加QQ一起讨论

联系 QQ: 3355168235