如何访问gRPC端点或HTTP1/REST网关的原始请求负载？

gRPC端点场景 #

想拿到gRPC的原始请求负载，不能依赖处理器里的context——因为此时请求已经被反序列化为结构体了，原始数据早就被处理掉了。你需要在反序列化之前通过拦截器+自定义编码解码器来捕获：

• 实现一个自定义的Codec，在Unmarshal方法中把原始字节数据存到上下文里：

  type RawPayloadCodec struct {
      inner grpc.Codec
  }
  
  func (c RawPayloadCodec) Marshal(v interface{}) ([]byte, error) {
      return c.inner.Marshal(v)
  }
  
  func (c RawPayloadCodec) Unmarshal(data []byte, v interface{}) error {
      // 将原始负载存入上下文，需自定义context key
      ctx := context.WithValue(context.Background(), "grpc_raw_payload", data)
      // 结合拦截器传递上下文
      return c.inner.Unmarshal(data, v)
  }
  
  func (c RawPayloadCodec) String() string {
      return c.inner.String()
  }
  

• 配合Unary或Stream拦截器，将携带原始负载的上下文传递给处理器：

  func RawPayloadInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
      if rawPayload, ok := ctx.Value("grpc_raw_payload").([]byte); ok {
          ctx = context.WithValue(ctx, "grpc_raw_payload", rawPayload)
      }
      return handler(ctx, req)
  }
  

• 启动gRPC服务时注册自定义Codec和拦截器：

  server := grpc.NewServer(
      grpc.WithCodec(RawPayloadCodec{inner: grpc.NewDefaultCodec()}),
      grpc.WithUnaryInterceptor(RawPayloadInterceptor),
  )
  

之后在处理器里，就能通过ctx.Value("grpc_raw_payload")拿到原始请求字节。

HTTP/REST网关场景 #

如果是gRPC-Gateway这类HTTP网关，直接用HTTP中间件就能在请求解析前捕获原始请求体：

• 编写HTTP中间件，读取原始请求体并存入上下文：

  func CaptureRawPayload(next http.Handler) http.Handler {
      return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
          rawBody, err := io.ReadAll(r.Body)
          if err != nil {
              http.Error(w, "failed to read request body", http.StatusBadRequest)
              return
          }
          // 放回body保证后续流程正常使用
          r.Body = io.NopCloser(bytes.NewBuffer(rawBody))
          // 将原始负载存入上下文
          ctx := context.WithValue(r.Context(), "http_raw_payload", rawBody)
          next.ServeHTTP(w, r.WithContext(ctx))
      })
  }
  

• 启动HTTP网关服务器时注册中间件：

  mux := runtime.NewServeMux()
  // 注册gRPC-Gateway路由...
  httpServer := &http.Server{
      Addr:    ":8080",
      Handler: CaptureRawPayload(mux),
  }
  httpServer.ListenAndServe()
  

之后在网关的处理器或后续逻辑里，就能从上下文中取出原始请求体。

关键提示 #

不管是gRPC还是HTTP网关，默认流程中处理器拿到的context都不会包含原始负载——因为请求已经被解析/反序列化过了。必须在请求进入业务处理器之前，通过拦截器（gRPC）或中间件（HTTP）提前捕获原始数据，再存入上下文供后续使用。

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