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

在Go语言中使用gRPC时,拦截器(Interceptor)是一个非常实用的机制,能够帮助我们在请求处理前后插入通用逻辑,比如日志记录、认证鉴权、错误处理、监控统计等。gRPC本身不直接提供中间件概念,但通过拦截器可以实现类似功能。
什么是gRPC拦截器
gRPC拦截器本质上是一个函数,它在gRPC方法执行前或执行后被调用。gRPC支持两种类型的拦截器:
Unary Interceptor:用于一元调用(即普通的请求-响应模式) Stream Interceptor:用于流式调用(客户端流、服务端流、双向流)
它们分别可以在grpc.Server和grpc.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服务器:
Seede AI
AI 驱动的设计工具
586 查看详情
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拦截器的使用,能让你的服务更具可观测性和可维护性。关键在于理解拦截器的执行时机和上下文传递机制,合理设计通用逻辑的边界。
以上就是Golang如何使用gRPC拦截器实现请求拦截_Golang gRPC拦截器实践详解的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1069753.html
微信扫一扫
支付宝扫一扫