mirror of
https://github.com/go-admin-team/go-admin.git
synced 2026-09-20 17:57:54 +00:00
127 lines
2.8 KiB
Go
127 lines
2.8 KiB
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/sirupsen/logrus"
|
|
"go-admin/models"
|
|
"go-admin/tools"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
//实例化
|
|
var logger = logrus.New()
|
|
|
|
// 日志记录到文件
|
|
func LoggerToFile() gin.HandlerFunc {
|
|
|
|
////写入文件
|
|
//src, err := os.OpenFile(config2.LogConfig.Dir+"/", os.O_APPEND|os.O_WRONLY, os.ModeAppend)
|
|
//if err != nil {
|
|
// fmt.Println("err", err)
|
|
//}
|
|
//
|
|
////设置输出
|
|
//logger.Out = src
|
|
//
|
|
////设置日志级别
|
|
//logger.SetLevel(logrus.DebugLevel)
|
|
//
|
|
////设置日志格式
|
|
//logger.SetFormatter(&logrus.TextFormatter{})
|
|
|
|
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()
|
|
|
|
// 日志格式
|
|
logger.Infof(" %3d %13v %15s %s %s \r\n",
|
|
statusCode,
|
|
latencyTime,
|
|
clientIP,
|
|
reqMethod,
|
|
reqUri,
|
|
)
|
|
logger.Infoln("-")
|
|
fmt.Println(
|
|
startTime.Format("\n2006-01-02 15:04:05.9999"),
|
|
"[INFO]",
|
|
reqMethod,
|
|
reqUri,
|
|
statusCode,
|
|
latencyTime,
|
|
reqUri,
|
|
clientIP,
|
|
)
|
|
if c.Request.Method != "GET" && c.Request.Method != "OPTIONS" {
|
|
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.GetCurrntTime()
|
|
sysOperLog.LatencyTime = (latencyTime).String()
|
|
sysOperLog.UserAgent = c.Request.UserAgent()
|
|
if c.Err() == nil {
|
|
sysOperLog.Status = "0"
|
|
} else {
|
|
sysOperLog.Status = "1"
|
|
}
|
|
_, _ = sysOperLog.Create()
|
|
}
|
|
}
|
|
}
|