gRPC拦截器那点事，希望帮到你

上一篇介绍了gRPC的接口认证，我们客户端需要实现gRPC提供的接口，然后在服务端业务接口实现中通过metadata获取认证信息，进行判断，那么当我们有几十个，几百个业务接口时，如果都在接口实现中去做，那将是一个噩梦，也不符合DRY(Don't Repeat Yourself)原则，今天一起来看看如何通过gRPC的拦截器做到统一接口认证工作

初识gRPC拦截器

• gRPC在grpc包中定义了一个变量,如下,这个变量叫做UnaryServerInterceptor（一元服务拦截器），

• 其类型是一个函数，这个函数有,4个入参，两个出参,介绍如下
• ctx context.Context 上下文
• req interface{} 用户请求的参数
• info UnaryServerInfo RPC 方法的所有信息，定义如下

• handler UnaryHandler RPC方法本身
• resp interface{} RPC方法执行结果

如何使用

• 首先定义一个拦截器

• 在服务端启动时将拦截器添加进去

• 如上就是服务端使用拦截器的所有步骤，客户端在访问服务端时就会被拦截

如何添加多个拦截器

• 有人说我看上面代码grpc.NewServer是一个可变参数，我传多个不就好了吗？真的是这样吗，我们来试试，代码如下我们添加了两个拦截器

• 很遗憾，服务端启动失败，报错信息如下,什么含义呢，意思是说，这个一元服务拦截器只能设置一个，不能重复，其实从名字就能看出，一元拦截器，就是说只能设置一个拦截器，gRPC有意的阻止拦截器链的形式

panic: The unary server interceptor was already set and may not be reset. [recovered] panic: The unary server interceptor was already set and may not be reset.

• 那我们如果真的需要拦截器链，该如何配置呢，核心思想是递归，代码如下:

• github上已经有这么一个项目，如下
• https://github.com/grpc-ecosystem/go-grpc-middleware
• 这个项目提供了拦截器的interceptor链式的功能，还有其它一些功能，大家可以去学习学习。

gRPC还有哪些拦截器

统一在grpc包下，其他拦截器如下

type UnaryClientInterceptor

这是一个客户端上的拦截器，在客户端真正发起调用之前，进行拦截，这是一个实验性的api，这是gRPC官方的说法

type StreamClientInterceptor

在流式客户端调用时，通过拦截clientstream的创建，返回一个自定义的clientstream,可以做一些额外的操作，这是一个实验性的api，这是gRPC官方的说法

type UnaryServerInterceptor (就是上面我们demo中的拦截器)

type StreamServerInterceptor

拦截服务器上流式rpc的执行

gRPC的拦截分类

• 按功能来分

1. 一元拦截器 UnaryInterceptor
2. 流式拦截器 StreamInterceptor

• 按端来分

1. 客户端拦截器 ClientInterceptor
2. 服务端拦截器 ServerInterceptor

• 大家平时使用时灵活使用即可，慢慢的就会融会贯通