GRPC自定义Codec统计压缩解压耗时遇空指针异常问题

核心原因 #

你的TimerCodec未正确关联gRPC默认的Protobuf Codec实例，导致调用Marshal时g.Codec字段为空，触发空指针异常。

具体修复步骤 #

1. 正确实现自定义Codec的初始化逻辑 #

创建TimerCodec时显式传入默认Protobuf Codec，而非依赖自动注入：

import (
    "time"
    "google.golang.org/grpc/encoding"
    "google.golang.org/grpc/encoding/proto"
)

type TimerCodec struct {
    Codec encoding.Codec // 底层实际处理编解码的Codec实例
}

// 初始化TimerCodec，主动绑定默认Protobuf Codec
func NewTimerCodec() *TimerCodec {
    return &TimerCodec{
        Codec: encoding.GetCodec(proto.Name), // 获取全局注册的默认proto codec
    }
}

// 覆写Marshal，加入计时逻辑
func (g *TimerCodec) Marshal(v any) ([]byte, error) {
    start := time.Now()
    data, err := g.Codec.Marshal(v)
    elapsed := time.Since(start)
    MeasureTimeAndCpu("marshal", elapsed) // 调用你的统计函数
    return data, err
}

// 覆写Unmarshal，加入计时逻辑
func (g *TimerCodec) Unmarshal(data []byte, v any) error {
    start := time.Now()
    err := g.Codec.Unmarshal(data, v)
    elapsed := time.Since(start)
    MeasureTimeAndCpu("unmarshal", elapsed)
    return err
}

// 必须返回默认proto codec的名称，确保gRPC能正确匹配消息类型
func (g *TimerCodec) Name() string {
    return proto.Name
}

2. 正确注册自定义Codec到客户端 #

有两种可靠的注册方式：

import "google.golang.org/grpc"

// 方式1：创建连接时显式指定Codec
conn, err := grpc.Dial("your-server:port",
    grpc.WithInsecure(),
    grpc.WithCodec(NewTimerCodec()), // 绑定自定义Codec到当前连接
)

// 方式2：替换全局默认的proto codec注册
encoding.RegisterCodec(NewTimerCodec())
// 后续创建的所有客户端连接都会自动使用该自定义Codec

3. 关键注意事项 #

• 自定义Codec的Name()方法必须返回proto.Name（即"proto"），否则gRPC客户端无法将Protobuf消息与你的Codec匹配。
• 绝不能依赖gRPC自动填充TimerCodec的Codec字段，必须主动初始化，gRPC只会调用你注册的Codec实例方法，不会自动封装底层Codec。
• 如果需求是统计压缩/解压的完整耗时（而非单纯编解码），你需要同时覆写gRPC的压缩器接口，将计时逻辑嵌入压缩/解压流程中。

额外排查点 #

• 检查是否在客户端连接时误用WithDefaultCallOptions但未指定Codec，导致连接使用空Codec实例。
• 确认NewTimerCodec()中encoding.GetCodec(proto.Name)未返回空值（默认情况下gRPC会自动注册proto codec，除非你主动移除过）。

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