如何在gRPC函数与拦截器间传递元数据？问题排查求助

核心问题 #

你在服务函数中使用context.WithValue生成了新的Context副本，但这个副本仅在服务函数内部有效——拦截器持有的还是调用handler前的原始Context，自然无法获取到新添加的UserID。需要注意：context.WithValue是不可变操作，每次调用都会返回新的Context实例，不会修改原Context的内容。

解决方法 #

方法1：使用指针类型的共享数据结构 #

通过在Context中存储一个指针类型的结构体，服务函数修改结构体的字段值，拦截器就能读取到更新后的数据：

1. 定义共享数据结构与统一Context Key（确保服务函数和拦截器都能访问）：

type ctxData struct {
    UserID string
}

// 用私有结构体作为Key，避免与其他包的Key冲突
var userIDKey = struct{}{}

2. 修改日志拦截器：

func GRPCLogger(ctx context.Context, req any, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp any, err error) {
    // 初始化共享数据并放入Context
    data := &ctxData{}
    ctx = context.WithValue(ctx, userIDKey, data)
    
    // 调用服务函数
    resp, err = handler(ctx, req)
    
    // 读取更新后的UserID
    if data.UserID != "" {
        log.Info().Str("userID", data.UserID).Msg("User ID from handler context in interceptor")
    }

    return resp, err
}

3. 修改服务函数：

func (s *YourGRPCServer) SomeGRPCFunction(ctx context.Context, req *pb.YourRequest) (*pb.YourResponse, error) {
    userID := "12345" // 实际场景从请求中动态获取
    
    // 更新共享数据的UserID字段
    if data, ok := ctx.Value(userIDKey).(*ctxData); ok {
        data.UserID = userID
    }

    return &pb.YourResponse{}, nil
}

方法2：提前在拦截器中提取UserID（若UserID来自请求） #

如果UserID可以从请求参数或元数据中直接提取，建议在拦截器中先完成提取并放入Context，服务函数直接使用即可，无需反向传递：

func GRPCLogger(ctx context.Context, req any, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp any, err error) {
    // 从请求中提取UserID（示例，根据实际请求结构调整）
    if yourReq, ok := req.(*pb.YourRequest); ok {
        userID := yourReq.UserID // 假设请求中包含UserID字段
        ctx = context.WithValue(ctx, userIDKey, userID)
    }

    resp, err = handler(ctx, req)
    
    // 直接从Context读取
    if userID, ok := ctx.Value(userIDKey).(string); ok {
        log.Info().Str("userID", userID).Msg("User ID from interceptor context")
    }

    return resp, err
}

注意事项 #

• Context Key的一致性：确保服务函数和拦截器使用的userIDKey是同一个变量（建议定义在公共包中，或使用私有结构体作为Key避免命名冲突）。
• 避免滥用Context：Context适合传递请求级别的数据，不适合传递复杂业务逻辑数据；如果是业务数据，建议通过函数参数直接传递。

内容的提问来源于stack exchange，提问作者Богдан Яцына