针对 SQLAlchemy 异步会话工厂 async_session 的优化方案

以下是针对 SQLAlchemy 异步会话工厂 async_session 的优化方案，结合最佳实践和性能考量：

---

优化后的代码

from sqlalchemy.ext.asyncio import async_sessionmaker, AsyncSession

async_session = async_sessionmaker(
    bind=engine,                  # 绑定异步引擎
    autoflush=False,              # 关闭自动刷新以提高批量操作性能
    expire_on_commit=False,       # 提交后不使对象过期，避免重复查询
    class_=AsyncSession,          # 显式指定异步会话类（可选，默认已包含）
    future=True                   # 启用 SQLAlchemy 2.0 兼容模式（如果引擎未默认开启）
)

---

优化点解析

1. 使用 async_sessionmaker 替代 sessionmaker

• 原因 ：async_sessionmaker 是 SQLAlchemy 专门为异步会话设计的工厂函数，内部优化了异步上下文管理逻辑。

• 原代码问题 ：

  # 原代码（适用于同步会话）
  from sqlalchemy.orm import sessionmaker
  async_session = sessionmaker(..., class_=AsyncSession)

• 优化后 ：

  # 优化代码（专为异步设计）
  from sqlalchemy.ext.asyncio import async_sessionmaker
  async_session = async_sessionmaker(...)

2. 关键参数优化

参数	推荐值	说明
autoflush	False	关闭自动刷新，减少不必要的数据库交互，适合批量操作或复杂事务场景。
expire_on_commit	False	提交后保留对象状态，避免后续访问属性时触发隐式查询。
future	True	启用 SQLAlchemy 2.0 兼容模式（需确保引擎也启用 future=True）。

3. 绑定引擎明确化

• 显式指定 bind=engine
  避免歧义，明确会话工厂与异步引擎的绑定关系。

4. 会话类显式声明（可选）

• class_=AsyncSession
  虽然 async_sessionmaker 默认使用 AsyncSession，显式声明可增强代码可读性，避免隐式依赖。

---

完整使用示例

from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
from sqlalchemy.orm import declarative_base

# 1. 创建异步引擎（优化引擎配置）
engine = create_async_engine(
    "postgresql+asyncpg://user:password@localhost/db",
    echo=True,                   # 开发时开启 SQL 日志
    pool_size=10,                # 连接池大小
    max_overflow=5,              # 允许临时超出连接池的数量
    pool_recycle=3600,           # 连接回收时间（秒）
    future=True                  # 启用 2.0 模式
)

# 2. 定义模型基类
Base = declarative_base()

# 3. 优化后的会话工厂
async_session = async_sessionmaker(
    bind=engine,
    autoflush=False,
    expire_on_commit=False
)

# 4. 使用示例（推荐通过上下文管理器管理会话）
async def get_db() -> AsyncGenerator[AsyncSession, None]:
    async with async_session() as session:
        yield session

---

优化效果总结

优化方向	具体改进
性能	减少自动刷新（autoflush=False）降低数据库负载，适合高频或批量操作。
资源管理	使用 async_sessionmaker 确保异步上下文兼容性，避免会话泄漏。
代码可维护性	显式参数配置提升可读性，减少因隐式行为导致的调试成本。
兼容性	启用 future=True 为 SQLAlchemy 2.0 迁移铺平道路。

---

常见问题排查

1. 会话未正确关闭

   • 现象：数据库连接池耗尽或超时。

   • 解决 ：始终通过 async with 或 finally 块确保会话关闭：

     async with async_session() as session:
         await session.execute(...)

2. 事务未提交/回滚

   • 现象：数据未持久化或锁未释放。
   • 解决 ：显式调用 await session.commit() 或在依赖项中集成自动提交逻辑（参考前文 get_apyyacr_db）。

3. 引擎配置不匹配

   • 现象 ：future=True 未启用导致兼容性问题。
   • 解决 ：确保引擎和会话工厂均启用 future=True。

---

通过上述优化，可以显著提升异步数据库会话的性能和可维护性，同时为复杂业务场景提供灵活的事务控制能力。