feat: SQLite数据库路径集中化与服务稳定性增强

lanshuizhiyue · lanshuizhiyue · commit fe2dd6baaee8 · 2026-03-24T15:44:15.000+08:00
diff --git a/src/backEnd/app.py b/src/backEnd/app.py
@@ -30,13 +30,12 @@ async def lifespan(app: FastAPI):
     """FastAPI 生命周期管理"""
     # 启动时，从配置加载刷新间隔
     initial_interval = get_refresh_interval()
-    print(f"[WebSocket] 启动 WebSocket 管理器，刷新间隔: {initial_interval} 分钟")
+    logger.info(f"启动 WebSocket 管理器，刷新间隔: {initial_interval} 分钟")
     ws_manager.start(initial_interval)
     logger.info(f"WebSocket 管理器已启动，刷新间隔: {initial_interval} 分钟")
     yield
     # 关闭时
     await ws_manager.stop()
-    print("[WebSocket] WebSocket 管理器已停止")
     logger.info("WebSocket 管理器已停止")
 
 app = FastAPI(lifespan=lifespan)
diff --git a/src/backEnd/main.py b/src/backEnd/main.py
@@ -3,6 +3,35 @@
 import tempfile
 import os
 
+
+def disable_windows_quick_edit():
+    """
+    禁用 Windows 控制台的 Quick Edit Mode。
+    Quick Edit Mode 会导致点击控制台窗口时进程输出被暂停，
+    必须按 Enter 才能恢复，这会导致后端服务卡住。
+    """
+    if sys.platform != "win32":
+        return
+    try:
+        import ctypes
+        kernel32 = ctypes.windll.kernel32
+        # STD_INPUT_HANDLE = -10
+        handle = kernel32.GetStdHandle(-10)
+        mode = ctypes.c_ulong()
+        if kernel32.GetConsoleMode(handle, ctypes.byref(mode)):
+            # ENABLE_QUICK_EDIT_MODE = 0x0040
+            # ENABLE_EXTENDED_FLAGS = 0x0080 (必须设置才能修改 Quick Edit)
+            mode.value &= ~0x0040  # 禁用 Quick Edit
+            mode.value |= 0x0080   # 启用 Extended Flags
+            kernel32.SetConsoleMode(handle, mode)
+            print("[INFO] Windows Quick Edit Mode 已禁用，控制台点击不会再暂停进程")
+    except Exception as e:
+        print(f"[WARNING] 无法禁用 Quick Edit Mode: {e}")
+
+
+# 在服务启动前禁用 Windows Quick Edit Mode
+disable_windows_quick_edit()
+
 # 配置 Python 模块导入路径 - 必须在所有项目模块导入之前完成
 current_dir = os.path.dirname(os.path.abspath(__file__))
 sqlmap_path = os.path.join(current_dir, "third_lib", "sqlmap")
@@ -164,7 +193,9 @@ def main(username, password):
 
         uvicorn.run(app=app, host="127.0.0.1", port=8775, reload=False, log_config='./uvicorn_config.json')
     except Exception as e:
-        print(e)
+        import traceback
+        print(f"\n[ERROR] 服务启动失败: {e}")
+        traceback.print_exc()
 
 if __name__ == "__main__":
     username = "admin"
diff --git a/src/backEnd/model/Database.py b/src/backEnd/model/Database.py
@@ -1,8 +1,11 @@
+import logging
 import sqlite3
 import threading
 import time
 
-from third_lib.sqlmap.lib.core.data import logger
+from third_lib.sqlmap.lib.core.data import logger as sqlmap_logger
+
+logger = logging.getLogger(__name__)
 from third_lib.sqlmap.lib.core.common import getSafeExString
 
 
@@ -33,6 +36,8 @@ def commit(self):
         self.connection.commit()
 
     def execute(self, statement, arguments=None):
+        max_retries = 5
+        retry_count = 0
         with self.lock:
             while True:
                 try:
