Go中请求Context对象传递问题咨询（Grpc Gateway架构场景）

架构背景 #

我们的架构为：Client -> gRPC Gateway -> Server，网关负责请求的处理与认证，完成后转发至Server应用；gRPC Gateway使用alice包串联中间件。

问题1：请求上下文在中间件间传递时，第二个中间件是从req.Header还是req.Context()获取前一个中间件传递的上下文？ #

• 结论：必须从req.Context()获取，而非Header。
• 原因：
  • req.Context()是Go程序内部的上下文传递机制，专门用于携带请求生命周期内的进程内部数据（如认证信息、追踪ID等）。前序中间件若要传递数据，需通过context.WithValue创建新上下文，再用req.WithContext(newCtx)生成新请求对象传给下一个中间件。
  • Header是HTTP协议层面的元数据载体，仅用于传递协议规定或自定义的公开请求头信息，不适合传递程序内部上下文数据。

问题2：在Go中使用req.WithContext(ctx)发送请求至Server应用时，上下文内容会进入请求Header吗？若不会，上下文会如何处理？ #

• 结论：req.WithContext(ctx)设置的上下文不会自动进入请求Header。
• 具体说明：
  • 该上下文仅存在于发起请求的gRPC Gateway进程内部，用于控制请求生命周期（如超时、取消）或携带进程内临时数据，不会随HTTP请求自动发送到Server。
  • 若需将上下文内的自定义数据传递给Server，需手动将数据写入请求Header（例如将认证Token放入Authorization头、追踪ID放入X-Trace-ID自定义头），Server端再从Header读取值并重新存入自身请求上下文供后续逻辑使用。

Server端无法获取一致上下文内容的解决方法 #

由于网关通过req.WithContext(ctx)设置的上下文是进程内部数据，不会自动传递到Server，需手动做Header转发：

网关端示例代码 #

// 中间件：将上下文内的用户ID写入请求Header
func injectUserID(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if userID, ok := r.Context().Value("user_id").(string); ok {
            r.Header.Set("X-User-ID", userID)
        }
        next.ServeHTTP(w, r)
    })
}

// 使用alice串联中间件
middlewareChain := alice.New(authMiddleware, injectUserID)

Server端示例代码 #

// 处理请求时从Header读取数据并存入上下文
func userHandler(w http.ResponseWriter, r *http.Request) {
    userID := r.Header.Get("X-User-ID")
    ctx := context.WithValue(r.Context(), "user_id", userID)
    
    // 后续业务逻辑使用该上下文
    fmt.Printf("Received user ID from context: %s\n", ctx.Value("user_id"))
}

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