main.go
package main
import (
"demo/gindemo/middleware"
"github.com/gin-gonic/gin"
"net/http"
)
func main() {
// 1.创建路由
r := gin.Default()
r.Use(middleware.Logger())
r.GET("/", func(c *gin.Context) {
c.String(http.StatusOK, "hello World!")
})
r.POST("/test", func(c *gin.Context) {
c.JSON(http.StatusOK, "hello World!")
})
r.Run(":8000")
}
logger.go
package middleware
import (
"bytes"
"encoding/json"
"fmt"
"github.com/gin-gonic/gin"
"io/ioutil"
"time"
)
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 (w BodyLogWriter) WriteString(s string) (int, error) {
w.body.WriteString(s)
return w.ResponseWriter.WriteString(s)
}
// 打印日志
func Logger() gin.HandlerFunc {
r