Function get_engine

api_key_manager/models.py:65–94

尝试创建数据库引擎并处理连接错误

(retries=5, delay=2)

# 创建带有重试机制的数据库引擎
def get_engine(retries=5, delay=2):
    """尝试创建数据库引擎并处理连接错误"""
    for attempt in range(retries):
        try:
            logger.info(f"尝试连接到数据库: {DB_HOST}:{DB_PORT}, 尝试 {attempt+1}/{retries}")
            # 添加connect_args以提高连接可靠性
            engine = create_engine(
                DATABASE_URL, 
                pool_pre_ping=True,  # 检查连接是否活跃
                pool_recycle=3600,   # 一小时后回收连接
                pool_size=20,        # 增加连接池大小，默认为5
                max_overflow=30,     # 增加溢出连接数，默认为10
                pool_timeout=60,     # 增加等待可用连接的超时时间（秒）
                connect_args={
                    "connect_timeout": 30  # 连接超时时间（秒）
                }
            )
            # 测试连接
            connection = engine.connect()
            connection.close()
            logger.info("数据库连接成功")
            return engine
        except Exception as e:
            logger.error(f"数据库连接失败: {str(e)}")
            if attempt < retries - 1:
                logger.info(f"将在 {delay} 秒后重试...")
                time.sleep(delay)
            else:
                logger.error("所有数据库连接尝试均失败")
                raise Exception("无法连接到数据库，已达到最大重试次数")

# 初始化数据库
try: