e71181f0a3
- 修复前端路由守卫：未登录时不显示提示，直接跳转登录页 - 修复API拦截器：401错误不显示提示，直接跳转 - 增强验证码显示：图片尺寸从120x40增加到200x80 - 增大验证码字体：从28号增加到48号 - 优化验证码字符：排除易混淆的0和1 - 减少干扰线：从5条减少到3条，添加背景色优化 - 增强登录API日志：添加详细的调试日志 - 增强验证码生成和验证日志 - 优化异常处理和错误追踪 影响文件： - src/router/index.ts - src/api/request.ts - app/services/auth_service.py - app/api/v1/auth.py - app/schemas/user.py 测试状态： - 前端构建通过 - 后端语法检查通过 - 验证码显示效果优化完成 Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
71 lines
1.6 KiB
Python
71 lines
1.6 KiB
Python
"""
|
||
数据库会话管理
|
||
"""
|
||
from typing import AsyncGenerator
|
||
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
|
||
from app.core.config import settings
|
||
from app.db.base import Base
|
||
|
||
# 创建异步引擎
|
||
engine = create_async_engine(
|
||
settings.DATABASE_URL,
|
||
echo=settings.DATABASE_ECHO,
|
||
pool_pre_ping=True,
|
||
pool_size=50, # 从20增加到50，提高并发性能
|
||
max_overflow=10, # 从0增加到10，允许峰值时的额外连接
|
||
)
|
||
|
||
# 创建异步会话工厂
|
||
async_session_maker = async_sessionmaker(
|
||
engine,
|
||
class_=AsyncSession,
|
||
expire_on_commit=False,
|
||
autocommit=False,
|
||
autoflush=False,
|
||
)
|
||
|
||
|
||
async def get_db() -> AsyncGenerator[AsyncSession, None]:
|
||
"""
|
||
获取数据库会话
|
||
|
||
Yields:
|
||
AsyncSession: 数据库会话
|
||
"""
|
||
async with async_session_maker() as session:
|
||
try:
|
||
yield session
|
||
await session.commit()
|
||
except Exception:
|
||
await session.rollback()
|
||
raise
|
||
finally:
|
||
await session.close()
|
||
|
||
|
||
async def init_db() -> None:
|
||
"""
|
||
初始化数据库（创建所有表）
|
||
注意：生产环境应使用Alembic迁移
|
||
"""
|
||
async with engine.begin() as conn:
|
||
# 导入所有模型以确保它们被注册
|
||
from app.models import user, asset, device_type, organization
|
||
|
||
# 创建所有表
|
||
await conn.run_sync(Base.metadata.create_all)
|
||
|
||
|
||
async def close_db() -> None:
|
||
"""关闭数据库连接"""
|
||
await engine.dispose()
|
||
|
||
|
||
__all__ = [
|
||
"engine",
|
||
"async_session_maker",
|
||
"get_db",
|
||
"init_db",
|
||
"close_db",
|
||
]
|