设为首页 加入收藏

TOP

Protobuf Options 自定义插件使用(一)
2023-07-23 13:33:27 】 浏览:53
Tags:Protobuf Options

ProtoOptions

简介

使用protobuf的options,我们可以添加我们自己的option,然后根据option生成代码添对proto中的参数进行处理,例如添加message的校验。

extension

能够对已经存在的消息进行扩展,这种扩展只是动态增加字段,而没有增加接口。
例子:扩展protobuf的FieldOptions

message MyFileOptions {
  optional bool ok = 1;
}

// 扩展protobuf的MessageOptions
extend google.protobuf.FileOptions {
  optional MyFileOptions my_file_option = 51234; // 取一个自定义不重复值
}

protobuf options

  • google.protobuf.FileOptions
  • google.protobuf.ServiceOptions
  • google.protobuf.MethodOptions
  • google.protobuf.MessageOptions
  • google.protobuf.FieldOptions
  • google.protobuf.EnumOptions
  • google.protobuf.EnumValueOptions
  • google.protobuf.OneofOptions
  • google.protobuf.ExtensionRangeOptions

实现自定义option

扩展实现

mkdir protoc-gen-my-option
cd protoc-gen-my-option

实现一个FileOptions

mkdir proto

# 在proto中实现自己的option,可以将EOF的内容拷贝到proto/my_option.proto中
echo > proto/my_option.proto  <<EOF
syntax = "proto3";

package my_option;

option go_package="protoc-gen-my-option/proto";

import "google/protobuf/descriptor.proto";

message MyFileOptions {
  optional bool ok = 1;
}

// 扩展protobuf的MessageOptions
extend google.protobuf.FileOptions {
  optional MyFileOptions my_file_option = 51234; // 取一个自定义不重复值
}
EOF 

生成后proto/my_option.proto中的内容

syntax = "proto3";

package my_option;

option go_package="protoc-gen-my-option/proto";

import "google/protobuf/descriptor.proto";

message MyFileOptions {
  optional bool ok = 1;
}

// 扩展protobuf的MessageOptions
extend google.protobuf.FileOptions {
  optional MyFileOptions my_file_option = 51234; // 取一个自定义不重复值
}

生成扩展的pb文件

protoc --go_out=../ proto/my_option.proto

解析扩展插件

创建工程文件目录

扩展插件的开头必须以protoc-gen开头,在创建的protoc-gen-my-option目录下编写插件,并初始化项目。

go mod init protoc-gen-my-option

解析demo

package main

import (
	"google.golang.org/protobuf/compiler/protogen"
	"google.golang.org/protobuf/proto"
	"log"
	my_proto "protoc-gen-my-option/proto"
)

func getFileOptions(f *protogen.File) *my_proto.MyFileOptions {
	log.Println("getFileOptions")
	// 解析my_options
	if myOptions, ok := proto.GetExtension(f.Proto.Options, my_proto.E_MyFileOption).(*my_proto.MyFileOptions); ok {
		log.Println("my_options:", myOptions)
		return myOptions
	} else {
		log.Println("get my_options failed.")
	}

	return nil
}

func main() {
	protogen.Options{}.Run(func(plugin *protogen.Plugin) error {
		log.Println("protogen plugin")
		// protc时会把proto文件传入,解析所有文件
		for _, f := range plugin.Files {
			// 只处理我们处理的文件
			if !f.Generate {
				continue
			}
			// 处理getFiledOptions
			getFileOptions(f)
		}
		return nil
	})
}

安装插件

go install .

使用MyFileOptions

mkdir protoc-gen-my-option/test_proto

# 使用my-option
echo > test_proto/test_proto.proto  <<EOF
syntax = "proto3";

package my_option;

option go_package="protoc-gen-my-option/test_proto";

import "my_option.proto";

// 使用FiledOptions

option (my_option.my_file_option).ok = true;
EOF 

生成的test_proto/test_proto.proto内容如下:

syntax = "proto3";

package my_option;

option go_package="protoc-gen-my-option/test_proto"
首页 上一页 1 2 下一页 尾页 1/2/2
】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
上一篇golang执行命令 && 实时获取输出.. 下一篇一文读懂Go Http Server原理

最新文章

热门文章

Hot 文章

Python

C 语言

C++基础

大数据基础

linux编程基础

C/C++面试题目