martsforever
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from typing import Annotated
|
||
|
||
from fastapi.params import Depends
|
||
from sqlalchemy import text
|
||
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
|
||
from sqlalchemy.orm import sessionmaker
|
||
|
||
from app.config.env import env
|
||
|
||
DATABASE_URL = f"mysql+asyncmy://{env.db_username}:{env.db_password}@{env.db_host}:{env.db_port}/{env.db_database}?charset=utf8mb4"
|
||
|
||
async_engine = create_async_engine(
|
||
DATABASE_URL,
|
||
# 连接池保持的连接数
|
||
pool_size=5,
|
||
# 允许超过pool_size的最大连接数
|
||
max_overflow=10,
|
||
# 获取连接的超时时间(秒)
|
||
pool_timeout=5,
|
||
# 连接回收时间(秒)
|
||
pool_recycle=60, # 针对 Docker NAT 环境优化
|
||
# 启用连接有效性检测
|
||
pool_pre_ping=True,
|
||
# 启用SQL语句日志输出，便于开发调试
|
||
echo=True,
|
||
# 启用SQLAlchemy 2.0风格的未来模式API
|
||
future=True,
|
||
)
|
||
|
||
# 使用这个 Session 工厂
|
||
async_session = sessionmaker(
|
||
async_engine,
|
||
class_=AsyncSession,
|
||
expire_on_commit=False
|
||
)
|
||
|
||
|
||
# 作用：用于在接口中注入得到会话实例对象session，在接口执行完毕之后，自动执行close动作关闭会话
|
||
async def get_async_session() -> AsyncSession:
|
||
async with async_session() as session:
|
||
yield session
|
||
|
||
|
||
# 作为类型使用，直接在接口参数中注入得到session会话实例对象
|
||
AsyncSessionDep = Annotated[AsyncSession, Depends(get_async_session)]
|
||
|
||
|
||
# 用于启动服务的时候检查数据库连接是否正常
|
||
async def check_database_connection():
|
||
"""检查数据库连接是否正常"""
|
||
try:
|
||
async with async_engine.begin() as conn:
|
||
print("Connecting Mysql...")
|
||
await conn.execute(text("select 1"))
|
||
# 打印连接成功信息及连接URL
|
||
print("✅ MySql connection successful：", DATABASE_URL)
|
||
except Exception as e:
|
||
# 打印连接失败信息及错误详情
|
||
print(f"❌ Database connection failed: {e}")
|
||
# 重新抛出异常，让上层处理
|
||
raise e
|
||
|
||
return async_engine
|