mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-08-30 20:55:28 +08:00
snaily
aea2f39952
- 重构 README.md，更新项目描述、结构、配置说明和 API 端点信息。 - 在 .env.example 中添加 MySQL 数据库配置项。 - 将数据库连接池回收时间从 1 小时减少到 30 分钟 (app/database/connection.py)。 - 修复认证成功后的重定向 URL，从 /keys 指向 /config (app/router/routes.py)。 - 微调认证页面的背景透明度 (app/templates/auth.html)。 - 添加 cryptography 依赖以支持 MySQL 8+ 认证 (requirements.txt)。 - 添加示例图片文件 (files/image*.png)。
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
"""
|
|
数据库连接池模块
|
|
"""
|
|
from databases import Database
|
|
from sqlalchemy import create_engine, MetaData
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
|
|
from app.config.config import settings
|
|
from app.log.logger import get_database_logger
|
|
|
|
logger = get_database_logger()
|
|
|
|
# 数据库URL
|
|
DATABASE_URL = f"mysql+pymysql://{settings.MYSQL_USER}:{settings.MYSQL_PASSWORD}@{settings.MYSQL_HOST}:{settings.MYSQL_PORT}/{settings.MYSQL_DATABASE}"
|
|
|
|
# 创建数据库引擎
|
|
# pool_pre_ping=True: 在从连接池获取连接前执行简单的 "ping" 测试，确保连接有效
|
|
engine = create_engine(DATABASE_URL, pool_pre_ping=True)
|
|
|
|
# 创建元数据对象
|
|
metadata = MetaData()
|
|
|
|
# 创建基类
|
|
Base = declarative_base(metadata=metadata)
|
|
|
|
# 创建数据库连接池，并配置连接池参数
|
|
# min_size/max_size: 连接池的最小/最大连接数
|
|
# pool_recycle=3600: 连接在池中允许存在的最大秒数（生命周期）。
|
|
# 设置为 3600 秒（1小时），确保在 MySQL 默认的 wait_timeout (通常8小时) 或其他网络超时之前回收连接。
|
|
# 如果遇到连接失效问题，可以尝试调低此值，使其小于实际的 wait_timeout 或网络超时时间。
|
|
# databases 库会自动处理连接失效后的重连尝试。
|
|
database = Database(DATABASE_URL, min_size=5, max_size=20, pool_recycle=1800) # Reduced recycle time to 30 mins
|
|
|
|
|
|
async def connect_to_db():
|
|
"""
|
|
连接到数据库
|
|
"""
|
|
try:
|
|
await database.connect()
|
|
logger.info("Connected to database")
|
|
except Exception as e:
|
|
logger.error(f"Failed to connect to database: {str(e)}")
|
|
raise
|
|
|
|
|
|
async def disconnect_from_db():
|
|
"""
|
|
断开数据库连接
|
|
"""
|
|
try:
|
|
await database.disconnect()
|
|
logger.info("Disconnected from database")
|
|
except Exception as e:
|
|
logger.error(f"Failed to disconnect from database: {str(e)}")
|