如何在请求中为中间件设置Context值？遇go-staticcheck问题

为什么会报这两个错误？ #

你遇到的staticcheck错误本质是代码逻辑无效：

• SA4006：你在signup函数里给局部变量r赋值了新的请求对象，但赋值后直接return，这个新的r没有被任何后续代码使用，完全是多余操作。
• SA4017：r.WithContext()不会修改原请求对象，只会返回一个带新Context的新请求对象，你调用它却没真正利用返回的新对象，属于无意义调用。

更深层的问题是你的错误传递逻辑设计有问题：ErrorHandler是先执行next.ServeHTTP(w, r)（也就是调用signup处理器），再去读取原请求r的Context。但你在signup里修改的是局部变量的Context，这个修改根本传不到ErrorHandler里，所以就算没有staticcheck报错，你的ErrorHandler也永远读不到处理器里设置的错误。

解决方案（推荐用方案一，更符合Go最佳实践） #

方案一：改用「处理器返回错误」的模式

放弃用Context传错误，定义支持返回错误的自定义Handler类型，让ErrorHandler统一处理返回的错误：

1. 定义自定义Handler类型

// 自定义支持返回错误的Handler
type ErrorHandlerFunc func(w http.ResponseWriter, r *http.Request) error

2. 修改ErrorHandler包装逻辑

func ErrorHandler(next ErrorHandlerFunc) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 执行处理器，捕获返回的错误
        if err := next(w, r); err != nil {
            switch e := err.(type) {
            case *app.AppError:
                w.WriteHeader(e.Status)
                render.JSON(w, r, e)
            default:
                internalErr := app.NewInternalError("internal-error", "Something went wrong", nil)
                w.WriteHeader(internalErr.Status)
                render.JSON(w, r, internalErr)
            }
        }
    })
}

3. 修改signup处理器，让它返回错误

func (h *Http) signup(w http.ResponseWriter, r *http.Request) error {
    var params dto.SignUpDTO
    decoder := json.NewDecoder(r.Body)
    defer r.Body.Close()
    
    if err := decoder.Decode(&params); err != nil {
        // 直接返回错误，交给ErrorHandler处理
        return app.NewBadRequestError("auth/invalid-signup-params", "invalid email or password", err)
    }

    // 其他业务逻辑...
    return nil
}

4. 注册路由时用ErrorHandler包装

http.Handle("/signup", ErrorHandler(h.signup))

方案二：如果一定要保留Context传错误的模式

这种模式需要让ErrorHandler能读到处理器修改后的Context，但必须修改执行逻辑：

• 处理器不能直接return，而是要把修改后的r传给后续处理（但HTTP处理器本身是终端逻辑，所以更合理的方式是在处理器里直接触发错误响应，或者用自定义的Context载体）。不过这种方式代码冗余，不推荐，示例如下：

修改ErrorHandler，先创建一个可传递错误的Context容器（比如用一个指针类型的错误变量）：

func ErrorHandler(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 用指针存储错误，让处理器可以修改
        errPtr := new(error)
        ctx := context.WithValue(r.Context(), auth.ErrorContextKey, errPtr)
        // 把带错误指针的Context传给处理器
        next.ServeHTTP(w, r.WithContext(ctx))

        if *errPtr != nil {
            switch e := (*errPtr).(type) {
            case *app.AppError:
                w.WriteHeader(e.Status)
                render.JSON(w, r, e)
            default:
                internalErr := app.NewInternalError("internal-error", "Something went wrong", nil)
                w.WriteHeader(internalErr.Status)
                render.JSON(w, r, internalErr)
            }
        }
    })
}

然后修改signup处理器：

func (h *Http) signup(w http.ResponseWriter, r *http.Request) {
    var params dto.SignUpDTO
    decoder := json.NewDecoder(r.Body)
    defer r.Body.Close()
    
    if err := decoder.Decode(&params); err != nil {
        // 取出错误指针，赋值错误
        if errPtr, ok := r.Context().Value(auth.ErrorContextKey).(*error); ok {
            *errPtr = app.NewBadRequestError("auth/invalid-signup-params", "invalid email or password", err)
        }
        return
    }
}

这种方式虽然能解决staticcheck的错误，但逻辑绕，不如方案一简洁直观。

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