Update Context for store

internal/scheduler/scheduler.go

@@ -2,6 +2,7 @@
 package scheduler
 
 import (
+	"context"
 	"gemini-balancer/internal/repository"
 	"gemini-balancer/internal/service"
 	"time"
@@ -15,7 +16,6 @@ type Scheduler struct {
 	logger          *logrus.Entry
 	statsService    *service.StatsService
 	keyRepo         repository.KeyRepository
-	// healthCheckService *service.HealthCheckService // 健康检查任务预留
 }
 
 func NewScheduler(statsSvc *service.StatsService, keyRepo repository.KeyRepository, logger *logrus.Logger) *Scheduler {
@@ -32,11 +32,13 @@ func NewScheduler(statsSvc *service.StatsService, keyRepo repository.KeyReposito
 func (s *Scheduler) Start() {
 	s.logger.Info("Starting scheduler and registering jobs...")
 
-	// --- 任务注册 ---
+	// 任务一：每小时执行一次的统计聚合
 	// 使用CRON表达式，精确定义“每小时的第5分钟”执行
 	_, err := s.gocronScheduler.Cron("5 * * * *").Tag("stats-aggregation").Do(func() {
 		s.logger.Info("Executing hourly request stats aggregation...")
-		if err := s.statsService.AggregateHourlyStats(); err != nil {
+		// 为后台定时任务创建一个新的、空的 context
+		ctx := context.Background()
+		if err := s.statsService.AggregateHourlyStats(ctx); err != nil {
 			s.logger.WithError(err).Error("Hourly stats aggregation failed.")
 		} else {
 			s.logger.Info("Hourly stats aggregation completed successfully.")
@@ -46,23 +48,14 @@ func (s *Scheduler) Start() {
 		s.logger.Errorf("Failed to schedule [stats-aggregation]: %v", err)
 	}
 
-	// 任务二：(预留) 自动健康检查 (例如：每10分钟一次)
-	/*
-	   _, err = s.gocronScheduler.Every(10).Minutes().Tag("auto-health-check").Do(func() {
-	       s.logger.Info("Executing periodic health check for all groups...")
-	       // s.healthCheckService.StartGlobalCheckTask() // 伪代码
-	   })
-	   if err != nil {
-	       s.logger.Errorf("Failed to schedule [auto-health-check]: %v", err)
-	   }
-	*/
-	// [NEW] --- 任务三: 清理软删除的API Keys ---
+	// 任务二：(预留) 自动健康检查
+
+	// 任务三：每日执行一次的软删除Key清理
 	// Executes once daily at 3:15 AM UTC.
 	_, err = s.gocronScheduler.Cron("15 3 * * *").Tag("cleanup-soft-deleted-keys").Do(func() {
 		s.logger.Info("Executing daily cleanup of soft-deleted API keys...")
 
-		// Let's assume a retention period of 7 days for now.
-		// In a real scenario, this should come from settings.
+		// [假设保留7天，实际应来自配置
 		const retentionDays = 7
 
 		count, err := s.keyRepo.HardDeleteSoftDeletedBefore(time.Now().AddDate(0, 0, -retentionDays))
@@ -77,9 +70,8 @@ func (s *Scheduler) Start() {
 	if err != nil {
 		s.logger.Errorf("Failed to schedule [cleanup-soft-deleted-keys]: %v", err)
 	}
-	// --- 任务注册结束 ---
 
-	s.gocronScheduler.StartAsync() // 异步启动，不阻塞应用主线程
+	s.gocronScheduler.StartAsync()
 	s.logger.Info("Scheduler started.")
 }