在 http.HandleFunc 中记录对传入 HTTP 请求的响应

Question

这是Go中的后续问题，如何检查写入http.ResponseWriter的http响应？因为那里的解决方案需要伪造请求，这对单元测试很有用，但不适用于实时服务器。我想将我的 Web 服务响应它从用户收到的请求而返回的 HTTP 响应转储到日志文件（或控制台）中。输出应该告诉我标头是什么和 JSON 有效负载。怎么做呢？如果有一个 httputil.DumpResponse 等价物将 http.ResponseWriter 作为参数而不是 http.Response 就完美了，但目前我只能从 http.ResponseWriter 访问 Headerr = mux.NewRouter()r.HandleFunc("/path", func (w http.ResponseWriter, r *http.Request) {    fmt.Printf("r.HandleFunc /path\n")    resp := server.NewResponse()    defer resp.Close()    r.ParseForm()    // Server does some work here    // ...    // Insert debug code here, something like    //    // dump = http.DumpResponseFromWriter(w)    // fmt.Printf("%s\n", dump)});http.Handle("/path", r)

Answer 1

中间件链

这个问题的一个常见解决方案是所谓的中间件链。有几个库提供此功能，例如negroni。

这是一种延续传递风格的形式，您可以像这样编写中间件函数（取自 negroni 的自述文件）：

func MyMiddleware(rw http.ResponseWriter, r *http.Request, next http.HandlerFunc) {

  // do some stuff before

  next(rw, r)

  // do some stuff after

}

然后 negroni 为您提供一个 HTTP 处理程序，以正确的顺序调用您的中间件。

我们可以以稍微不同的方式实现这个解决方案，而不是一种不太神奇但功能更多的方法（如在函数式编程中）。定义处理程序组合子如下：

func NewFooHandler(next http.HandlerFunc) http.HandlerFunc {

    return func(w http.ResponseWriter, r *http.Request) {

        // do some stuff before

        next(r,w)

        // do some stuff after

    }

}

然后将您的链定义为一个组合：

h := NewFooHandler(NewBarHandler(NewBazHandler(Sink)))

现在h是http.HandlerFuncfoo，然后是 bar，然后是 baz。Sink只是一个空的最后一个处理程序，什么都不做（“完成”链。）

将此解决方案应用于您的问题

定义一个处理程序组合子：

func NewResponseLoggingHandler(next http.HandlerFunc) http.HandlerFunc {

    return func(w http.ResponseWriter, r *http.Request) {

        // switch out response writer for a recorder

        // for all subsequent handlers

        c := httptest.NewRecorder()

        next(c, r)

        // copy everything from response recorder

        // to actual response writer

        for k, v := range c.HeaderMap {

            w.Header()[k] = v

        }

        w.WriteHeader(c.Code)

        c.Body.WriteTo(w)

    }

}

现在问题归结为处理程序管理。您可能希望将此处理程序应用于某个类别中的所有链。为此，您可以再次使用组合器（这在某种程度上相当于 negroni 的Classic()方法）：

func NewDefaultHandler(next http.HandlerFunc) http.HandlerFunc {

    return NewResponseLoggingHandler(NewOtherStuffHandler(next))

}

在此之后，每当您启动这样的链时：

h := NewDefaultHandler(...)

它将自动包含响应日志记录和您在NewDefaultHandler.