怎样开发一个grpc拦截器
在开发grpc服务时，我们经常会遇到一些通用的需求，比如：日志、链路追踪、鉴权等。这些需求可以通过grpc拦截器来实现。本文使用go语言来实现一个 grpc一元模式(Unary)拦截器，上报链路追踪信息。
原始类型定义
我们可以在grpc的源码包里(interceptor.go)，找到一元模式拦截器的类型定义：
// UnaryServerInterceptor provides a hook to intercept the execution of a unary RPC on the server. info// contains all the information of this RPC the interceptor can operate on. And handler is the wrapper// of the service method implementation. It is the responsibility of the interceptor to invoke handler// to complete the RPC.type UnaryServerInterceptor func(ctx context.Context, req any, info *UnaryServerInfo, handler UnaryHandler) (resp any, err error)
从上面的定义可以看出，一元模式拦截器是一个函数，接收四个参数，返回两个参数。下面我们来看一下这四个参数的含义：
-
ctx：上下文对象。
-
req：请求参数
-
info：包含了RPC的元信息，比如服务名、方法名等。
-
handler 实例的方法，用来调用实际的RPC方法。
我们只需要实现一个上述类型的函数，在里面实现我们的功能，然后再执行handler函数，就可以实现一个拦截器了。
实现拦截器
我们新建一个项目grpcdemo。
服务定义
我们先在项目目录下新建一个proto文件，定义一个服务：
定义一个Makefile:
protos:protoc --proto_path=./ --go_out=pb --go-grpc_out=pb --go_opt=paths=source_relative --go-grpc_opt=paths=source_relative ./*.prototidy:go mod tidyrun:go mod tidygo run main.go
执行以下命令，生成go代码：
make protos
代码开发
第一步，新建一个tracing.go，初始化链路追踪器：
第二步，在main.go文件中，添加相关代码：
在上面的代码中，我们启动了一个grpc服务，监听8091端口。在启动grpc服务前，初始化了链路追踪信息，然后在grpc服务中，使用了自定义的拦截器。在自定义拦截器中，我们上报了链路追踪信息。
启动jaeger服务
具体的启动方式，可以参考官方文档：www.jaegertracing.io/docs/1.26/g…
测试
我们使用goland的grpc插件，来测试一下：
#GRPC localhost:8091/pb.HelloService/Hello{"name": "ZhangSan"}#GRPC localhost:8091/pb.HelloService/HelloAgain{"name": "ZhangSan"}
测试结果：
我们再打开jaeger的UI，查看链路追踪信息：可以看到，我们的链路追踪信息已经上报到了jaeger服务。
链接：https://juejin.cn/post/7313241977735577627
（版权归原作者所有，侵删）
发表评论