96 lines
2.3 KiB
Go
96 lines
2.3 KiB
Go
package services
|
|
|
|
import (
|
|
"gpt-load/internal/config"
|
|
"gpt-load/internal/models"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// LogCleanupService 负责清理过期的请求日志
|
|
type LogCleanupService struct {
|
|
db *gorm.DB
|
|
settingsManager *config.SystemSettingsManager
|
|
leaderService *LeaderService
|
|
stopCh chan struct{}
|
|
}
|
|
|
|
// NewLogCleanupService 创建新的日志清理服务
|
|
func NewLogCleanupService(db *gorm.DB, settingsManager *config.SystemSettingsManager, leaderService *LeaderService) *LogCleanupService {
|
|
return &LogCleanupService{
|
|
db: db,
|
|
settingsManager: settingsManager,
|
|
leaderService: leaderService,
|
|
stopCh: make(chan struct{}),
|
|
}
|
|
}
|
|
|
|
// Start 启动日志清理服务
|
|
func (s *LogCleanupService) Start() {
|
|
go s.run()
|
|
logrus.Debug("Log cleanup service started")
|
|
}
|
|
|
|
// Stop 停止日志清理服务
|
|
func (s *LogCleanupService) Stop() {
|
|
close(s.stopCh)
|
|
logrus.Info("Log cleanup service stopped")
|
|
}
|
|
|
|
// run 运行日志清理的主循环
|
|
func (s *LogCleanupService) run() {
|
|
ticker := time.NewTicker(2 * time.Hour)
|
|
defer ticker.Stop()
|
|
|
|
// 启动时先执行一次清理
|
|
s.cleanupExpiredLogs()
|
|
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
s.cleanupExpiredLogs()
|
|
case <-s.stopCh:
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// cleanupExpiredLogs 清理过期的请求日志
|
|
func (s *LogCleanupService) cleanupExpiredLogs() {
|
|
if !s.leaderService.IsLeader() {
|
|
logrus.Debug("Not the leader, skipping log cleanup.")
|
|
return
|
|
}
|
|
|
|
// 获取日志保留天数配置
|
|
settings := s.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 := s.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")
|
|
}
|
|
}
|