Go日志中间件如何获取后续中间件更新的Request Context变量？

你的核心问题是：日志中间件必须前置（因为认证失败要提前记录日志），但后续中间件通过req.WithContext()生成的新请求实例，无法被原始的req指针捕获，导致拿不到Context里的变量。

这里有个直接的解决方案：在日志中间件里加一层请求捕获的包装，把后续中间件传递的最新请求实例保存下来，具体代码修改如下：

import (
    "log"
    "net/http"
    "github.com/urfave/negroni"
)

func wrapHandlerWithLogging(wrappedHandler http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
        log.Printf("--> %s %s", req.Method, req.URL.Path)

        lrw := negroni.NewResponseWriter(w)
        var finalReq *http.Request

        // 包装handler，捕获经过后续中间件修改后的请求实例
        captureHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            finalReq = r
            wrappedHandler.ServeHTTP(w, r)
        })

        // 执行捕获包装后的handler链
        captureHandler.ServeHTTP(lrw, req)

        // 现在可以从finalReq里获取最新的Context变量
        if email, ok := finalReq.Context().Value(someContextKey).(string); ok {
            log.Printf("Authenticated user: %s", email)
        }

        statusCode := lrw.Status()
        log.Printf("<-- %d %s", statusCode, http.StatusText(statusCode))
    })
}

原理说明 #

• 后续中间件调用req.WithContext()时会返回新的请求实例，原始的req指针不会指向这个新实例。
• 新增的captureHandler会在执行真正的handler链前，把当前收到的请求实例（也就是经过所有前置中间件修改后的最新请求）赋值给finalReq。
• 不管认证中间件是否提前返回，finalReq都会保存到当时最新的请求状态，确保你能拿到Context里的变量。

如果用Negroni的标准中间件链写法，只需把这个日志中间件放在链的最前面即可：

n := negroni.New()
n.Use(negroni.NewLogger()) // 可替换为你的自定义日志中间件
n.Use(yourAuthMiddleware())
n.UseHandler(yourBusinessHandler())

这样既保证认证失败时日志能正常记录，又能拿到认证通过后设置的Context变量。

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