refactor: 重构数据库orm为异步 #I9MRQS

This commit is contained in:
insistence
2024-05-17 15:04:30 +08:00
parent 18103e3d38
commit 1972f1cb6d
52 changed files with 1414 additions and 1246 deletions

View File

@@ -2,16 +2,13 @@ from config.database import *
from utils.log_util import logger
def get_db_pro():
async def get_db():
"""
每一个请求处理完毕后会关闭当前连接,不同的请求使用不同的连接
:return:
"""
current_db = SessionLocal()
try:
async with AsyncSessionLocal() as current_db:
yield current_db
finally:
current_db.close()
async def init_create_table():
@@ -20,8 +17,6 @@ async def init_create_table():
:return:
"""
logger.info("初始化数据库连接...")
Base.metadata.create_all(bind=engine)
async with async_engine.begin() as conn:
await conn.run_sync(Base.metadata.create_all)
logger.info("数据库连接成功")
get_db = get_db_pro