refactor(database): use asyncio

app/dependencies/__init__.py

@@ -1,2 +1,4 @@
+from __future__ import annotations
+
 from .database import get_db as get_db
-from .user import get_current_user as get_current_user
+from .user import get_current_user as get_current_user

app/dependencies/database.py

@@ -1,6 +1,8 @@
 from __future__ import annotations
 
-from sqlmodel import Session, create_engine
+from sqlalchemy.ext.asyncio import create_async_engine
+from sqlmodel import SQLModel
+from sqlmodel.ext.asyncio.session import AsyncSession
 
 try:
     import redis
@@ -9,7 +11,7 @@ except ImportError:
 from app.config import settings
 
 # 数据库引擎
-engine = create_engine(settings.DATABASE_URL)
+engine = create_async_engine(settings.DATABASE_URL)
 
 # Redis 连接
 if redis:
@@ -19,11 +21,16 @@ else:
 
 
 # 数据库依赖
-def get_db():
-    with Session(engine) as session:
+async def get_db():
+    async with AsyncSession(engine) as session:
         yield session
 
 
+async def create_tables():
+    async with engine.begin() as conn:
+        await conn.run_sync(SQLModel.metadata.create_all)
+
+
 # Redis 依赖
 def get_redis():
     return redis_client

app/dependencies/user.py

@@ -9,14 +9,16 @@ from .database import get_db
 
 from fastapi import Depends, HTTPException
 from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
-from sqlmodel import Session, select
+from sqlalchemy.orm import joinedload, selectinload
+from sqlmodel import select
+from sqlmodel.ext.asyncio.session import AsyncSession
 
 security = HTTPBearer()
 
 
 async def get_current_user(
     credentials: HTTPAuthorizationCredentials = Depends(security),
-    db: Session = Depends(get_db),
+    db: AsyncSession = Depends(get_db),
 ) -> DBUser:
     """获取当前认证用户"""
     token = credentials.credentials
@@ -27,9 +29,31 @@ async def get_current_user(
     return user
 
 
-async def get_current_user_by_token(token: str, db: Session) -> DBUser | None:
-    token_record = get_token_by_access_token(db, token)
+async def get_current_user_by_token(token: str, db: AsyncSession) -> DBUser | None:
+    token_record = await get_token_by_access_token(db, token)
     if not token_record:
         return None
-    user = db.exec(select(DBUser).where(DBUser.id == token_record.user_id)).first()
+    user = (
+        await db.exec(
+            select(DBUser)
+            .options(
+                joinedload(DBUser.lazer_profile),  # pyright: ignore[reportArgumentType]
+                joinedload(DBUser.lazer_counts),  # pyright: ignore[reportArgumentType]
+                joinedload(DBUser.daily_challenge_stats),  # pyright: ignore[reportArgumentType]
+                joinedload(DBUser.avatar),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.lazer_statistics),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.lazer_achievements),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.lazer_profile_sections),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.statistics),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.team_membership),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.rank_history),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.active_banners),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.lazer_badges),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.lazer_monthly_playcounts),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.lazer_previous_usernames),  # pyright: ignore[reportArgumentType]
+                selectinload(DBUser.lazer_replays_watched),  # pyright: ignore[reportArgumentType]
+            )
+            .where(DBUser.id == token_record.user_id)
+        )
+    ).first()
     return user