在go中,如何检查写入http.ResponseWriter的http响应?

时间:2015-01-16 12:23:09

标签: http go

可能有一些显而易见的事情我不知道但是我正在尝试调试我的go服务器写的HTTP响应。

我看到有httputil.DumpResponse可用,但它需要一个http.Response对象,而我可用的是http.ResponseWriter

有没有办法从http.ResponseWriter中提取http.Response,以便我可以检查对控制台或日志的响应内容?

上下文:

我正在使用https://github.com/RangelReale/osin编写简单的服务器端身份验证,这是默认示例,但无法理解为什么前端(使用http://ember-simple-auth.com)将身份验证失败(密码错误)解释为成功。

以下是片段:

r = mux.NewRouter()
r.HandleFunc("/token", func (w http.ResponseWriter, r *http.Request) {

    fmt.Printf("r.HandleFunc /token\n")

    resp := server.NewResponse()
    defer resp.Close()

    r.ParseForm()

    grantType := r.FormValue("grant_type")
    username := r.FormValue("username")
    password := r.FormValue("password")
    fmt.Printf("/token : grantType=%s  username=%s  password=%s\n", grantType, username, password)

    if ar := server.HandleAccessRequest(resp, r); ar != nil {
        if username == "user" && password == "correct-password" {
            ar.Authorized = true
        } else {
            ar.Authorized = false
        }
        server.FinishAccessRequest(resp, r, ar)
    }

    osin.OutputJSON(resp, w, r)

    // Debug - doesn't work yet
    dump, err := httputil.DumpResponse(w, true)
    if err != nil {
        fmt.Printf("%s\n", dump)
    }
});
http.Handle("/token", r)

1 个答案:

答案 0 :(得分:6)

写一个*httptest.ResponseRecorder(实现http.ResponseWriter)并检查它。

包中的示例:

package main

import (
    "fmt"
    "log"
    "net/http"
    "net/http/httptest"
)

func main() {
    handler := func(w http.ResponseWriter, r *http.Request) {
        http.Error(w, "something failed", http.StatusInternalServerError)
    }

    req, err := http.NewRequest("GET", "http://example.com/foo", nil)
    if err != nil {
        log.Fatal(err)
    }

    w := httptest.NewRecorder()
    handler(w, req)

    fmt.Printf("%d - %s", w.Code, w.Body.String())
}

编辑以在评论中回答问题:

如果我理解你的问题,那么是的,你可以使用闭包。

请考虑以下内容作为您的处理程序:

func MyHandler(w http.ResponseWriter, r *http.Request) {
    // do useful stuff...
}

然后,您可以使用servemux注册以下闭包以获得所需的效果:

http.HandleFunc("/my/url", func(w http.ResponseWriter, r *http.Request) {
    // first call MyHandler
    MyHandler(w, r)

    // then log whatever you need
    log.Printf("%#v\n", w)
})

如果这个模式证明对你有用,那么你可以编写一个更高阶的方法来包装任何func(http.ResponseWriter, *http.Request)这样的闭包。不过,这本身就是一个话题。

相关问题