GRPC开发(8)-grpc服务开发-截取器

截取器

截取器可以对grpc server做一些例如验证、log等前置操作
s := grpc.NewServer(grpc.UnaryInterceptor(filter))

grpc-ecosystem项目中的链式截取器的支持

不过GRPC框架中只能为每个服务设置一个截取器，因此所有的截取工作只能在一个函数中完成。
开源的grpc-ecosystem项目中的go-grpc-middleware包已经基于GRPC对截取器实现了链式截取器的支持。
以下是go-grpc-middleware包中链式截取器的简单用法

import "github.com/grpc-ecosystem/go-grpc-middleware"
myServer := grpc.NewServer(
	grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(filter1, filter2, ...)),
	grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(filter1, filter2, ...)),
)

一个可用的例子哦

s := grpc.NewServer(grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(filter)))

//截取器增加了对GRPC方法异常的捕获
func filter(ctx context.Context, req interface{},info *grpc.UnaryServerInfo,handler grpc.UnaryHandler,) (resp interface{}, err error) {
	log.Println("fileter:", info)
	defer func() {
		if r := recover(); r != nil {
			err = fmt.Errorf("panic: %v", r)
		}
	}()
	return handler(ctx, req)
}