gRPC拦截器是Go中用于在请求前后插入通用逻辑的钩子函数，分为服务端和客户端的一元与流式拦截器。通过grpc.UnaryInterceptor注册服务端拦截器可实现日志、错误恢复等，如loggingInterceptor记录请求信息；客户端拦截器如authInterceptor可添加认证头。多个拦截器可通过grpc-middleware.ChainUnaryServer组合使用，便于统一处理认证、监控等横切关注点，提升代码复用性与可维护性。
在Go语言中使用gRPC拦截器可以让你在请求处理前后插入通用逻辑，比如日志记录、认证、监控、错误处理等。gRPC本身不直接提供拦截器功能，但通过 grpc-go 提供的中间件机制（即拦截器）可以轻松实现。
什么是gRPC拦截器
gRPC拦截器是一种钩子函数，可以在gRPC方法执行前或后运行。分为两种类型：
- Unary Interceptor：用于处理一元调用（普通请求-响应模式）
- Stream Interceptor：用于处理流式调用（客户端流、服务端流、双向流）
你可以为客户端和服务端分别设置拦截器。
服务端一元拦截器使用方法
定义一个服务端一元拦截器函数，其签名必须符合 grpc.UnaryServerInterceptor 类型：
立即学习“go语言免费学习笔记（深入）”；
示例：实现一个简单的日志和错误恢复拦截器
func loggingInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
// 请求前
log.Printf("Received request: %s", info.FullMethod)
// 处理请求
resp, err = handler(ctx, req)
// 请求后
if err != nil {
log.Printf("Error handling request: %v", err)
} else {
log.Printf("Request completed successfully")
}
return resp, err
}
注册到gRPC服务器：
server := grpc.NewServer(grpc.UnaryInterceptor(loggingInterceptor))
pb.RegisterYourServiceServer(server, &yourService{})
客户端一元拦截器使用方法
客户端拦截器可用于添加认证头、重试、超时等逻辑。
定义一个客户端一元拦截器：
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-token")
// 调用实际的RPC
return invoker(ctx, method, req, reply, cc, opts...)
}
创建客户端时使用：
conn, err := grpc.Dial("localhost:50051",
grpc.WithInsecure(),
grpc.WithUnaryInterceptor(authInterceptor),
)
if err != nil {
log.Fatal(err)
}
多个拦截器的组合
如果你需要多个拦截器，可以使用第三方库如 github.com/grpc-ecosystem/go-grpc-middleware 来链式组合。
安装：
go get github.com/grpc-ecosystem/go-grpc-middleware使用示例：
import "github.com/grpc-ecosystem/go-grpc-middleware"
// 组合一元拦截器
unaryInterceptors := []grpc.UnaryServerInterceptor{
loggingInterceptor,
recoveryInterceptor, // 恢复panic
}
server := grpc.NewServer(
grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(unaryInterceptors...)),
)
这个方式让代码更清晰，易于维护。
基本上就这些。掌握拦截器后，你可以统一处理横切关注点，避免在每个方法中重复写日志、认证等逻辑。关键是理解拦截器的函数签名和执行时机，然后根据业务需求封装通用功能。不复杂但容易忽略细节，比如上下文传递和错误处理。
