使用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))










