feat: 添加module表status字段及相关接口优化

1. 在module表新增status字段（0：待确认；1：正常；2：弃用）
2. 修改/document/generate-cases接口，创建模块时设置status=0
3. 修改/case/restore接口，恢复用例时同步更新模块及其父模块的status为1
4. 修改/case/list接口，支持module_status参数过滤，默认只显示status=1的模块
5. 修改/case/list接口，返回module_path字段
6. 修改/plan/case/list接口，返回module_path字段
7. 修改/module/tree接口，默认筛选status=1的数据
8. 优化数据库连接池配置，添加连接验证和自动重试机制

common/sqlSession.py

@@ -2,6 +2,7 @@
 from sqlalchemy import create_engine, text
 from sqlalchemy.orm import sessionmaker
 from urllib.parse import quote_plus as urlquote
+import time
 
 from const import sparkatp_sql_uri
 
@@ -10,6 +11,10 @@ from logger import logger
 _ENGINE_CACHE = {}
 _SESSION_FACTORY_CACHE = {}
 
+# 重试配置
+MAX_RETRY_ATTEMPTS = 3
+RETRY_DELAY = 1  # 秒
+
 
 """
 sql操作：
@@ -32,28 +37,52 @@ class SqlSession:
     def build_postgres_uri(host, port, user, password, database):
         return f"postgresql+psycopg2://{user}:{urlquote(str(password))}@{host}:{port}/{database}"
 
-    def get_session(self):
-        session_factory = _SESSION_FACTORY_CACHE.get(self.sql_uri)
-        if session_factory is None:
-            engine = _ENGINE_CACHE.get(self.sql_uri)
-            if engine is None:
-                engine = create_engine(
-                    self.sql_uri,
-                    pool_size=20,
-                    max_overflow=30,
-                    pool_pre_ping=True,
-                    pool_recycle=1200,
-                    pool_timeout=60,
-                    pool_use_lifo=True,
-                    connect_args={
-                        'connect_timeout': 10,
-                        'options': '-c timezone=Asia/Shanghai'
-                    }
-                )
-                _ENGINE_CACHE[self.sql_uri] = engine
-            session_factory = sessionmaker(bind=engine, autoflush=False, expire_on_commit=False)
-            _SESSION_FACTORY_CACHE[self.sql_uri] = session_factory
-        return session_factory()
+    def get_session(self, retry_count=0):
+        try:
+            session_factory = _SESSION_FACTORY_CACHE.get(self.sql_uri)
+            if session_factory is None:
+                engine = _ENGINE_CACHE.get(self.sql_uri)
+                if engine is None:
+                    engine = create_engine(
+                        self.sql_uri,
+                        pool_size=10,
+                        max_overflow=20,
+                        pool_recycle=180,  # 3分钟回收连接，避免连接被服务器断开
+                        pool_pre_ping=True,  # 获取连接前先验证有效性
+                        pool_timeout=15,  # 获取连接超时时间
+                        connect_args={
+                            'connect_timeout': 5,
+                            'options': '-c timezone=Asia/Shanghai',
+                            'keepalives': 1,
+                            'keepalives_idle': 30,
+                            'keepalives_interval': 5,
+                            'keepalives_count': 3
+                        }
+                    )
+                    _ENGINE_CACHE[self.sql_uri] = engine
+                session_factory = sessionmaker(bind=engine, autoflush=False, expire_on_commit=False)
+                _SESSION_FACTORY_CACHE[self.sql_uri] = session_factory
+            session = session_factory()
+            
+            # 验证连接是否有效
+            try:
+                session.execute(text("SELECT 1"))
+            except Exception as e:
+                logger.warning(f"连接验证失败，尝试重新获取连接: {e}")
+                # 清除缓存的session factory，强制创建新连接
+                _SESSION_FACTORY_CACHE.pop(self.sql_uri, None)
+                _ENGINE_CACHE.pop(self.sql_uri, None)
+                raise
+            
+            return session
+        except Exception as e:
+            if retry_count < MAX_RETRY_ATTEMPTS:
+                logger.warning(f"获取数据库连接失败，第 {retry_count + 1} 次重试: {e}")
+                time.sleep(RETRY_DELAY * (retry_count + 1))
+                return self.get_session(retry_count + 1)
+            else:
+                logger.error(f"获取数据库连接失败，已重试 {MAX_RETRY_ATTEMPTS} 次: {e}")
+                raise
 
     def query(self, *args):
         return self._session.query(*args)