mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-03 16:00:25 +08:00
47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package logger
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/sirupsen/logrus"
|
||
)
|
||
|
||
var Log = logrus.New()
|
||
|
||
func init() {
|
||
// customFormatter := new(logrus.TextFormatter)
|
||
// customFormatter.TimestampFormat = "2006-01-02 15:04:05.000"
|
||
// customFormatter.FullTimestamp = true
|
||
Log.SetFormatter(new(LogFormatter))
|
||
Log.SetReportCaller(true)
|
||
Log.SetLevel(logrus.DebugLevel)
|
||
}
|
||
|
||
type LogFormatter struct{}
|
||
|
||
func (l *LogFormatter) Format(entry *logrus.Entry) ([]byte, error) {
|
||
timestamp := time.Now().Local().Format("2006-01-02 15:04:05.000")
|
||
level := entry.Level
|
||
logMsg := fmt.Sprintf("%s [%s]", timestamp, strings.ToUpper(level.String()))
|
||
// 如果存在调用信息，且为error级别以上记录文件及行号
|
||
if caller := entry.Caller; caller != nil {
|
||
var fp string
|
||
// 全路径切割，只获取项目相关路径，
|
||
// 即/Users/hml/Desktop/project/go/mayfly-go/server/test.go只获取/server/test.go
|
||
ps := strings.Split(caller.File, "mayfly-go/")
|
||
if len(ps) >= 2 {
|
||
fp = ps[1]
|
||
} else {
|
||
fp = ps[0]
|
||
}
|
||
logMsg = logMsg + fmt.Sprintf(" [%s:%d]", fp, caller.Line)
|
||
}
|
||
for k, v := range entry.Data {
|
||
logMsg = logMsg + fmt.Sprintf(" [%s=%v]", k, v)
|
||
}
|
||
logMsg = logMsg + fmt.Sprintf(" : %s\n", entry.Message)
|
||
return []byte(logMsg), nil
|
||
}
|