Go-Chi路由中处理函数能否覆盖中间件设置的Content-Type？

我现有一个始终返回JSON格式的Go API，使用Chi路由并在主函数中通过如下中间件配置：

func router() http.Handler {
    r := chi.NewRouter()
    r.Use(render.SetContentType(render.ContentTypeJSON))
    // ...其他路由配置
}

现在我希望部分函数返回不同类型的文件，若在路由处理函数中按如下方式设置Content-Type：

func handleRequest(w http.ResponseWriter, r *http.Request) {
    fileBytes, err := ioutil.ReadFile("test.png")
    if err != nil {
        panic(err)
    }
    w.WriteHeader(http.StatusOK)
    w.Header().Set("Content-Type", "application/octet-stream")
    w.Write(fileBytes)
    return
}

请问此操作能否覆盖该函数对应的中间件Content-Type设置？

可以覆盖，但你当前代码里有个关键顺序问题——先调用WriteHeader再修改Header是无效的。

w.WriteHeader(http.StatusOK)会直接把当前已有的Header写入响应，之后再调用w.Header().Set()修改的内容根本不会被发送给客户端。正确的操作顺序应该是先设置Header，再发送响应头：

func handleRequest(w http.ResponseWriter, r *http.Request) {
    fileBytes, err := ioutil.ReadFile("test.png")
    if err != nil {
        panic(err)
    }
    // 建议用更精准的Content-Type，比如image/png比application/octet-stream更合适
    w.Header().Set("Content-Type", "image/png")
    w.WriteHeader(http.StatusOK)
    w.Write(fileBytes)
    return
}

核心逻辑是：Chi的render.SetContentType中间件只是在请求处理流程的早期设置了Content-Type头，而http.ResponseWriter的Header对象是允许后续处理函数修改的，只要在发送响应头（调用WriteHeader或第一次调用Write）之前完成修改，就能完全覆盖中间件设置的值。

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