go-zero(九) RPC拦截器的使用

go zero 拦截器详解

在微服务架构中，拦截器(Interceptor)是一种强大的机制，用于在 RPC 调用的生命周期中注入自定义逻辑。go-zero可以设置多个拦截器，比如身份验证、日志记录、请求限流、性能监控等，这些都可以通过拦截器实现

一、拦截器基础概念

1. 什么是拦截器？

拦截器是 gRPC 框架提供的一种机制，允许开发者在 RPC 调用的不同阶段插入自定义逻辑。根据应用场景，拦截器分为两种类型：

• 服务端拦截器(Server Interceptor)：在服务端处理请求前后执行自定义逻辑
• 客户端拦截器(Client Interceptor)：在客户端发送请求前后执行自定义逻辑

2. 拦截器的主要作用

拦截器能够在 RPC 调用链中实现多种功能：

• 日志记录：捕获请求和响应信息，便于调试和监控
• 认证授权：验证请求者身份和权限
• 请求追踪：实现分布式链路追踪
• 性能度量：记录请求处理时间和资源使用情况
• 错误处理：统一处理和转换错误信息
• 限流熔断：控制请求流量，防止系统过载
• 请求参数校验：在服务层面验证请求参数

二、拦截器与中间件的区别与联系

在 go-zero 框架中，拦截器和中间件都是用于在请求处理流程中注入自定义逻辑的机制，但它们各自适用于不同的场景和协议层。

1. 概念与作用域

中间件 (Middleware):

• 作用域: 作用于 HTTP 请求处理流程
• 应用场景: REST API 服务
• 位置: API 网关层、HTTP 服务层
• 实现方式: 基于标准库的 http.Handler 接口

拦截器 (Interceptor):

• 作用域: 作用于 RPC 调用流程
• 应用场景: 微服务间内部通信
• 位置: RPC 服务层
• 实现方式: 基于 gRPC 的拦截器接口

2. 详细对比

特性	中间件 (Middleware)	拦截器 (Interceptor)
协议层	HTTP	gRPC (基于HTTP/2)
数据格式	多种格式 (JSON, XML, 表单等)	Protocol Buffers
调用方式	同步请求-响应	同步/异步、流式
客户端支持	主要在服务端使用	同时支持客户端和服务端
执行模式	请求-响应模式 (洋葱模型)	请求拦截和响应拦截
性能特性	HTTP 协议开销较大	gRPC 性能更优
适用场景	外部 API 接口、Web 服务	微服务内部通信、高性能场景

三、服务端拦截器

服务端拦截器在 RPC 服务处理请求前后执行，可以用于请求验证、权限检查、日志记录等场景。

AddUnaryInterceptors 是在 gRPC 服务器初始化时调用的方法，用于将一个或多个服务端拦截器注册到 gRPC 服务器上.

1. 服务端拦截器定义

服务端拦截器必须遵循 gRPC 定义的函数签名：

func(ctx context.Context, req interface{
   
   }, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{
   
   }, error)

各参数含义：

• ctx: 请求上下文，包含元数据
• req: 客户端请求参数
• info: RPC 方法的相关信息，如方法名
• handler: 实际处理请求的处理器

2. 实现服务拦截器

接下来我们使用两个，拦截器作为演示

日志拦截器
记录每个 RPC 请求的详细信息，包括方法名、请求参数、处理时间等：

func LoggingInterceptor(ctx context.Context, req interface{
   
   }, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{