Golang如何使用gRPC拦截器实现请求拦截_Golang gRPC拦截器实践详解

gRPC拦截器是go中实现通用逻辑的关键机制,支持一元和流式调用。服务端可用来做日志、耗时统计,客户端可用于认证、重试。通过grpc.UnaryInterceptor注册,结合grpc-middleware库可链式组合多个拦截器,提升服务可观测性与可维护性。

Golang如何使用gRPC拦截器实现请求拦截_Golang gRPC拦截器实践详解

Go语言中使用gRPC时,拦截器(Interceptor)是一个非常实用的机制,能够帮助我们在请求处理前后插入通用逻辑,比如日志记录、认证鉴权、错误处理、监控统计等。gRPC本身不直接提供中间件概念,但通过拦截器可以实现类似功能。

什么是gRPC拦截器

gRPC拦截器本质上是一个函数,它在gRPC方法执行前或执行后被调用。gRPC支持两种类型的拦截器:

  • Unary Interceptor:用于一元调用(即普通的请求-响应模式)
  • stream Interceptor:用于流式调用(客户端流、服务端流、双向流)

它们分别可以在grpc.Servergrpc.ClientConn上注册,实现服务端和客户端的拦截逻辑。

服务端一元拦截器实现

服务端的一元拦截器可用于统一处理请求日志、panic恢复、耗时统计等。下面是一个简单的日志和耗时统计拦截器示例:

立即学习go语言免费学习笔记(深入)”;

 func loggingInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {     start := time.Now()     log.Printf("Received request: %s", info.FullMethod)          // 调用实际的处理函数     resp, err := handler(ctx, req)          log.Printf("Finished request: %s, duration: %v, error: %v", info.FullMethod, time.Since(start), err)     return resp, err } 

注册该拦截器到gRPC服务器:

Golang如何使用gRPC拦截器实现请求拦截_Golang gRPC拦截器实践详解

ViiTor实时翻译

ai实时多语言翻译专家!强大的语音识别、AR翻译功能。

Golang如何使用gRPC拦截器实现请求拦截_Golang gRPC拦截器实践详解116

查看详情 Golang如何使用gRPC拦截器实现请求拦截_Golang gRPC拦截器实践详解

 server := grpc.NewServer(grpc.UnaryInterceptor(loggingInterceptor)) pb.RegisterYourServiceServer(server, &yourService{}) 

客户端一元拦截器实现

客户端拦截器常用于添加认证头、重试机制或请求日志。例如,添加JWT Token到每个请求:

 func authInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {     ctx = metadata.appendToOutgoingContext(ctx, "authorization", "Bearer your-jwt-token")     return invoker(ctx, method, req, reply, cc, opts...) } 

创建客户端连接时启用拦截器:

 conn, err := grpc.Dial("localhost:50051", grpc.WithUnaryInterceptor(authInterceptor), grpc.WithInsecure()) if err != nil {     log.Fatal(err) } 

结合多个拦截器

实际项目中通常需要多个拦截器协同工作。可以使用grpc-middleware库来简化组合:

 import "github.com/grpc-ecosystem/go-grpc-middleware"  interceptors := []grpc.UnaryServerInterceptor{     loggingInterceptor,     recoveryInterceptor, // 恢复panic     validationInterceptor, // 参数校验 }  combined := grpc_middleware.ChainUnaryServer(interceptors...) server := grpc.NewServer(grpc.UnaryInterceptor(combined)) 

这个方式让代码更清晰,也便于测试和维护。

基本上就这些。掌握gRPC拦截器的使用,能让你的服务更具可观测性和可维护性。关键在于理解拦截器的执行时机和上下文传递机制,合理设计通用逻辑的边界。

上一篇
下一篇
text=ZqhQzanResources