Huangzj
52 lines
995 B
Go
52 lines
995 B
Go
/*
|
||
* @Author : huangzj
|
||
* @Time : 2020/12/8 14:45
|
||
* @Description：
|
||
*/
|
||
|
||
package gin
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/gin-gonic/gin"
|
||
"log"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func Logger() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
t := time.Now()
|
||
|
||
// 请求前
|
||
fmt.Println(fmt.Sprintf("记录日志，本次请求的参数为：%v", c.Params))
|
||
|
||
c.Next()
|
||
|
||
// 请求后
|
||
latency := time.Since(t)
|
||
log.Print(latency)
|
||
|
||
// 获取发送的 status
|
||
status := c.Writer.Status()
|
||
log.Println(status)
|
||
}
|
||
}
|
||
|
||
//在请求前后者请求后自定义中间件处理
|
||
func TestMiddleWare(t *testing.T) {
|
||
router := gin.Default() //Default会返回两个中间件 Logger(), Recovery().这接口调用之前和之后进行处理，所以我们也可以自定义中间件
|
||
|
||
router.Use(Logger())
|
||
|
||
router.GET("/test", func(c *gin.Context) {
|
||
example := c.MustGet("example").(string)
|
||
|
||
// 打印："12345"
|
||
log.Println(example)
|
||
})
|
||
|
||
// 监听并在 0.0.0.0:8080 上启动服务
|
||
router.Run(":8080")
|
||
}
|