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";

import "my_option.proto";

// 使用FiledOptions

option (my_option.my_file_option).ok = true;

驗證插件決議MyFileOptions

protoc -I=./proto --go_out=../ --my-option_out=../ --proto_path=test_proto test_option.proto

成功后會輸出以下內容,能夠看到my_options中的ok被成功決議，

2023/02/18 20:12:16 protogen plugin
2023/02/18 20:12:16 getFileOptions
2023/02/18 20:12:16 my_options: ok:true

決議擴展插件后生成我們自己的代碼

在上面的demo中添加代碼生成邏輯

package main

import (
	"fmt"
	"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 genCode(myOptions *my_proto.MyFileOptions) string {
	if myOptions.Ok != nil && myOptions.GetOk() {
		return `
			func HelloMyOptions() {
				fmt.Println("Hello MyOptions, MyOptions is ok!")
			}
	`
	}
	return `
		func HelloMyOptions() {
			fmt.Println("Hello MyOptions, MyOptions is not ok...")
		}
	`
}

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
			myOptions := getFileOptions(f)

			head := fmt.Sprintf(`
				package %s
				import "fmt"
			`, f.GoPackageName) // go檔案, package名是必須的，import生成后一般IDE會自動import,可以不用寫，除非某些場景需要提前加好import

			outFile := plugin.NewGeneratedFile(fmt.Sprintf("%s.my.go", f.GeneratedFilenamePrefix), f.GoImportPath)
			outFile.Write([]byte(head))
			outFile.Write([]byte(genCode(myOptions)))
		}
		return nil
	})
}

重新安裝插件，并生成我們的代碼

go install .
protoc -I=./proto --go_out=../ --my-option_out=../ --proto_path=test_proto test_option.proto

在test_proto目錄中會生成test_option.my.go,內容如下

package test_proto

import "fmt"

func HelloMyOptions() {
	fmt.Println("Hello MyOptions, MyOptions is ok!")
}

以上代碼地址

標籤：其他

上一篇：記憶體對齊

下一篇：day13-自定義攔截器