'commit'

DbKit/Db.py

@@ -1,6 +1,8 @@
 import os
 import logging
 import re
+import asyncio
+import functools
 from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
 from sqlalchemy.orm import sessionmaker
 from sqlalchemy.sql import text
@@ -14,6 +16,42 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(level
 logger = logging.getLogger(__name__)
 
 
+def db_retry(max_retries=3, delay=1):
+    """
+    数据库操作重试装饰器
+    
+    Args:
+        max_retries: 最大重试次数，默认为3次
+        delay: 重试间隔时间（秒），默认为1秒
+    """
+    def decorator(func):
+        @functools.wraps(func)
+        async def wrapper(*args, **kwargs):
+            # 检查是否提供了 session，如果提供了 session，通常意味着是在一个更大的事务中，不建议在这里重试
+            session = kwargs.get('session')
+            if session is not None:
+                return await func(*args, **kwargs)
+
+            last_exception = None
+            for attempt in range(max_retries):
+                try:
+                    return await func(*args, **kwargs)
+                except Exception as e:
+                    last_exception = e
+                    # 只有在还有重试机会时才打印警告
+                    if attempt < max_retries - 1:
+                        logger.warning(f"数据库操作失败 (尝试 {attempt + 1}/{max_retries})，正在进行重试: {str(e)}")
+                        await asyncio.sleep(delay)
+                    else:
+                        logger.error(f"数据库操作在 {max_retries} 次尝试后仍然失败: {str(e)}")
+            
+            # 如果循环结束仍未返回，说明最后一次尝试也失败了，抛出异常
+            if last_exception:
+                raise last_exception
+        return wrapper
+    return decorator
+
+
 class Db:
     """通用数据库操作封装类，提供数据库连接和操作功能"""
     # 单例实例
@@ -341,6 +379,7 @@ class Db:
 
         return processed_sql
 
+    @db_retry()
     async def find(self, sql, params=None, session=None):
         """
         执行SQL查询并返回结果（异步版本）
@@ -783,6 +822,7 @@ class Db:
         # 如果所有策略都失败，返回默认查询
         return "SELECT COUNT(*)"
 
+    @db_retry()
     async def execute_update(self, sql, params=None, session=None):
         """执行SQL更新操作（插入、更新、删除）（异步版本）
 
@@ -890,6 +930,7 @@ class Db:
             if is_own_session and session:
                 await session.close()
 
+    @db_retry()
     async def save(self, table_name, data, primary_key, session=None):
         """插入数据到指定表，并返回插入后的主键值（异步版本）
 
@@ -948,6 +989,7 @@ class Db:
             if is_own_session and session:
                 await session.close()
 
+    @db_retry()
     async def update(self, table_name, data, primary_key, session=None):
         """根据主键更新指定表中的数据（异步版本）
 
@@ -1013,6 +1055,7 @@ class Db:
             if is_own_session and session:
                 await session.close()
 
+    @db_retry()
     async def batch_insert(self, table_name, data_list, primary_key=None, session=None):
         """
         批量插入数据到指定表（异步版本）
@@ -1092,6 +1135,7 @@ class Db:
             if is_own_session and session:
                 await session.close()
 
+    @db_retry()
     async def batch_update(self, table_name, data_list, primary_key, session=None):
         """
         批量更新指定表中的数据（异步版本）