实现缓存功能

logger/logger.go

@@ -1,8 +1,10 @@
 package logger
 
 import (
+	"fmt"
 	"io"
 	"os"
+	"path/filepath"
 	"sync"
 
 	"github.com/sirupsen/logrus"
@@ -15,7 +17,7 @@ var (
 )
 
 // InitLogger 初始化日志系统
-func InitLogger(logFile, level string, maxSize, maxBackups, maxAge int) error {
+func InitLogger(logFile, level string, maxSize, maxBackups, maxAge int, _ bool) error {
 	logMutex.Lock()
 	defer logMutex.Unlock()
 
@@ -30,21 +32,42 @@ func InitLogger(logFile, level string, maxSize, maxBackups, maxAge int) error {
 		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 {
-			logrus.Warn("无法打开日志文件，将使用标准输出", "error", err)
-			log.SetOutput(os.Stdout)
+			// 无法打开日志文件时，回退到标准输出
+			log.Println("无法打开日志文件，将使用标准输出:", err)
 		} else {
-			// 同时输出到文件和标准输出
-			log.SetOutput(io.MultiWriter(file, os.Stdout))
+			outputTargets = append(outputTargets, file)
+			defer file.Sync() // 确保日志内容被写入磁盘
 		}
-	} else {
-		log.SetOutput(os.Stdout)
 	}
 
+	// 无论是否指定日志文件，都同时输出到标准输出
+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 {
@@ -61,13 +84,20 @@ func Close() {
 	logMutex.Lock()
 	defer logMutex.Unlock()
 
-	if !initialized {
+	if !initialized || log == nil {
 		return
 	}
 
 	// 执行日志刷新
 	log.Warn("日志系统已关闭")
+	
+	// 确保日志被写入磁盘
+	if loggerOutput, ok := log.Out.(*os.File); ok {
+		loggerOutput.Sync()
+	}
+	
 	initialized = false
+	log = nil
 }
 
 // Info 记录信息级别日志
@@ -122,6 +152,20 @@ func Warn(msg string, fields ...interface{}) {
 	}
 }
 
+// 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)