共计 1646 个字符,预计需要花费 5 分钟才能阅读完成。
前言
上篇文章《Go – 如何编写 ProtoBuf 插件 (二)》,分享了基于 自定义选项
定义了 interceptor
插件,而后在 helloworld.proto
中应用了插件,最初在 golang
代码中获取到应用的插件信息。
接上篇,持续分享。
既然获取到了插件信息,咱们就能够应用它们。本文次要分享在 grpc.ServerOption
中的 grpc.UnaryInterceptor
中应用。
演示代码
还是以上篇文章中 helloworld.proto
为例。
// 生成 helloworld.pb.go
// 生成 helloworld_grpc.pb.go
// 应用的 protoc --version 为 libprotoc 3.18.1
// 应用的 protoc-gen-go --version 为 protoc-gen-go v1.27.1
// 应用的 protoc-gen-go-grpc --version 为 protoc-gen-go-grpc 1.1.0
// 在根目录下执行 protoc 命令
protoc --go_out=helloworld/gen --go-grpc_out=helloworld/gen helloworld/helloworld.proto
一、基于上篇文章中获取 options
的代码进行批改,次要是将其存入到构造体即可。
// 演示代码,构造体
var handlers = &struct {Methods map[string]*options.MethodHandler // FullMethod : Handler
Services map[string]*options.ServiceHandler // FullMethod : Handler
}{Methods: make(map[string]*options.MethodHandler),
Services: make(map[string]*options.ServiceHandler),
}
二、在 grpc.NewServer
中应用拦截器。
// 演示代码
serverOptions := []grpc.ServerOption{grpc.UnaryInterceptor(unaryServerInterceptor()),
}
srv := grpc.NewServer(serverOptions...)
resolveFileDescriptor() // 解析 options 扩大项
三、在 unaryServerInterceptor()
办法中,能够依据以后申请的服务名和办法名获取到对应设置的 options
。
// 演示代码
fullMethod := strings.Split(info.FullMethod, "/")
serviceName := fullMethod[1]
// 获取 service options
getServiceHandler(serviceName)
// 获取 method options
getMethodHandler(info.FullMethod)
四、本人写一个 grpcclient
调用一下即可。
--- /helloworld.Greeter/SayHello1 ---
service use interceptor authorization: login_token
method use interceptor whitelist: ip_whitelist
method use interceptor logger: true
至此,在 grpc.UnaryInterceptor
中就能够获取到 options
了,其余演示代码我就不贴了。
最初,通过获取到的 options
,便能够执行本人定义的具体方法。
小结
通过最近的“如何编写 ProtoBuf 插件”这三篇文章,置信你对编写 ProtoBuf 插件有一点小的意识,心愿对你可能有所帮忙。
举荐浏览
- Go – 如何编写 ProtoBuf 插件 (二)?
- Go – 如何编写 ProtoBuf 插件 (一)?
- Go – 对于 protoc 工具的小纳闷
- Go – 对于 .proto 文件的小思考
- Go – 基于逃逸剖析来晋升程序性能
正文完