自定义拦截器

如果框架自带的拦截器都没有满足需求的，则可以自定义或者使用社区提供的拦截器。

自定义server中间件

步骤一：编写拦截器

import (  ...  grpc2 "google.golang.org/grpc"  ...)func UnaryServerInterceptorDemo() grpc2.UnaryServerInterceptor {    return func(ctx context.Context, req interface{}, info *grpc2.UnaryServerInfo, handler grpc2.UnaryHandler) (_ interface{}, err error) {
      fmt.Println("======= demo before=======")      defer func() {        if err := recover(); err != nil {          //打印错误堆栈信息          fmt.Println(err)
        }      }()
      resp, err := handler(ctx, req)
      fmt.Println("======= demo after=======")
      return resp, err    }}

步骤二：将拦截器加入到server中

import (  "github.com/go-eagle/eagle/pkg/transport/grpc")  grpcServer := grpc.NewServer(      grpc.Network("tcp"),      grpc.Address(":9000"),      grpc.Timeout(3*time.Second),      grpc.UnaryInterceptor(UnaryServerInterceptorDemo()),    )

到此自定义拦截器开发完毕，这里需要注意拦截器的顺序问题。

拦截器顺序demo

import (  ...  grpc2 "google.golang.org/grpc"  ...)
func UnaryServerInterceptorDemo() grpc2.UnaryServerInterceptor {    return func(ctx context.Context, req interface{}, info *grpc2.UnaryServerInfo, handler grpc2.UnaryHandler) (_ interface{}, err error) {
        fmt.Println("======= demo before=======")        defer func() {            if err := recover(); err != nil {                //打印错误堆栈信息                fmt.Println(err)
            }        }()
        resp, err := handler(ctx, req)
        fmt.Println("======= demo after=======")
        return resp, err    }}
func UnaryServerInterceptorDemo2() grpc2.UnaryServerInterceptor {    return func(ctx context.Context, req interface{}, info *grpc2.UnaryServerInfo, handler grpc2.UnaryHandler) (_ interface{}, err error) {
        fmt.Println("======= demo2 before=======")
        defer func() {            if err := recover(); err != nil {                //打印错误堆栈信息                fmt.Println(err)
            }        }()
        resp, err := handler(ctx, req)
        fmt.Println("======= demo2 after=======")        return resp, err    }}

输出结果

======= demo before============== demo2 before============= demo2 after============== demo after========