parent
acec082490
commit
42b406e0c8
@ -0,0 +1,45 @@
|
|||||||
|
package middleware
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/taoshihan1991/imaptool/tools"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewMidLogger() gin.HandlerFunc {
|
||||||
|
logger := tools.Logger()
|
||||||
|
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 |",
|
||||||
|
statusCode,
|
||||||
|
latencyTime,
|
||||||
|
clientIP,
|
||||||
|
reqMethod,
|
||||||
|
reqUri,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
@ -1,87 +1,102 @@
|
|||||||
package tools
|
package tools
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"github.com/gin-gonic/gin"
|
|
||||||
"github.com/sirupsen/logrus"
|
"github.com/sirupsen/logrus"
|
||||||
|
"log"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var logrusObj *logrus.Logger
|
||||||
|
|
||||||
func Logger() *logrus.Logger {
|
func Logger() *logrus.Logger {
|
||||||
|
if logrusObj != nil {
|
||||||
|
src, _ := setOutputFile()
|
||||||
|
//设置输出
|
||||||
|
logrusObj.Out = src
|
||||||
|
return logrusObj
|
||||||
|
}
|
||||||
|
|
||||||
|
//实例化
|
||||||
|
logger := logrus.New()
|
||||||
|
src, _ := setOutputFile()
|
||||||
|
//设置输出
|
||||||
|
logger.Out = src
|
||||||
|
//设置日志级别
|
||||||
|
logger.SetLevel(logrus.DebugLevel)
|
||||||
|
//设置日志格式
|
||||||
|
logger.SetFormatter(&logrus.TextFormatter{
|
||||||
|
TimestampFormat: "2006-01-02 15:04:05",
|
||||||
|
})
|
||||||
|
logrusObj = logger
|
||||||
|
return logger
|
||||||
|
}
|
||||||
|
func setOutputFile() (*os.File, error) {
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
logFilePath := ""
|
logFilePath := ""
|
||||||
if dir, err := os.Getwd(); err == nil {
|
if dir, err := os.Getwd(); err == nil {
|
||||||
logFilePath = dir + "/logs/"
|
logFilePath = dir + "/logs/"
|
||||||
}
|
}
|
||||||
if err := os.MkdirAll(logFilePath, 0777); err != nil {
|
_, err := os.Stat(logFilePath)
|
||||||
fmt.Println(err.Error())
|
if os.IsNotExist(err) {
|
||||||
|
if err := os.MkdirAll(logFilePath, 0777); err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
}
|
}
|
||||||
logFileName := now.Format("2006-01-02") + ".log"
|
logFileName := now.Format("2006-01-02") + ".log"
|
||||||
//日志文件
|
//日志文件
|
||||||
fileName := path.Join(logFilePath, logFileName)
|
fileName := path.Join(logFilePath, logFileName)
|
||||||
if _, err := os.Stat(fileName); err != nil {
|
if _, err := os.Stat(fileName); err != nil {
|
||||||
if _, err := os.Create(fileName); err != nil {
|
if _, err := os.Create(fileName); err != nil {
|
||||||
fmt.Println(err.Error())
|
log.Println(err.Error())
|
||||||
|
return nil, err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
//写入文件
|
//写入文件
|
||||||
src, err := os.OpenFile(fileName, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
|
src, err := os.OpenFile(fileName, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println("err", err)
|
log.Println(err)
|
||||||
|
return nil, err
|
||||||
}
|
}
|
||||||
|
return src, nil
|
||||||
//实例化
|
|
||||||
logger := logrus.New()
|
|
||||||
|
|
||||||
//设置输出
|
|
||||||
logger.Out = src
|
|
||||||
|
|
||||||
//设置日志级别
|
|
||||||
logger.SetLevel(logrus.DebugLevel)
|
|
||||||
|
|
||||||
//设置日志格式
|
|
||||||
logger.SetFormatter(&logrus.TextFormatter{
|
|
||||||
TimestampFormat: "2006-01-02 15:04:05",
|
|
||||||
})
|
|
||||||
return logger
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func LoggerToFile() gin.HandlerFunc {
|
//func LoggerToFile() gin.HandlerFunc {
|
||||||
logger := Logger()
|
// logger := Logger()
|
||||||
return func(c *gin.Context) {
|
// return func(c *gin.Context) {
|
||||||
// 开始时间
|
// // 开始时间
|
||||||
startTime := time.Now()
|
// startTime := time.Now()
|
||||||
|
//
|
||||||
// 处理请求
|
// // 处理请求
|
||||||
c.Next()
|
// c.Next()
|
||||||
|
//
|
||||||
// 结束时间
|
// // 结束时间
|
||||||
endTime := time.Now()
|
// endTime := time.Now()
|
||||||
|
//
|
||||||
// 执行时间
|
// // 执行时间
|
||||||
latencyTime := endTime.Sub(startTime)
|
// latencyTime := endTime.Sub(startTime)
|
||||||
|
//
|
||||||
// 请求方式
|
// // 请求方式
|
||||||
reqMethod := c.Request.Method
|
// reqMethod := c.Request.Method
|
||||||
|
//
|
||||||
// 请求路由
|
// // 请求路由
|
||||||
reqUri := c.Request.RequestURI
|
// reqUri := c.Request.RequestURI
|
||||||
|
//
|
||||||
// 状态码
|
// // 状态码
|
||||||
statusCode := c.Writer.Status()
|
// statusCode := c.Writer.Status()
|
||||||
|
//
|
||||||
// 请求IP
|
// // 请求IP
|
||||||
clientIP := c.ClientIP()
|
// clientIP := c.ClientIP()
|
||||||
|
//
|
||||||
//日志格式
|
// //日志格式
|
||||||
logger.Infof("| %3d | %13v | %15s | %s | %s |",
|
// logger.Infof("| %3d | %13v | %15s | %s | %s |",
|
||||||
statusCode,
|
// statusCode,
|
||||||
latencyTime,
|
// latencyTime,
|
||||||
clientIP,
|
// clientIP,
|
||||||
reqMethod,
|
// reqMethod,
|
||||||
reqUri,
|
// reqUri,
|
||||||
)
|
// )
|
||||||
}
|
// }
|
||||||
}
|
//}
|
||||||
|
Loading…
Reference in new issue