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("耗时:%sn", 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