如何在gRPC UnaryClientInterceptor中使用reply参数？

首先你对拦截器的理解有个小偏差：UnaryClientInterceptor不是只在请求发送前执行，它包裹了整个gRPC调用的完整生命周期——从调用发起前，到调用invoker（实际发起请求并获取响应），再到响应返回后的处理阶段，都在拦截器的控制范围内。

reply参数的实用场景主要有这几个：

• 预初始化reply实例
  有些自定义的响应类型需要提前分配资源、设置默认值，或者依赖特定的初始化逻辑才能正常被gRPC的反序列化填充。拦截器可以在这里统一完成reply的初始化，不用每个业务调用都重复写这段代码。比如：

  func myInterceptor(ctx context.Context, method string, req, reply any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
      // 针对特定响应类型做预初始化
      if resp, ok := reply.(*MyCustomResponse); ok {
          resp.DefaultField = "default_value"
          resp.Buffer = make([]byte, 1024)
      }
      return invoker(ctx, method, req, reply, cc, opts...)
  }
  

• 响应返回后处理
  当invoker调用完成（也就是服务端响应已经被反序列化到reply里），拦截器可以继续操作这个reply对象：比如记录响应日志、统一修改响应字段、做响应校验等。举个日志的例子：

  func loggingInterceptor(ctx context.Context, method string, req, reply any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
      err := invoker(ctx, method, req, reply, cc, opts...)
      // 调用完成后，打印响应内容
      if err == nil {
          log.Printf("Method %s response: %v", method, reply)
      }
      return err
  }
  

• 测试场景下模拟响应
  在单元测试时，你可以编写拦截器直接给reply赋值，然后跳过真正的invoker调用，模拟服务端返回指定响应，这样不用启动真实的gRPC服务就能测试业务逻辑：

  func mockInterceptor(ctx context.Context, method string, req, reply any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
      if method == "/MyService/MyMethod" {
          // 直接填充模拟响应
          if resp, ok := reply.(*MyResponse); ok {
              resp.Result = "mocked_result"
          }
          return nil
      }
      // 其他方法走真实调用
      return invoker(ctx, method, req, reply, cc, opts...)
  }
  

内容的提问来源于stack exchange，提问作者mrmcgreg