mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-23 02:48:34 +08:00
72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
package logger
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"pandax/base/config"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/sirupsen/logrus"
|
||
)
|
||
|
||
func InitLog(logConf *config.Log) *logrus.Logger {
|
||
var Log = logrus.New()
|
||
Log.SetFormatter(new(LogFormatter))
|
||
Log.SetReportCaller(true)
|
||
|
||
// 如果不存在日志配置信息，则默认debug级别
|
||
if logConf == nil {
|
||
Log.SetLevel(logrus.DebugLevel)
|
||
return nil
|
||
}
|
||
|
||
// 根据配置文件设置日志级别
|
||
if level := logConf.Level; level != "" {
|
||
l, err := logrus.ParseLevel(level)
|
||
if err != nil {
|
||
panic(any(fmt.Sprintf("日志级别不存在: %s", level)))
|
||
}
|
||
Log.SetLevel(l)
|
||
} else {
|
||
Log.SetLevel(logrus.DebugLevel)
|
||
}
|
||
|
||
if logFile := logConf.File; logFile != nil {
|
||
//写入文件
|
||
file, err := os.OpenFile(logFile.GetFilename(), os.O_CREATE|os.O_APPEND|os.O_WRONLY, os.ModeAppend|0666)
|
||
if err != nil {
|
||
panic(any(fmt.Sprintf("创建日志文件失败: %s", err.Error())))
|
||
}
|
||
|
||
Log.Out = file
|
||
}
|
||
return Log
|
||
}
|
||
|
||
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/pandax/test.go只获取/test.go
|
||
ps := strings.Split(caller.File, "pandax/")
|
||
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
|
||
}
|