记录 http.ResponseWriter 内容

Question

前提：我发现了类似的问题，但不适用于我的情况，所以请不要将其标记为重复。

我在 Go 中有一个 HTTP 服务器，我创建了一个中间件来记录请求、响应时间，我也想记录响应。

我已经在包下httputil.DumpRequest调用的函数中使用了。如何正确获取响应正文、状态和标头并将它们与其他数据一起记录？HTTPRequestlogw http.ResponseWriter

我的问题是：我想拦截响应标头、状态和正文，并将其与请求和响应时间一起记录

这是代码：

log "core/logger"
...
func RequestLoggerMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()

        defer func() {
            log.Info(
                fmt.Sprintf(
                    "[Request: %s] [Execution time: %v] [Response: %s]",
                    log.HTTPRequest(r),
                    time.Since(start),
                    // RESPONSE DATA HERE !!!!!!!
                ))
        }()
        next.ServeHTTP(w, r)
    })
}

Answer 1

谢谢@Sivachandran的回复。它几乎是完美的，只是它没有实现http.ResponseWriter因为指针。

为了完整起见，我在这里发布了正确的解决方案代码，因为即使这个问题得到了负分，也不容易找到任何有关它的文档。

Stackoverflow是一个交流问题的好地方，在我看来，这是一个非常好又难的问题，无论对于中等水平的 Golang 程序员来说，所以它根本不值得负分！

这就是解决方案，享受：

// RequestLoggerMiddleware is the middleware layer to log all the HTTP requests
func RequestLoggerMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()
        rww := NewResponseWriterWrapper(w)
        w.Header()
        defer func() {
            log.Info(
                fmt.Sprintf(
                    "[Request: %s] [Execution time: %v] [Response: %s]",
                    log.HTTPRequest(r),
                    time.Since(start),
                    rww.String(),
                ))
        }()
        next.ServeHTTP(rww, r)
    })
}

// ResponseWriterWrapper struct is used to log the response
type ResponseWriterWrapper struct {
    w          *http.ResponseWriter
    body       *bytes.Buffer
    statusCode *int
}

// NewResponseWriterWrapper static function creates a wrapper for the http.ResponseWriter
func NewResponseWriterWrapper(w http.ResponseWriter) ResponseWriterWrapper {
    var buf bytes.Buffer
    var statusCode int = 200
    return ResponseWriterWrapper{
        w:          &w,
        body:       &buf,
        statusCode: &statusCode,
    }
}

func (rww ResponseWriterWrapper) Write(buf []byte) (int, error) {
    rww.body.Write(buf)
    return (*rww.w).Write(buf)
}

// Header function overwrites the http.ResponseWriter Header() function
func (rww ResponseWriterWrapper) Header() http.Header {
    return (*rww.w).Header()

}

// WriteHeader function overwrites the http.ResponseWriter WriteHeader() function
func (rww ResponseWriterWrapper) WriteHeader(statusCode int) {
    (*rww.statusCode) = statusCode
    (*rww.w).WriteHeader(statusCode)
}

func (rww ResponseWriterWrapper) String() string {
    var buf bytes.Buffer

    buf.WriteString("Response:")

    buf.WriteString("Headers:")
    for k, v := range (*rww.w).Header() {
        buf.WriteString(fmt.Sprintf("%s: %v", k, v))
    }

    buf.WriteString(fmt.Sprintf(" Status Code: %d", *(rww.statusCode)))

    buf.WriteString("Body")
    buf.WriteString(rww.body.String())
    return buf.String()
}