文档：https://godoc.org/github.com/grpc-ecosystem/go-grpc-middleware

gRPC是一个非常棒的RPC中间件，它在Golang世界中被广泛采用。然而，上游的gRPC代码库相对比较简单。
这个包及其大部分子包为gRPC提供了通常需要的中间件:用于退休的客户端拦截器、用于输入验证和验证的服务器端拦截器、用于链接所述拦截器的功能、元数据便利方法等等。

func ChainStreamClient(interceptors …grpc.StreamClientInterceptor) grpc.StreamClientInterceptor

  • 从许多拦截器链中创建一个拦截器。执行按从左到右的顺序执行

func ChainStreamServer(interceptors …grpc.StreamServerInterceptor) grpc.StreamServerInterceptor

  • 从许多拦截器链中创建一个拦截器。执行按从左到右的顺序执行

func ChainUnaryClient(interceptors …grpc.UnaryClientInterceptor) grpc.UnaryClientInterceptor

  • 从许多拦截器链中创建一个拦截器。执行按从左到右的顺序执行

func ChainUnaryServer(interceptors …grpc.UnaryServerInterceptor) grpc.UnaryServerInterceptor

  • 从许多拦截器链中创建一个拦截器。执行按从左到右的顺序执行

func WithStreamServerChain(interceptors …grpc.StreamServerInterceptor) grpc.ServerOption
func WithUnaryServerChain(interceptors …grpc.UnaryServerInterceptor) grpc.ServerOption

  1. WithStreamServerChain，WithUnaryServerChain是为了简化代码的作用
  2. 例如：
  3. s := grpc.NewServer(grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(
  4.     grpc_auth.UnaryServerInterceptor(AuthInterceptor))))
  5. 可以写为
  6. s := grpc.NewServer(grpc_middleware.WithUnaryServerChain(
  7.     grpc_auth.UnaryServerInterceptor(AuthInterceptor)))

自定义拦截器，服务端：

  1. package main
  3. import (
  4.     "context"
  5.     "google.golang.org/grpc"
  6.     "google.golang.org/grpc/codes"
  7.     "google.golang.org/grpc/status"
  8.     "log"
  9.     "net"
  10.     pb "study/grpc/protobuf/person"
  11.     "github.com/grpc-ecosystem/go-grpc-middleware"
  12. )
  14. // 全局恐慌
  15. func RecoveryInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
  16.     defer func() {
  17.         if e := recover(); e != nil {
  18.             err = status.Errorf(codes.Internal, "Panic err: %v", e)
  19.         }
  20.     }()
  21.     return handler(ctx, req)
  22. }
  24. // 日志记录
  25. func LoggingInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
  26.     log.Printf("gRPC method: %s, %v", info.FullMethod, req)
  27.     resp, err := handler(ctx, req)
  28.     log.Printf("gRPC method: %s, %v", info.FullMethod, resp)
  29.     return resp, err
  30. }
  32. func main() {
  33.     lis, err := net.Listen("tcp", port)
  34.     if err != nil {
  35.         log.Fatalf("failed to listen: %v", err)
  36.     }
  37.     opts := grpc_middleware.WithUnaryServerChain(
  38.             RecoveryInterceptor, // 全局恐慌
  39.             LoggingInterceptor,  // 日志记录
  40.             )
  41.     s := grpc.NewServer(opts) //起一个服务
  42.     pb.RegisterPersonserverServer(s,&Server{})
  44.     if err := s.Serve(lis); err != nil {
  45.         log.Fatalf("failed to serve: %v", err)
  46.     }
  47. }

自定义拦截器，客户端：

  1. func MyInterceptor() grpc.UnaryClientInterceptor{
  2.     // 自己的处理逻辑
  3.     return func(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
  4.         // 自己的处理逻辑
  5.         return invoker(ctx, method, req, reply, cc, opts...)
  6.     }
  7. }