mirror of
https://github.com/go-admin-team/go-admin.git
synced 2026-09-23 10:51:09 +00:00
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package middleware
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"go-admin/app/admin/models"
|
|
"go-admin/app/admin/models/system"
|
|
"go-admin/app/admin/service"
|
|
"go-admin/common/global"
|
|
"go-admin/common/log"
|
|
"go-admin/tools"
|
|
"go-admin/tools/config"
|
|
)
|
|
|
|
// LoggerToFile 日志记录到文件
|
|
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()
|
|
|
|
// 日志格式
|
|
logData := map[string]interface{}{
|
|
"statusCode": statusCode,
|
|
"latencyTime": latencyTime,
|
|
"clientIP": clientIP,
|
|
"method": reqMethod,
|
|
"uri": reqUri,
|
|
}
|
|
log.Info(logData)
|
|
global.RequestLogger.Info(logData)
|
|
|
|
if c.Request.Method != "GET" && c.Request.Method != "OPTIONS" && config.LoggerConfig.EnabledDB {
|
|
SetDBOperLog(c, clientIP, statusCode, reqUri, reqMethod, latencyTime)
|
|
}
|
|
}
|
|
}
|
|
|
|
// SetDBOperLog 写入操作日志表 fixme 该方法后续即将弃用
|
|
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()
|
|
sysOperaLog := system.SysOperaLog{}
|
|
sysOperaLog.OperIp = clientIP
|
|
sysOperaLog.OperLocation = tools.GetLocation(clientIP)
|
|
sysOperaLog.Status = tools.IntToString(statusCode)
|
|
sysOperaLog.OperName = tools.GetUserName(c)
|
|
sysOperaLog.RequestMethod = c.Request.Method
|
|
sysOperaLog.OperUrl = reqUri
|
|
if reqUri == "/login" {
|
|
sysOperaLog.BusinessType = "10"
|
|
sysOperaLog.Title = "用户登录"
|
|
sysOperaLog.OperName = "-"
|
|
} else if strings.Contains(reqUri, "/api/v1/logout") {
|
|
sysOperaLog.BusinessType = "11"
|
|
} else if strings.Contains(reqUri, "/api/v1/getCaptcha") {
|
|
sysOperaLog.BusinessType = "12"
|
|
sysOperaLog.Title = "验证码"
|
|
} else {
|
|
if reqMethod == "POST" {
|
|
sysOperaLog.BusinessType = "1"
|
|
} else if reqMethod == "PUT" {
|
|
sysOperaLog.BusinessType = "2"
|
|
} else if reqMethod == "DELETE" {
|
|
sysOperaLog.BusinessType = "3"
|
|
}
|
|
}
|
|
sysOperaLog.Method = reqMethod
|
|
if len(menuList) > 0 {
|
|
sysOperaLog.Title = menuList[0].Title
|
|
}
|
|
b, _ := c.Get("body")
|
|
sysOperaLog.OperParam, _ = tools.StructToJsonStr(b)
|
|
sysOperaLog.CreateBy = tools.GetUserIdUint(c)
|
|
sysOperaLog.OperTime = tools.GetCurrentTime()
|
|
sysOperaLog.LatencyTime = (latencyTime).String()
|
|
sysOperaLog.UserAgent = c.Request.UserAgent()
|
|
if c.Err() == nil {
|
|
sysOperaLog.Status = "0"
|
|
} else {
|
|
sysOperaLog.Status = "1"
|
|
}
|
|
msgID := tools.GenerateMsgIDFromContext(c)
|
|
db, err := tools.GetOrm(c)
|
|
if err != nil {
|
|
log.Errorf("msgID[%s] 获取Orm失败, error:%s", msgID, err)
|
|
}
|
|
serviceOperaLog := service.SysOperaLog{}
|
|
serviceOperaLog.Orm = db
|
|
_ = serviceOperaLog.InsertSysOperaLog(sysOperaLog.Generate())
|
|
}
|