gin拦截response的数据

package main

import (
    "bytes"
    "fmt"
    "github.com/gin-gonic/gin"
"net/http"
)

type bodyLogWriter struct {
    gin.ResponseWriter
    body *bytes.Buffer
}

func (w bodyLogWriter) Write(b []byte) (int, error) {
    w.body.Write(b)
    return w.ResponseWriter.Write(b)
}

func ginBodyLogMiddleware(c *gin.Context) {
    blw := &bodyLogWriter{body: bytes.NewBufferString(""), ResponseWriter: c.Writer}
    c.Writer = blw
    c.Next()
    statusCode := c.Writer.Status()
    fmt.Println(statusCode)
    //if statusCode >= 400 {
        //ok this is an request with error, let's make a record for it
        // now print body (or log in your preferred way)
        fmt.Println("Response body: " + blw.body.String())
    //}
}
func main() {
    r := gin.Default()
    r.Use(ginBodyLogMiddleware)

    r.GET("/ping", func(c *gin.Context) {
        c.String(http.StatusOK, "PONG")
    })

    r.GET("/status", func(c *gin.Context) {
        c.Status(http.StatusOK)
    })

    r.GET("/hello", func(c *gin.Context) {
        c.JSON(http.StatusOK,gin.H{"afa":232})
    })

    r.Run()
}