使用gRPC拦截器可统一处理日志、认证等逻辑，无需修改业务代码。2. 一元拦截器通过grpc.UnaryServerInterceptor实现，用于记录请求耗时与日志。3. 流式拦截器通过grpc.StreamServerInfo处理流式RPC调用。4. 在grpc.NewServer时注册拦截器选项。5. 客户端可通过grpc.WithUnaryInterceptor添加认证头信息，提升可观测性与安全性。
在Go语言中使用gRPC拦截器可以统一处理请求前后的逻辑，比如日志记录、认证、错误处理等。通过拦截器，你可以在不修改业务代码的前提下增强服务行为。下面是一个使用gRPC一元拦截器和流式拦截器的简单示例。
1. 定义并实现一元拦截器
一元拦截器用于处理普通的RPC调用（非流式）。你可以用它来记录请求耗时、打印日志或进行身份验证。
注意：拦截器函数需要符合
grpc.UnaryServerInterceptor
类型定义。
示例：实现一个日志和耗时统计的一元拦截器
func loggingUnaryInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
start := time.Now()
log.Printf(“Received unary request: %s”, info.FullMethod)
resp, err := handler(ctx, req)
log.Printf(“Completed unary request: %s, duration: %v, error: %v”,
info.FullMethod, time.Since(start), err)
return resp, err
}
2. 定义并实现流式拦截器
流式拦截器用于处理客户端流、服务端流或双向流的gRPC方法。
立即学习“go语言免费学习笔记（深入）”；
示例：一个简单的流式拦截器
func loggingStreamInterceptor(srv interface{}, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {
start := time.Now()
log.Printf(“Received stream request: %s”, info.FullMethod)
err := handler(srv, ss)
log.Printf(“Completed stream request: %s, duration: %v, error: %v”,
info.FullMethod, time.Since(start), err)
return err
}
3. 注册拦截器到gRPC服务器
在启动gRPC服务器时，通过
grpc.UnaryInterceptor
和
grpc.StreamInterceptor
选项注册拦截器。
func main() {
lis, err := net.Listen(“tcp”, “:50051”)
if err != nil {
log.Fatalf(“failed to listen: %v”, err)
}
opts := []grpc.ServerOption{
grpc.UnaryInterceptor(loggingUnaryInterceptor),
grpc.StreamInterceptor(loggingStreamInterceptor),
}
s := grpc.NewServer(opts…)
pb.RegisterYourServiceServer(s, &server{})
log.Println(“gRPC server listening on :50051”)
if err := s.Serve(lis); err != nil {
log.Fatalf(“failed to serve: %v”, err)
}
}
4. 客户端也可使用拦截器
如果你希望在客户端也做类似操作（如添加认证头），可以设置客户端拦截器。
func authClientInterceptor(ctx context.Context, method string, req, reply interface{},
cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts …grpc.CallOption) error {
newCtx := metadata.AppendToOutgoingContext(ctx, “authorization”, “Bearer “)
return invoker(newCtx, method, req, reply, cc, opts…)
}
// 使用
conn, err := grpc.Dial(address, grpc.WithInsecure(),
grpc.WithUnaryInterceptor(authClientInterceptor))基本上就这些。拦截器是gRPC中非常实用的功能，合理使用能极大提升代码可维护性和可观测性。
以上就是Golang使用gRPC拦截器处理请求示例的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1405556.html
微信扫一扫 
支付宝扫一扫 
