如何从 http.ResponseWriter 获取数据以进行日志记录

“纵有疾风来，人生不言弃”，这句话送给正在学习Golang的朋友们，也希望在阅读本文《如何从 http.ResponseWriter 获取数据以进行日志记录》后，能够真的帮助到大家。我也会在后续的文章中，陆续更新Golang相关的技术文章，有好的建议欢迎大家在评论留言，非常感谢！

问题内容

我不知道这个问题是否有意义，但我想知道是否有任何方法可以获取写入http.ResponseWriter. 我需要它来记录。

我在 Go 中编写了一个 API。

func api1(w http.ResponseWriter, req *http.Request) {       
    var requestData MyStruct
    err := json.NewDecoder(req.Body).Decode(&requestData)
    if err != nil {
        writeError(w, "JSON request is not in correct format")
        return
    }
    log.Println(" Request Data :", req.Body) // I am logging req
    result, err := compute()                 // getting result from a function

    if err != nil {
        errRes := ErrorResponse{"ERROR", err}
        response, er = json.Marshal(errRes) // getting error response
    } else {
        response, er = json.Marshal(result)
    }
    if er != nil {
        http.Error(w, er.Error(), 500) // writing error
        return
    }
    io.WriteString(w, string(response)) // writing response
}

目的是创建一个包含请求和响应数据的日志。响应可以是错误响应或已处理响应。

我在想如果我能得到写在 http.ResponseWriter 上的数据，我就可以创建一个有意义的日志。

这可能吗？如果没有，请建议我如何实现这一目标。

正确答案

您可以使用io.MultiWriter - 它创建一个写入器，将其写入复制到所有提供的写入器。所以要记录响应

func api1(w http.ResponseWriter, req *http.Request) {
    var log bytes.Buffer
    rsp := io.MultiWriter(w, &log)
    // from this point on use rsp instead of w, ie
    err := json.NewDecoder(req.Body).Decode(&requestData)
    if err != nil {
        writeError(rsp, "JSON request is not in correct format")
        return
    }
    ...
}

rsp现在，您在两者中都写入了信息的副本，w并且log您可以将log缓冲区的内容保存到光盘上，以便在控制台等上显示它​​。

您可以使用io.TeeReader创建一个读取器，将它从给定读取器读取的内容写入给定 Writer - 这将允许您将副本保存req.Body到日志中，即

func api1(w http.ResponseWriter, req *http.Request) {
    var log bytes.Buffer
    tee := io.TeeReader(req.Body, &log)
    err := json.NewDecoder(tee).Decode(&requestData)
    ...
}

现在，由于 json 解码器从表单中读取tee，因此也将其内容req.Body复制到log缓冲区中。

文中关于golang的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何从 http.ResponseWriter 获取数据以进行日志记录》文章吧，也可关注golang学习网公众号了解相关技术文章。