如果已设置标头，如何使用自定义 http 处理程序/中间件？

Question

我试图在 go 中链接 HTTP 处理程序以提供一些附加功能，如下所示：

package router

import (

    // snip

    "github.com/gorilla/mux"

    "github.com/gorilla/handlers"

    "net/http"

)

// snip

r := mux.NewRouter()

/* routing code */

var h http.Handler

h = r

if useGzip {

    h = handlers.CompressHandler(h)

}

if useLogFile {

    fn := pathToLog

    accessLog, err := os.OpenFile(fn, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0666)

    if err != nil {

        panic(err)

    }

    h = handlers.CombinedLoggingHandler(accessLog, h)

}

// etc...

问题是，如果任何HTTP头已经由一个控制器设置大猩猩/ MUX路由器点（例如，w.WriteHeader(404)或w.Header().Set("Content-Type", "application/json")） -此默默休息的任何“包装”处理程序试图设置或添加自己的头，像压缩处理程序。我看不到任何错误，除非我忘记在某处捕获错误，但浏览器收到无效响应。

有没有什么优雅的方法来处理这个问题，除了将标题藏在某处然后让最终处理程序编写它们之外？这似乎意味着重写处理程序的代码，如果可能的话，我很想避免这样做。

Answer 1

一旦您调用w.WriteHeader(404)，标题就会连接到电线上。所以你不能再添加它了。最好的方法是缓冲状态代码并将其写入链的末尾。例如，您可以提供自己的包装器，http.ResponseWriter以便重新实现WriteHeader()以保存状态值。然后添加方法Commit()来实际编写它。调用Commit()最后一个处理程序。当然，您必须以某种方式确定最后一个处理程序。

Answer 2

我经历了同样的默默失败的行为。但仅在我使用 WritheHeader 设置 StatusOK 以外的状态代码的处理程序中。我认为 CompressHandler 的这一部分出了问题：if h.Get("Content-Type") == "" {    h.Set("Content-Type", http.DetectContentType(b))}在我自己的处理程序中显式设置内容类型时，这似乎已解决：w.Header().Set("Content-Type", "text/html; charset=utf-8")w.WriteHeader(code)