Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions backend/core/registrar.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from backend.common.log import set_custom_logfile, setup_logging
from backend.core.conf import settings
from backend.core.path_conf import STATIC_DIR, UPLOAD_DIR
from backend.database.db import create_table
from backend.database.db import create_tables
from backend.database.redis import redis_client
from backend.middleware.access_middleware import AccessMiddleware
from backend.middleware.jwt_auth_middleware import JwtAuthMiddleware
Expand All @@ -40,7 +40,7 @@ async def register_init(app: FastAPI) -> AsyncGenerator[None, None]:
:return:
"""
# 创建数据库表
await create_table()
await create_tables()
# 初始化 limiter
await FastAPILimiter.init(
redis=redis_client,
Expand Down
6 changes: 5 additions & 1 deletion backend/database/db.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
yield session


async def create_table() -> None:
async def create_tables() -> None:
"""创建数据库表"""
async with async_engine.begin() as coon:
await coon.run_sync(MappedBase.metadata.create_all)
Expand All @@ -86,7 +86,11 @@ def uuid4_str() -> str:
return str(uuid4())


# SQLA 数据库链接
SQLALCHEMY_DATABASE_URL = create_database_url()

# SALA 异步引擎和会话
async_engine, async_db_session = create_async_engine_and_session(SQLALCHEMY_DATABASE_URL)

# Session Annotated
CurrentSession = Annotated[AsyncSession, Depends(get_db)]