设为首页 加入收藏

TOP

Protobuf Options 自定义插件使用(二)
2023-07-23 13:33:27 】 浏览:55
Tags:Protobuf Options
;; 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!")
}

以上代码地址

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

最新文章

热门文章

Hot 文章

Python

C 语言

C++基础

大数据基础

linux编程基础

C/C++面试题目