Go - 如何编写 ProtoBuf 插件 (一) ?

2021-12-06 15:33:40 浏览数 (1)

文章目录:

  • 前言
  • 自定义选项
  • 需求场景
  • 推荐阅读

前言

我们要知道 proto3proto2 的语法,并不是完全兼容的。

具体可查阅官方文档:

  • Overview[1]
  • Language Guide (proto2)[2]
  • Language Guide (proto3)[3]

如果上述链接无法打开,可以访问这个文档:Overview - 语雀[4]

自定义选项

proto3 中,常见的实现插件的方式是使用 自定义选项,也就是 extend 标签,其中支持的 extend Options 有:

  • MethodOptions
  • ServiceOptions
  • EnumOptions
  • EnumValueOptions
  • MessageOptions
  • FieldOptions
  • FileOptions
  • OneofOptions
  • ExtensionRangeOptions

具体写法可参考:

代码语言:javascript复制
import "google/protobuf/descriptor.proto";

extend google.protobuf.MessageOptions {
  optional string my_option = 51234;
}

message MyMessage {
  option (my_option) = "Hello world!";
}

需求场景

假设,我们的需求场景是这样的:

我们有很多的拦截器,其中不同的 service 可能会使用一个或多个拦截器,不同的 method 也可能会使用一个或多个拦截器,在 helloworld.proto

  • service Greeter{} 支持登录令牌验证
  • rpc SayHello1() 支持 IP 白名单限制和记录日志
  • rpc SayHello2() 支持禁止记录日志
代码语言:javascript复制
// helloworld.proto

service Greeter {
  rpc SayHello1 (HelloRequest) returns (HelloReply) {}
  rpc SayHello2 (HelloRequest) returns (HelloReply) {}
}

message HelloRequest {
  string name = 1;
}

message HelloReply {
  string message = 1;
}

我们需要在 proto 文件中,定义出 service 使用了哪些拦截器?定义出 method 使用了哪些拦截器?这样 proto 文件就会更加语义化,更加清晰明确,当大家看到定义的文件时,对使用的拦截器一目了然。

如何实现这个功能?

这时,我们就需要用到 MethodOptionsServiceOptions 选项,通过名字大概也能猜到 MethodOptions 是定义方法选项的,ServiceOptions 是定义服务选项的。

代码语言:javascript复制
extend google.protobuf.MethodOptions {
  ...
}

extend google.protobuf.ServiceOptions {
  ...
}

大家有实现的思路吗?欢迎留言评论 ~

参考资料

[1]Overview: https://developers.google.com/protocol-buffers/docs/overview

[2]Language Guide (proto2): https://developers.google.com/protocol-buffers/docs/proto

[3]Language Guide (proto3): https://developers.google.com/protocol-buffers/docs/proto3

[4]Overview - 语雀: https://www.yuque.com/xinliangnote/blog/mumt7v

0 人点赞