fix memory manager startup

app/startup/lifecycle.py

@@ -4,8 +4,8 @@ from contextlib import asynccontextmanager
 from fastapi import FastAPI
 
 from app.core.config import global_vars
-from app.chain.torrents import TorrentsChain
 from app.startup.command_initializer import init_command, stop_command, restart_command
+from app.startup.memory_initializer import init_memory_manager, stop_memory_manager
 from app.startup.modules_initializer import init_modules, stop_modules
 from app.startup.monitor_initializer import stop_monitor, init_monitor
 from app.startup.plugins_initializer import init_plugins, stop_plugins, sync_plugins
@@ -35,8 +35,6 @@ async def lifespan(app: FastAPI):
     init_modules()
     # 初始化路由
     init_routers(app)
-    # 资源缓存
-    TorrentsChain()
     # 初始化插件
     init_plugins()
     # 初始化定时器
@@ -47,6 +45,8 @@ async def lifespan(app: FastAPI):
     init_command()
     # 初始化工作流
     init_workflow()
+    # 初始化内存管理
+    init_memory_manager()
     # 插件同步到本地
     sync_plugins_task = asyncio.create_task(init_plugin_system())
     try:
@@ -56,6 +56,7 @@ async def lifespan(app: FastAPI):
         print("Shutting down...")
         # 停止信号
         global_vars.stop_system()
+        # 取消同步插件任务
         try:
             sync_plugins_task.cancel()
             await sync_plugins_task
@@ -63,6 +64,8 @@ async def lifespan(app: FastAPI):
             pass
         except Exception as e:
             print(str(e))
+        # 停止内存管理器
+        stop_memory_manager()
         # 停止工作流
         stop_workflow()
         # 停止命令

app/startup/memory_initializer.py

@@ -0,0 +1,19 @@
+from app.core.config import settings
+from app.helper.memory import MemoryManager
+
+
+def init_memory_manager():
+    """
+    初始化内存监控器
+    """
+    memory_manager = MemoryManager()
+    # 设置内存阈值和启动监控
+    memory_manager.set_threshold(settings.CACHE_CONF['memory'])
+    memory_manager.start_monitoring()
+
+
+def stop_memory_manager():
+    """
+    停止内存监控器
+    """
+    MemoryManager().stop_monitoring()