feat: 配置分离

internal/services/log_cleanup.go

@@ -0,0 +1,102 @@
+package services
+
+import (
+	"gpt-load/internal/config"
+	"gpt-load/internal/db"
+	"gpt-load/internal/models"
+	"time"
+
+	"github.com/sirupsen/logrus"
+)
+
+// LogCleanupService 负责清理过期的请求日志
+type LogCleanupService struct {
+	stopCh chan struct{}
+}
+
+// NewLogCleanupService 创建新的日志清理服务
+func NewLogCleanupService() *LogCleanupService {
+	return &LogCleanupService{
+		stopCh: make(chan struct{}),
+	}
+}
+
+// Start 启动日志清理服务
+func (s *LogCleanupService) Start() {
+	go s.run()
+	logrus.Info("Log cleanup service started")
+}
+
+// Stop 停止日志清理服务
+func (s *LogCleanupService) Stop() {
+	close(s.stopCh)
+	logrus.Info("Log cleanup service stopped")
+}
+
+// run 运行日志清理的主循环
+func (s *LogCleanupService) run() {
+	// 每天凌晨2点执行清理任务
+	ticker := time.NewTicker(24 * time.Hour)
+	defer ticker.Stop()
+
+	// 启动时先执行一次清理
+	s.cleanupExpiredLogs()
+
+	for {
+		select {
+		case <-ticker.C:
+			s.cleanupExpiredLogs()
+		case <-s.stopCh:
+			return
+		}
+	}
+}
+
+// cleanupExpiredLogs 清理过期的请求日志
+func (s *LogCleanupService) cleanupExpiredLogs() {
+	if db.DB == nil {
+		logrus.Error("Database connection is not available for log cleanup")
+		return
+	}
+
+	// 获取日志保留天数配置
+	settingsManager := config.GetSystemSettingsManager()
+	if settingsManager == nil {
+		logrus.Error("System settings manager is not available for log cleanup")
+		return
+	}
+
+	settings := settingsManager.GetSettings()
+	retentionDays := settings.RequestLogRetentionDays
+
+	if retentionDays <= 0 {
+		logrus.Debug("Log retention is disabled (retention_days <= 0)")
+		return
+	}
+
+	// 计算过期时间点
+	cutoffTime := time.Now().AddDate(0, 0, -retentionDays).UTC()
+
+	// 执行删除操作
+	result := db.DB.Where("timestamp < ?", cutoffTime).Delete(&models.RequestLog{})
+	if result.Error != nil {
+		logrus.WithError(result.Error).Error("Failed to cleanup expired request logs")
+		return
+	}
+
+	if result.RowsAffected > 0 {
+		logrus.WithFields(logrus.Fields{
+			"deleted_count":  result.RowsAffected,
+			"cutoff_time":    cutoffTime.Format(time.RFC3339),
+			"retention_days": retentionDays,
+		}).Info("Successfully cleaned up expired request logs")
+	} else {
+		logrus.Debug("No expired request logs found to cleanup")
+	}
+}
+
+// CleanupNow 立即执行一次日志清理
+func (s *LogCleanupService) CleanupNow() {
+	logrus.Info("Manual log cleanup triggered")
+	s.cleanupExpiredLogs()
+}