118 lines
2.8 KiB
Go
118 lines
2.8 KiB
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"go-admin/logger"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"go-admin/app/admin/models"
|
|
"go-admin/tools"
|
|
config2 "go-admin/tools/config"
|
|
)
|
|
|
|
// 日志记录到文件
|
|
func LoggerToFile() gin.HandlerFunc {
|
|
|
|
return func(c *gin.Context) {
|
|
// 开始时间
|
|
startTime := time.Now()
|
|
|
|
// 处理请求
|
|
c.Next()
|
|
|
|
// 结束时间
|
|
endTime := time.Now()
|
|
|
|
// 执行时间
|
|
latencyTime := endTime.Sub(startTime)
|
|
|
|
// 请求方式
|
|
reqMethod := c.Request.Method
|
|
|
|
// 请求路由
|
|
reqUri := c.Request.RequestURI
|
|
|
|
// 状态码
|
|
statusCode := c.Writer.Status()
|
|
|
|
// 请求IP
|
|
clientIP := c.ClientIP()
|
|
|
|
// 日志格式
|
|
fmt.Printf("%s [INFO] %s %s %3d %13v %15s \r\n",
|
|
startTime.Format("2006-01-02 15:04:05"),
|
|
reqMethod,
|
|
reqUri,
|
|
statusCode,
|
|
latencyTime,
|
|
clientIP,
|
|
)
|
|
|
|
logger.Info(
|
|
"log request",
|
|
logger.Field("status", statusCode),
|
|
logger.Field("spend_time", latencyTime),
|
|
logger.Field("client_id", clientIP),
|
|
logger.Field("method", reqMethod),
|
|
logger.Field("uri", reqUri),
|
|
)
|
|
|
|
if c.Request.Method != "GET" && c.Request.Method != "OPTIONS" && config2.LoggerConfig.EnabledDB {
|
|
SetDBOperLog(c, clientIP, statusCode, reqUri, reqMethod, latencyTime)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 写入操作日志表
|
|
// 该方法后续即将弃用
|
|
func SetDBOperLog(c *gin.Context, clientIP string, statusCode int, reqUri string, reqMethod string, latencyTime time.Duration) {
|
|
menu := models.Menu{}
|
|
menu.Path = reqUri
|
|
menu.Action = reqMethod
|
|
menuList, _ := menu.Get()
|
|
sysOperLog := models.SysOperLog{}
|
|
sysOperLog.OperIp = clientIP
|
|
sysOperLog.OperLocation = tools.GetLocation(clientIP)
|
|
sysOperLog.Status = tools.IntToString(statusCode)
|
|
sysOperLog.OperName = tools.GetUserName(c)
|
|
sysOperLog.RequestMethod = c.Request.Method
|
|
sysOperLog.OperUrl = reqUri
|
|
if reqUri == "/login" {
|
|
sysOperLog.BusinessType = "10"
|
|
sysOperLog.Title = "用户登录"
|
|
sysOperLog.OperName = "-"
|
|
} else if strings.Contains(reqUri, "/api/v1/logout") {
|
|
sysOperLog.BusinessType = "11"
|
|
} else if strings.Contains(reqUri, "/api/v1/getCaptcha") {
|
|
sysOperLog.BusinessType = "12"
|
|
sysOperLog.Title = "验证码"
|
|
} else {
|
|
if reqMethod == "POST" {
|
|
sysOperLog.BusinessType = "1"
|
|
} else if reqMethod == "PUT" {
|
|
sysOperLog.BusinessType = "2"
|
|
} else if reqMethod == "DELETE" {
|
|
sysOperLog.BusinessType = "3"
|
|
}
|
|
}
|
|
sysOperLog.Method = reqMethod
|
|
if len(menuList) > 0 {
|
|
sysOperLog.Title = menuList[0].Title
|
|
}
|
|
b, _ := c.Get("body")
|
|
sysOperLog.OperParam, _ = tools.StructToJsonStr(b)
|
|
sysOperLog.CreateBy = tools.GetUserName(c)
|
|
sysOperLog.OperTime = tools.GetCurrentTime()
|
|
sysOperLog.LatencyTime = (latencyTime).String()
|
|
sysOperLog.UserAgent = c.Request.UserAgent()
|
|
if c.Err() == nil {
|
|
sysOperLog.Status = "0"
|
|
} else {
|
|
sysOperLog.Status = "1"
|
|
}
|
|
_, _ = sysOperLog.Create()
|
|
}
|