Add grade hot cache

app/scheduler/cache_scheduler.py

@@ -6,6 +6,11 @@ from app.config import settings
 from app.dependencies.database import get_db, get_redis
 from app.dependencies.fetcher import get_fetcher
 from app.log import logger
+from app.scheduler.user_cache_scheduler import (
+    schedule_user_cache_cleanup_task,
+    schedule_user_cache_preload_task,
+    schedule_user_cache_warmup_task,
+)
 
 
 class CacheScheduler:
@@ -42,17 +47,26 @@ class CacheScheduler:
         
         # 启动时执行一次排行榜缓存刷新
         await self._refresh_ranking_cache()
+        
+        # 启动时执行一次用户缓存预热
+        await self._warmup_user_cache()
 
         beatmap_cache_counter = 0
         ranking_cache_counter = 0
+        user_cache_counter = 0
+        user_cleanup_counter = 0
         
         # 从配置文件获取间隔设置
         check_interval = 5 * 60  # 5分钟检查间隔
         beatmap_cache_interval = 30 * 60  # 30分钟beatmap缓存间隔
         ranking_cache_interval = settings.ranking_cache_refresh_interval_minutes * 60  # 从配置读取
+        user_cache_interval = 15 * 60  # 15分钟用户缓存预加载间隔
+        user_cleanup_interval = 60 * 60  # 60分钟用户缓存清理间隔
         
         beatmap_cache_cycles = beatmap_cache_interval // check_interval
         ranking_cache_cycles = ranking_cache_interval // check_interval
+        user_cache_cycles = user_cache_interval // check_interval
+        user_cleanup_cycles = user_cleanup_interval // check_interval
 
         while self.running:
             try:
@@ -64,6 +78,8 @@ class CacheScheduler:
 
                 beatmap_cache_counter += 1
                 ranking_cache_counter += 1
+                user_cache_counter += 1
+                user_cleanup_counter += 1
 
                 # beatmap缓存预热
                 if beatmap_cache_counter >= beatmap_cache_cycles:
@@ -74,6 +90,16 @@ class CacheScheduler:
                 if ranking_cache_counter >= ranking_cache_cycles:
                     await self._refresh_ranking_cache()
                     ranking_cache_counter = 0
+                
+                # 用户缓存预加载
+                if user_cache_counter >= user_cache_cycles:
+                    await self._preload_user_cache()
+                    user_cache_counter = 0
+                
+                # 用户缓存清理
+                if user_cleanup_counter >= user_cleanup_cycles:
+                    await self._cleanup_user_cache()
+                    user_cleanup_counter = 0
 
             except asyncio.CancelledError:
                 break
@@ -110,16 +136,37 @@ class CacheScheduler:
                 schedule_ranking_refresh_task,
             )
 
-            # 获取数据库会话
-            async for session in get_db():
+            # 使用独立的数据库会话
+            from app.dependencies.database import with_db
+            async with with_db() as session:
                 await schedule_ranking_refresh_task(session, redis)
-                break  # 只需要一次会话
 
             logger.info("Ranking cache refresh completed successfully")
 
         except Exception as e:
             logger.error(f"Ranking cache refresh failed: {e}")
 
+    async def _warmup_user_cache(self):
+        """用户缓存预热"""
+        try:
+            await schedule_user_cache_warmup_task()
+        except Exception as e:
+            logger.error(f"User cache warmup failed: {e}")
+
+    async def _preload_user_cache(self):
+        """用户缓存预加载"""
+        try:
+            await schedule_user_cache_preload_task()
+        except Exception as e:
+            logger.error(f"User cache preload failed: {e}")
+
+    async def _cleanup_user_cache(self):
+        """用户缓存清理"""
+        try:
+            await schedule_user_cache_cleanup_task()
+        except Exception as e:
+            logger.error(f"User cache cleanup failed: {e}")
+
 
 # Beatmap缓存调度器（保持向后兼容）
 class BeatmapsetCacheScheduler(CacheScheduler):