53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
|
|
from sqlalchemy.orm import sessionmaker, declarative_base
|
|
from ..kangdaApi.parseConfig import parse_config
|
|
from .config import settings
|
|
|
|
# import yaml
|
|
|
|
# config_path = "app/config/config.yaml"
|
|
|
|
# config = parse_config(config_path)
|
|
|
|
# config_kangda = config["kangda"]
|
|
|
|
# print("sssss", config_kangda)
|
|
|
|
# 创建异步引擎
|
|
engine = create_async_engine(
|
|
f"mysql+aiomysql://{settings.DB_USER}:{settings.DB_PASSWORD}@{settings.DB_HOST}:{settings.DB_PORT}/{settings.DB_NAME}",
|
|
pool_size=settings.DB_POOL_SIZE,
|
|
max_overflow=settings.DB_MAX_OVERFLOW,
|
|
pool_timeout=settings.DB_POOL_TIMEOUT,
|
|
pool_recycle=settings.DB_POOL_RECYCLE,
|
|
echo=False
|
|
)
|
|
|
|
# engine = create_async_engine(
|
|
# f"mysql+aiomysql://{config_kangda["DB_USER"]}:{config_kangda["DB_PASSWORD"]}@{config_kangda["DB_HOST"]}:{config_kangda["DB_PORT"]}/{config_kangda["DB_NAME"]}",
|
|
# pool_size=config_kangda["DB_POOL_SIZE"],
|
|
# max_overflow=config_kangda["DB_MAX_OVERFLOW"],
|
|
# pool_timeout=config_kangda["DB_POOL_TIMEOUT"],
|
|
# pool_recycle=config_kangda["DB_POOL_RECYCLE"],
|
|
# echo=False
|
|
# )
|
|
|
|
# 创建异步会话工厂, expire_on_commit 禁用提交后过期对象--> 提交后原对象仍然可以继续使用.
|
|
async_session = sessionmaker(
|
|
engine, class_=AsyncSession, expire_on_commit=False
|
|
)
|
|
|
|
# 创建基类
|
|
Base = declarative_base()
|
|
|
|
# 获取数据库会话
|
|
async def get_db():
|
|
async with async_session() as session:
|
|
try:
|
|
yield session
|
|
await session.commit()
|
|
except Exception:
|
|
await session.rollback()
|
|
raise
|
|
finally:
|
|
await session.close() |