grpc拦截器的使用

gRPC作为通用RPC框架，内置了拦截器功能。包括服务器端的拦截器和客户端拦截器，使用上大同小异。主要作用是在rpc调用的前后进行额外处理。

拦截器在很多场景中使用，比如调用接口前验证用户是否登录，比如接口中判断用户的useragent做一些反爬的策略等等，大量的情况就是把请求拦截一下，做一下接口的预处理，我们不可能在每个接口中都写一遍，需要做统一的拦截器。

实现简单的拦截器

1）服务端：grpc.UnaryInterceptor(interceptor)

interceptor是自定义的拦截器函数，追踪函数的参数可知，interceptor是一个:

type UnaryServerInterceptor func(ctx context.Context, req interface{}, info *UnaryServerInfo, handler UnaryHandler) (resp interface{}, err error)

参数含义： 

• ctx context.Context：请求上下文
• req interface{}：RPC 方法的请求参数
• info *UnaryServerInfo：RPC 方法的所有信息
• handler UnaryHandler：RPC 方法真正执行的逻辑

 案例：

//拦截器
interceptor :=  func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error){
	fmt.Println("接收到了一个新的请求")
	res,err := handler(ctx, req)
	fmt.Println("请求已完成")
	return res, err
}
opt := grpc.UnaryInterceptor(interceptor)
g := grpc.NewServer(opt)
//...

2）客户端：grpc.WithUnaryInterceptor(interceptor)

interceptor是自定义的拦截器函数，追踪函数的参数可知，interceptor是一个:

type UnaryClientInterceptor func(ctx context.Context, method string, req, reply interface{}, cc *ClientConn, invoker UnaryInvoker, opts ...CallOption) error

案例：

interceptor := func(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error{
	start := time.Now()
	err := invoker(ctx, method, req, reply, cc, opts...)
	fmt.Printf("耗时：%s\n", time.Since(start))
	return err
}
opt := grpc.WithUnaryInterceptor(interceptor)
conn, err := grpc.Dial(":8083", grpc.WithInsecure(), opt)
//...

3）一些开源的拦截器

https://github.com/grpc-ecosystem/go-grpc-middleware