proto_code_gen

使用protobuf的项目,一般用protoc生成*.pb.go文件,proto_code_gen可以分析生成的*.pb.go文件,做一些预处理

这样proto_code_gen就可以集成到项目的工作链中

proto_code_gen目前提供3个功能:

• 生成proto的Message结构的只读接口(应用场景:proto当作配置数据的格式)
• 生成proto的Message的消息号(应用场景:proto当作网络消息协议)
• 根据proto的Message的自定义的struct tag生成对应的代码(应用场景:给proto的Message增加一些自定义的设置,如struct tag)

应用场景1: 配置数据的只读接口

某些应用场景,会使用protobuf的结构来当作配置数据的格式,proto_code_gen提供了一种生成protobuf只读接口的功能,类似c++中的const.

如examples里的examples/reader_gen.go.template模板对应生成代码examples/gen/cfg_reader_gen.go

如https://github.com/fish-tennis/gserver/tree/main/gen目录下的代码就是使用proto_code_gen生成的

应用场景2: 自动生成网络消息协议号

某些应用场景,会使用消息号+protobuf来做网络协议,proto_code_gen可以根据proto的message名自动生成网络消息协议号,并解决消息号冲突的问题.

应用场景3:生成模板代码

项目中,我们经常希望能给proto生成message增加一些自定义的设置,如struct tag.

增加自定义的struct tag,可以用protoc-go-inject-tag

但是protoc-go-inject-tag只能给message的字段加struct tag,因为golang并没有给struct结构体提供tag接口.

proto_code_gen提供了一种给message增加类似struct tag的方式

Step1: 在proto文件中,使用自定义tag,示例参考examples/proto/cfg.proto

// file: examples/proto/cfg.proto
syntax = "proto3";

package pb;
option go_package = "/pb";

// 测试message's struct tag
// @StructTagOfExample
message Example {
  int32 Field1 = 1;
  string Field2 = 2;
}

Step2: protoc生成*.pb.go文件,如examples/pb/cfg.pb.go

Step3: 配置代码模板,示例参考examples/message_gen.go.template,examples/reader_gen.go.template

模板使用go自带的text/template

Step4: 运行protoc_code_gen -input=/examples/pb/*.pb.go -config=./proto_code_gen.yaml

会根据模板生成对应的代码,如examples/gen/example_gen.go,examples/gen/cfg_reader_gen.go

// file: examples/gen/example_gen.go
package gen
import (
  . "github.com/fish-tennis/proto_code_gen/examples/pb"
)
// 测试message's struct tag
func ExampleToString(m *Example) string {
  // just a test code
  return m.String()
}

使用proto_code_gen

获取

go get github.com/fish-tennis/proto_code_gen

运行

protoc_code_gen -input=/dir/*.pb.go -config=./proto_code_gen.yaml

参考

protoc-go-inject-tag

text/template