@@ -44,14 +49,22 @@ def execute(self, statement, arguments=None):
                     if "locked" not in getSafeExString(ex):
                         raise
                     else:
-                        time.sleep(1)
+                        retry_count += 1
+                        if retry_count > max_retries:
+                            logger.error(f"Database locked after {max_retries} retries, giving up")
+                            raise sqlite3.OperationalError(f"Database locked after {max_retries} retries")
+                        wait_time = min(0.05 * (2 ** retry_count), 1.0)
+                        logger.warning(f"Database locked, retry {retry_count}/{max_retries} after {wait_time:.2f}s")
+                        time.sleep(wait_time)
                 else:
                     break
 
         if statement.lstrip().upper().startswith("SELECT"):
             return self.cursor.fetchall()
 
     def only_execute(self, statement, arguments=None):
+        max_retries = 5
+        retry_count = 0
         with self.lock:
             while True:
                 try:
@@ -63,19 +76,30 @@ def only_execute(self, statement, arguments=None):
                     if "locked" not in getSafeExString(ex):
                         raise
                     else:
-                        time.sleep(1)
+                        retry_count += 1
+                        if retry_count > max_retries:
+                            logger.error(f"Database locked after {max_retries} retries, giving up")
+                            raise sqlite3.OperationalError(f"Database locked after {max_retries} retries")
+                        wait_time = min(0.05 * (2 ** retry_count), 1.0)
+                        logger.warning(f"Database locked, retry {retry_count}/{max_retries} after {wait_time:.2f}s")
+                        time.sleep(wait_time)
                 else:
                     break
 
         return self.cursor
 
     def init(self):
         self.execute(
-            "CREATE TABLE logs(id INTEGER PRIMARY KEY AUTOINCREMENT, taskid INTEGER, datetime TEXT, level TEXT, message TEXT)")
+            "CREATE TABLE IF NOT EXISTS logs(id INTEGER PRIMARY KEY AUTOINCREMENT, taskid INTEGER, datetime TEXT, level TEXT, message TEXT)")
         self.execute(
-            "CREATE TABLE data(id INTEGER PRIMARY KEY AUTOINCREMENT, taskid INTEGER, status INTEGER, content_type INTEGER, value TEXT)")
+            "CREATE TABLE IF NOT EXISTS data(id INTEGER PRIMARY KEY AUTOINCREMENT, taskid INTEGER, status INTEGER, content_type INTEGER, value TEXT)")
         self.execute(
-            "CREATE TABLE errors(id INTEGER PRIMARY KEY AUTOINCREMENT, taskid INTEGER, error TEXT)")
+            "CREATE TABLE IF NOT EXISTS errors(id INTEGER PRIMARY KEY AUTOINCREMENT, taskid INTEGER, error TEXT)")
+        
+        # IPC 表是进程间临时通信数据，每次启动时清空旧数据，避免无限增长
+        self.execute("DELETE FROM logs")
+        self.execute("DELETE FROM data")
+        self.execute("DELETE FROM errors")
         
         # 创建持久化请求头规则表
         self.execute("""
diff --git a/src/backEnd/pyproject.toml b/src/backEnd/pyproject.toml
@@ -3,7 +3,9 @@ name = "sqlmapwebui"
 version = "0.1.0"
 description = "Add your description here"
 requires-python = ">=3.10"
-dependencies = []
+dependencies = [
+    "psutil",
+]
 
 [project.optional-dependencies]
 thirdparty = [
diff --git a/src/backEnd/service/taskService.py b/src/backEnd/service/taskService.py
@@ -1,4 +1,3 @@
-import pdb
 import os
 import sys
 import asyncio
diff --git a/src/backEnd/utils/task_monitor.py b/src/backEnd/utils/task_monitor.py
@@ -3,10 +3,19 @@
 from model.TaskStatus import TaskStatus
 from third_lib.sqlmap.lib.core.data import logger
 
-import psutil
-
 from model.DataStore import DataStore
 
+# psutil 导入容错处理
+try:
+    import psutil
+    # 模块级别预热：psutil.cpu_percent 首次调用 interval=0 会返回 0.0（无意义）
+    # 需要先执行一次 interval=1 的调用来初始化内部状态
+    psutil.cpu_percent(interval=1)
+    _has_psutil = True
+except ImportError:
+    psutil = None
+    _has_psutil = False
+
 
 def get_max_tasks_count():
     """
@@ -17,8 +26,11 @@ def get_max_tasks_count():
     # 获取逻辑核心数
     logical_cores = os.cpu_count() or 1
 
-    # 获取当前 CPU 平均占用率（过去 1 秒的平均值）
-    cpu_usage = psutil.cpu_percent(interval=1)
+    # 获取当前 CPU 平均占用率（非阻塞，立即返回）
+    if _has_psutil:
+        cpu_usage = psutil.cpu_percent(interval=0)
+    else:
+        cpu_usage = 50  # psutil 不可用时，默认假设 50% CPU 使用率
 
     # 根据 CPU 使用率动态调整最大任务数
     # 如果 CPU 使用率较高，则减少最大任务数
diff --git a/src/backEnd/utils/websocket_manager.py b/src/backEnd/utils/websocket_manager.py
@@ -155,30 +155,24 @@ def _restart_refresh_task(self) -> None:
     
     async def _refresh_loop(self) -> None:
         """定时刷新循环"""
-        print(f"[WebSocket] 定时刷新循环已启动，间隔: {self._refresh_interval} 分钟")
         logger.info(f"定时刷新循环已启动，间隔: {self._refresh_interval} 分钟")
         while self._is_running:
             try:
                 # 等待指定的刷新间隔（转换为秒）
                 wait_seconds = self._refresh_interval * 60
-                print(f"[WebSocket] 下一次刷新将在 {self._refresh_interval} 分钟后 ({wait_seconds}秒)")
-                logger.info(f"下一次刷新将在 {self._refresh_interval} 分钟后 ({wait_seconds}秒)")
+                logger.debug(f"下一次刷新将在 {self._refresh_interval} 分钟后 ({wait_seconds}秒)")
                 await asyncio.sleep(wait_seconds)
                 
                 # 只有有连接时才广播
                 if self._active_connections:
-                    print(f"[WebSocket] 定时刷新触发，准备广播到 {self.connection_count} 个连接")
                     logger.info(f"定时刷新触发，准备广播到 {self.connection_count} 个连接")
                     await self.broadcast_refresh()
                 else:
-                    print("[WebSocket] 定时刷新触发，但没有活跃连接，跳过广播")
                     logger.info("定时刷新触发，但没有活跃连接，跳过广播")
             except asyncio.CancelledError:
-                print("[WebSocket] 定时刷新任务被取消")
                 logger.info("定时刷新任务被取消")
                 break
             except Exception as e:
-                print(f"[WebSocket] 定时刷新循环出错: {e}")
                 logger.error(f"定时刷新循环出错: {e}")
                 await asyncio.sleep(5)  # 出错后短暂等待再重试
     
@@ -193,7 +187,6 @@ def start(self, initial_interval: int = None) -> None:
                 self._refresh_interval = max(1, min(60, initial_interval))
             self._is_running = True
             self._refresh_task = asyncio.create_task(self._refresh_loop())
-            print(f"[WebSocket] WebSocket 管理器已启动，刷新间隔: {self._refresh_interval}分钟")
             logger.info(f"WebSocket 管理器已启动，刷新间隔: {self._refresh_interval}分钟")
     
     async def stop(self) -> None:
diff --git a/src/backEnd/uv.lock b/src/backEnd/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`		`-import pdb`
`2`	`1`	`import os`
`3`	`2`	`import sys`
`4`	`3`	`import asyncio`