如何在Go gRPC拦截器中无需重新序列化访问原始序列化数据？

针对你遇到的stats.InPayload.Data字段被移除的问题，以下两种方案可以让你无需重新序列化就能获取原始请求数据：

方案一：使用新版stats.Handler API获取 #

新版本gRPC（v1.40+）将InPayload.Data替换为Serialized接口，通过GetSerialized()方法即可直接获取原始序列化字节数据，无需修改原有stats handler的核心逻辑：

import "google.golang.org/grpc/stats"

func (h *myStatsHandler) HandleRPC(ctx context.Context, s stats.RPCStats) {
    switch payload := s.(type) {
    case *stats.InPayload:
        // 直接获取原始序列化数据
        rawData := payload.GetSerialized()
        // 在这里处理rawData，比如日志记录、校验等
    }
}

这种方式是官方推荐的方案，性能开销极低，不会侵入业务逻辑。

方案二：自定义编解码器，在拦截器中传递原始数据 #

如果必须在拦截器中直接获取原始数据，可以通过自定义编解码器，在解码阶段将原始数据存入上下文，供后续拦截器使用：

1. 实现自定义编解码器 #

import (
    "context"
    "google.golang.org/grpc"
)

// 自定义编解码器，包裹默认proto编解码器
type rawDataCodec struct {
    grpc.Codec
}

func (c *rawDataCodec) Unmarshal(data []byte, v interface{}) error {
    // 将原始数据存入当前请求的上下文
    ctx := grpc.UnaryServerInterceptorContext()
    ctx = context.WithValue(ctx, "grpc_raw_request", data)
    grpc.SetUnaryServerInterceptorContext(ctx)
    // 执行原解码逻辑
    return c.Codec.Unmarshal(data, v)
}

2. 启动服务器时指定编解码器和拦截器 #

// 定义拦截器，从上下文取出原始数据
func rawRequestInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
    if rawData, ok := ctx.Value("grpc_raw_request").([]byte); ok {
        // 在这里使用原始序列化数据
    }
    return handler(ctx, req)
}

// 启动gRPC服务器
server := grpc.NewServer(
    grpc.ForceCodec(&rawDataCodec{Codec: grpc.NewDefaultCodec()}),
    grpc.UnaryInterceptor(rawRequestInterceptor),
)

注意事项 #

• 该方案仅适用于Unary RPC，若需要支持流式RPC，需要额外实现grpc.StreamCodec接口；
• 上下文key建议定义为私有常量，避免冲突，比如var rawRequestKey = struct{}{}{}。

内容的提问来源于stack exchange，提问作者Muhammed Jishin Jamal TCP