179 lines
3.4 KiB
Go
179 lines
3.4 KiB
Go
package logger
|
||
|
||
import (
|
||
"fmt"
|
||
"io"
|
||
"os"
|
||
"path/filepath"
|
||
"sync"
|
||
|
||
"github.com/sirupsen/logrus"
|
||
)
|
||
|
||
var (
|
||
log *logrus.Logger
|
||
logMutex sync.Mutex
|
||
initialized bool
|
||
)
|
||
|
||
// InitLogger 初始化日志系统
|
||
func InitLogger(logFile, level string, maxSize, maxBackups, maxAge int, _ bool) error {
|
||
logMutex.Lock()
|
||
defer logMutex.Unlock()
|
||
|
||
if initialized {
|
||
return nil
|
||
}
|
||
|
||
log = logrus.New()
|
||
|
||
// 配置日志格式
|
||
log.SetFormatter(&logrus.TextFormatter{
|
||
FullTimestamp: true,
|
||
})
|
||
|
||
// 创建日志目录
|
||
if logFile != "" {
|
||
logDir := filepath.Dir(logFile)
|
||
if logDir != "." {
|
||
if err := os.MkdirAll(logDir, 0755); err != nil {
|
||
return fmt.Errorf("创建日志目录失败: %w", err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// 设置输出目标
|
||
outputTargets := []io.Writer{}
|
||
|
||
if logFile != "" {
|
||
// 使用标准库打开文件，支持追加写入
|
||
file, err := os.OpenFile(logFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
|
||
if err != nil {
|
||
// 无法打开日志文件时，回退到标准输出
|
||
log.Println("无法打开日志文件，将使用标准输出:", err)
|
||
} else {
|
||
outputTargets = append(outputTargets, file)
|
||
defer file.Sync() // 确保日志内容被写入磁盘
|
||
}
|
||
}
|
||
|
||
// 无论是否指定日志文件，都同时输出到标准输出
|
||
if len(outputTargets) > 0 {
|
||
outputTargets = append(outputTargets, os.Stdout)
|
||
} else {
|
||
// 如果没有指定日志文件，仅使用标准输出
|
||
outputTargets = append(outputTargets, os.Stdout)
|
||
}
|
||
|
||
// 设置日志输出
|
||
log.SetOutput(io.MultiWriter(outputTargets...))
|
||
|
||
// 设置日志级别
|
||
logLevel, err := logrus.ParseLevel(level)
|
||
if err != nil {
|
||
logLevel = logrus.InfoLevel
|
||
}
|
||
log.SetLevel(logLevel)
|
||
|
||
initialized = true
|
||
return nil
|
||
}
|
||
|
||
// Close 关闭日志系统
|
||
func Close() {
|
||
logMutex.Lock()
|
||
defer logMutex.Unlock()
|
||
|
||
if !initialized || log == nil {
|
||
return
|
||
}
|
||
|
||
// 执行日志刷新
|
||
log.Warn("日志系统已关闭")
|
||
|
||
// 确保日志被写入磁盘
|
||
if loggerOutput, ok := log.Out.(*os.File); ok {
|
||
loggerOutput.Sync()
|
||
}
|
||
|
||
initialized = false
|
||
log = nil
|
||
}
|
||
|
||
// Info 记录信息级别日志
|
||
func Info(msg string, fields ...interface{}) {
|
||
if !initialized {
|
||
return
|
||
}
|
||
|
||
if len(fields) > 0 {
|
||
log.WithFields(toFields(fields)).Info(msg)
|
||
} else {
|
||
log.Info(msg)
|
||
}
|
||
}
|
||
|
||
// Error 记录错误级别日志
|
||
func Error(msg string, fields ...interface{}) {
|
||
if !initialized {
|
||
return
|
||
}
|
||
|
||
if len(fields) > 0 {
|
||
log.WithFields(toFields(fields)).Error(msg)
|
||
} else {
|
||
log.Error(msg)
|
||
}
|
||
}
|
||
|
||
// Debug 记录调试级别日志
|
||
func Debug(msg string, fields ...interface{}) {
|
||
if !initialized {
|
||
return
|
||
}
|
||
|
||
if len(fields) > 0 {
|
||
log.WithFields(toFields(fields)).Debug(msg)
|
||
} else {
|
||
log.Debug(msg)
|
||
}
|
||
}
|
||
|
||
// Warn 记录警告级别日志
|
||
func Warn(msg string, fields ...interface{}) {
|
||
if !initialized {
|
||
return
|
||
}
|
||
|
||
if len(fields) > 0 {
|
||
log.WithFields(toFields(fields)).Warn(msg)
|
||
} else {
|
||
log.Warn(msg)
|
||
}
|
||
}
|
||
|
||
// Fatal 记录致命级别日志并退出程序
|
||
func Fatal(msg string, fields ...interface{}) {
|
||
if !initialized {
|
||
// 如果日志系统未初始化，使用标准库log
|
||
log.Fatal(msg)
|
||
}
|
||
|
||
if len(fields) > 0 {
|
||
log.WithFields(toFields(fields)).Fatal(msg)
|
||
} else {
|
||
log.Fatal(msg)
|
||
}
|
||
}
|
||
|
||
// toFields 将键值对转换为logrus字段
|
||
func toFields(keyValues []interface{}) logrus.Fields {
|
||
fields := make(logrus.Fields)
|
||
for i := 0; i < len(keyValues); i += 2 {
|
||
if i+1 < len(keyValues) {
|
||
fields[keyValues[i].(string)] = keyValues[i+1]
|
||
}
|
||
}
|
||
return fields
|
||
}
|