feat(logger): add size-based log rotation support

Author: 0xJacky

app.example.ini

@@ -27,6 +27,7 @@ DB = 0
 Prefix = cosy
 
 [log]
+EnableRotate = true
 EnableFileLog = false
 Dir = logs
 MaxSize = 10

docs/logger/start.md

@@ -89,10 +89,11 @@ func initDB() *gorm.DB {
 
 ## 日志文件配置
 
-Logger 支持日志文件轮转功能，可通过以下配置启用：
+Logger 支持日志文件轮转功能，可通过以下配置启用或关闭：
 
 ```ini
 [log]
+EnableRotate = true
 EnableFileLog = true
 Dir = logs
 MaxSize = 100
@@ -103,6 +104,7 @@ Compress = true
 
 配置参数说明：
 
+- `EnableRotate`: 是否启用按大小轮转，默认 true；关闭后日志文件仅追加不会分片
 - `EnableFileLog`: 是否启用文件日志（默认：false）
 - `Dir`: 日志文件存储目录
 - `MaxSize`: 单个日志文件最大体积，单位 MB（默认：100 MB）

logger/file.go

@@ -1,10 +1,15 @@
 package logger
 
 import (
+	"fmt"
+	"os"
+	"path/filepath"
 	"time"
 
 	"github.com/gin-gonic/gin"
+	"github.com/uozi-tech/cosy/settings"
 	"go.uber.org/zap/zapcore"
+	"gopkg.in/natefinch/lumberjack.v2"
 )
 
 func GetFileEncoder(mode string) zapcore.Encoder {
@@ -31,3 +36,66 @@ func GetFileEncoder(mode string) zapcore.Encoder {
 
 	return zapcore.NewJSONEncoder(encoderConfig)
 }
+
+// NewFileCores builds zap cores with size-based log rotation.
+// Error and above go to error.log; lower levels go to info.log.
+func NewFileCores(mode string, highPriority, lowPriority zapcore.LevelEnabler) ([]zapcore.Core, error) {
+	if err := os.MkdirAll(settings.LogSettings.Dir, 0755); err != nil {
+		return nil, fmt.Errorf("create log directory: %w", err)
+	}
+
+	encoder := GetFileEncoder(mode)
+
+	var (
+		errorSyncer zapcore.WriteSyncer
+		infoSyncer  zapcore.WriteSyncer
+		err         error
+	)
+
+	if settings.LogSettings.EnableRotate {
+		errorSyncer = newRotateWriteSyncer(filepath.Join(settings.LogSettings.Dir, "error.log"))
+		infoSyncer = newRotateWriteSyncer(filepath.Join(settings.LogSettings.Dir, "info.log"))
+	} else {
+		errorSyncer, err = newPlainWriteSyncer(filepath.Join(settings.LogSettings.Dir, "error.log"))
+		if err != nil {
+			return nil, fmt.Errorf("open error log file: %w", err)
+		}
+		infoSyncer, err = newPlainWriteSyncer(filepath.Join(settings.LogSettings.Dir, "info.log"))
+		if err != nil {
+			return nil, fmt.Errorf("open info log file: %w", err)
+		}
+	}
+
+	errorCore := zapcore.NewCore(
+		encoder,
+		errorSyncer,
+		highPriority,
+	)
+
+	infoCore := zapcore.NewCore(
+		encoder,
+		infoSyncer,
+		lowPriority,
+	)
+
+	return []zapcore.Core{errorCore, infoCore}, nil
+}
+
+func newRotateWriteSyncer(filename string) zapcore.WriteSyncer {
+	return zapcore.AddSync(&lumberjack.Logger{
+		Filename:   filename,
+		MaxSize:    settings.LogSettings.MaxSize,
+		MaxBackups: settings.LogSettings.MaxBackups,
+		MaxAge:     settings.LogSettings.MaxAge,
+		LocalTime:  true,
+		Compress:   settings.LogSettings.Compress,
+	})
+}
+
+func newPlainWriteSyncer(filename string) (zapcore.WriteSyncer, error) {
+	f, err := os.OpenFile(filename, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o644)
+	if err != nil {
+		return nil, err
+	}
+	return zapcore.AddSync(f), nil
+}

logger/logger.go

@@ -3,14 +3,12 @@ package logger
 import (
 	"log"
 	"os"
-	"path/filepath"
 	"time"
 
 	"github.com/gin-gonic/gin"
 	"github.com/uozi-tech/cosy/settings"
 	"go.uber.org/zap"
 	"go.uber.org/zap/zapcore"
-	"gopkg.in/natefinch/lumberjack.v2"
 )
 
 var logger *zap.SugaredLogger
@@ -109,34 +107,11 @@ func Init(mode string) {
 	}
 
 	if settings.LogSettings.EnableFileLog {
-		if err := os.MkdirAll(settings.LogSettings.Dir, 0755); err != nil {
-			log.Fatal(err)
+		fileCores, err := NewFileCores(mode, highPriority, lowPriority)
+		if err != nil {
+			log.Fatalf("init file logger: %v", err)
 		}
-
-		errorLogWriter := &lumberjack.Logger{
-			Filename:   filepath.Join(settings.LogSettings.Dir, "error.log"),
-			MaxSize:    settings.LogSettings.MaxSize,
-			MaxBackups: settings.LogSettings.MaxBackups,
-			MaxAge:     settings.LogSettings.MaxAge,
-			LocalTime:  true,
-			Compress:   settings.LogSettings.Compress,
-		}
-
-		infoLogWriter := &lumberjack.Logger{
-			Filename:   filepath.Join(settings.LogSettings.Dir, "info.log"),
-			MaxSize:    settings.LogSettings.MaxSize,
-			MaxBackups: settings.LogSettings.MaxBackups,
-			MaxAge:     settings.LogSettings.MaxAge,
-			LocalTime:  true,
-			Compress:   settings.LogSettings.Compress,
-		}
-
-		fileEncoder := GetFileEncoder(mode)
-
-		cores = append(cores,
-			zapcore.NewCore(fileEncoder, zapcore.AddSync(errorLogWriter), highPriority),
-			zapcore.NewCore(fileEncoder, zapcore.AddSync(infoLogWriter), lowPriority),
-		)
+		cores = append(cores, fileCores...)
 	}
 
 	// Initialize enhanced logger (console + file only)

settings/log.go

@@ -1,6 +1,10 @@
 package settings
 
 type LogConfig struct {
+	// EnableRotate controls whether to use size-based log rotation.
+	// Default true to prevent unbounded growth.
+	EnableRotate bool
+
 	// EnableFileLog is a flag to enable file logging.
 	EnableFileLog bool
 
@@ -32,4 +36,6 @@ type LogConfig struct {
 	Compress bool
 }
 
-var LogSettings = &LogConfig{}
+var LogSettings = &LogConfig{
+	EnableRotate: true,
+}