protoc-go-inject-tag

janbar

fork这个仓库是因为需要修改pb.go中bytes类型为json.RawMessage,方便序列化和反序列化。

执行protoc --proto_path=. --go_out=paths=source_relative:. janbar.proto可以生成pb文件。

执行protoc-go-inject-tag.exe -i janbar.pb.go可以修改源码中的bytes为json.RawMessage。

注意需要按照规则填写gotags:"json.RawMessage"的gotags前缀的标签。如果有多个则最后一个生效。

protoc-go-inject-tag

Why?

Golang protobuf doesn't support custom tags to generated structs. This tool injects custom tags to generated protobuf files, which is commonly used for validating fields, omitting fields from JSON data, etc.

Install

• protobuf version 3

  For OS X:

  brew install protobuf
  

• go support for protobuf: go get -u github.com/golang/protobuf/{proto,protoc-gen-go}

• go install github.com/favadi/protoc-go-inject-tag or download the binaries from the releases page.

Usage

$ protoc-go-inject-tag -h
Usage of protoc-go-inject-tag:
  -XXX_skip string
        tags that should be skipped (applies 'tag:"-"') for unknown fields (deprecated since protoc-gen-go v1.4.0)
  -input string
        pattern to match input file(s)
  -verbose
        verbose logging
  -remove_tag_comment
        removes tag comments from the generated file(s)

Add a comment with the following syntax before fields, and these will be injected into the resulting .pb.go file. This can be specified above the field, or trailing the field.

// @gotags: custom_tag:"custom_value"

Example

// file: test.proto
syntax = "proto3";

package pb;
option go_package = "/pb";

message IP {
  // @gotags: valid:"ip"
  string Address = 1;

  // Or:
  string MAC = 2; // @gotags: validate:"omitempty"
}

Generate your .pb.go files with the protoc command as normal:

protoc --proto_path=. --go_out=paths=source_relative:. test.proto

Then run protoc-go-inject-tag against the generated files (e.g test.pb.go):

$ protoc-go-inject-tag -input=./test.pb.go
# or
$ protoc-go-inject-tag -input="*.pb.go"

The custom tags will be injected to test.pb.go:

type IP struct {
 // @gotags: valid:"ip"
 Address string `protobuf:"bytes,1,opt,name=Address,json=address" json:"Address,omitempty" valid:"ip"`
}

Remove gotag comments from generated output

Utilizing the -remove_tag_comment flag, you can remove the gotag comment that is normally annotated to the generated code. This allows more seamless support with libraries like swag/openapi generators that use code comments to generate openapi files.

Deprecated functionality

Skip XXX_* fields

To skip the tag for the generated XXX_* fields (unknown fields), use the -XXX_skip=yaml,xml flag. This is deprecated, as this functionality hasn't existed in protoc-gen-go since v1.4.x.

inject_tag keyword

Since v1.3.0, we recommend using @gotags: rather than @inject_tags:, as @gotags is more indicative of the language the comment is for. We don't plan on removing @inject_tags: support anytime soon, however we strongly recommend switching to @gotags.