60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package middlewares
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/sirupsen/logrus"
|
|
"hospital-open-api/global"
|
|
"time"
|
|
)
|
|
|
|
// Logrus 日志中间件
|
|
func Logrus() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
// 开始时间
|
|
startTime := time.Now()
|
|
|
|
// 处理请求
|
|
c.Next()
|
|
|
|
// 获取 请求 参数
|
|
params := make(map[string]string)
|
|
|
|
paramsRaw, ok := c.Get("params")
|
|
if ok {
|
|
requestParams, ok := paramsRaw.(map[string]string)
|
|
if ok || len(requestParams) > 0 {
|
|
params = requestParams
|
|
}
|
|
}
|
|
|
|
// 结束时间
|
|
endTime := time.Now()
|
|
|
|
// 执行时间
|
|
latencyTime := fmt.Sprintf("%6v", endTime.Sub(startTime))
|
|
|
|
// 请求方式
|
|
reqMethod := c.Request.Method
|
|
|
|
// 请求路由
|
|
reqUri := c.Request.RequestURI
|
|
|
|
// 状态码
|
|
statusCode := c.Writer.Status()
|
|
|
|
// 请求IP
|
|
clientIP := c.ClientIP()
|
|
|
|
// 日志格式
|
|
global.Logger.WithFields(logrus.Fields{
|
|
"http_status": statusCode,
|
|
"total_time": latencyTime,
|
|
"ip": clientIP,
|
|
"method": reqMethod,
|
|
"uri": reqUri,
|
|
"params": params,
|
|
}).Info("access")
|
|
}
|
|
}
|