fix(session): fix wrong usages of with_db

This commit is contained in:
MingxuanGame
2025-10-02 15:26:54 +00:00
parent 9a77c8d246
commit 2e1d922f59
2 changed files with 21 additions and 37 deletions

View File

@@ -135,8 +135,7 @@ class VerifySessionMiddleware(BaseHTTPMiddleware):
token = auth_header[7:] # 移除"Bearer "前缀
# 创建专用数据库会话
db = with_db()
try:
async with with_db() as db:
# 获取token记录
token_record = await get_token_by_access_token(db, token)
if not token_record:
@@ -145,8 +144,6 @@ class VerifySessionMiddleware(BaseHTTPMiddleware):
# 获取用户
user = (await db.exec(select(User).where(User.id == token_record.user_id))).first()
return user
finally:
await db.close()
except Exception as e:
logger.debug(f"[Verify Session Middleware] Error getting user: {e}")
@@ -163,8 +160,7 @@ class VerifySessionMiddleware(BaseHTTPMiddleware):
session_token = auth_header[7:]
# 获取数据库和Redis连接
db = with_db()
try:
async with with_db() as db:
redis = get_redis()
# 查找会话
@@ -173,8 +169,6 @@ class VerifySessionMiddleware(BaseHTTPMiddleware):
return None
return SessionState(session, user, redis, db)
finally:
await db.close()
except Exception as e:
logger.error(f"[Verify Session Middleware] Error getting session state: {e}")