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...)), )
这个方式让代码更清晰，易于维护。
基本上就这些。掌握拦截器后，你可以统一处理横切关注点，避免在每个方法中重复写日志、认证等逻辑。关键是理解拦截器的函数签名和执行时机，然后根据业务需求封装通用功能。不复杂但容易忽略细节，比如上下文传递和错误处理。