mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 08:20:25 +08:00
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package logger
|
||
|
||
import (
|
||
"fmt"
|
||
"mayfly-go/pkg/config"
|
||
"mayfly-go/pkg/global"
|
||
"os"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/sirupsen/logrus"
|
||
)
|
||
|
||
var Log = logrus.New()
|
||
|
||
func Init() {
|
||
Log.SetFormatter(new(LogFormatter))
|
||
Log.SetReportCaller(true)
|
||
|
||
logConf := config.Conf.Log
|
||
// 如果不存在日志配置信息，则默认debug级别
|
||
if logConf == nil {
|
||
Log.SetLevel(logrus.DebugLevel)
|
||
return
|
||
}
|
||
|
||
// 根据配置文件设置日志级别
|
||
if level := logConf.Level; level != "" {
|
||
l, err := logrus.ParseLevel(level)
|
||
if err != nil {
|
||
panic(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(fmt.Sprintf("创建日志文件失败: %s", err.Error()))
|
||
}
|
||
|
||
Log.Out = file
|
||
}
|
||
|
||
global.Log = 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()))
|
||
// 如果存在调用信息，记录方法信息及行号
|
||
if caller := entry.Caller; caller != nil {
|
||
logMsg = logMsg + fmt.Sprintf(" [%s:%d]", caller.Function, 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
|
||
}
|