理解SQLAlchemy的连接池机制
在使用sqlalchemy连接postgresql等关系型数据库时，开发者可能会观察到即使在代码中明确“关闭”了会话，数据库端仍然显示有活跃的连接。这并非错误，而是sqlalchemy连接池（connection pooling）机制的正常行为。
连接池的核心思想是为了提高数据库操作的效率和性能。每次建立新的数据库连接都是一个相对耗时的操作。通过维护一个预先建立好的连接池，SQLAlchemy可以在需要时从池中获取连接，使用完毕后再将连接返回池中，而不是立即关闭。这样，后续的请求可以直接复用现有连接，避免了频繁地创建和销毁连接的开销。
默认情况下，SQLAlchemy的连接池会保持一定数量的连接处于开放状态，以便快速响应新的会话请求。对于异步引擎（如create_async_engine），这个默认的池大小通常是5个连接。
正确处理异步会话的关闭
在SQLAlchemy的异步编程模型中，推荐使用上下文管理器（async with语句）来管理会话的生命周期。这种方式能够确保会话在使用完毕后被正确地处理，包括将其关联的连接返回到连接池。
考虑以下代码示例：
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine, async_sessionmaker
# 数据库引擎配置，echo=False可避免打印SQL日志，future=True使用2.0风格API
db_engine = create_async_engine('', echo=False, future=True)
# 会话工厂配置，expire_on_commit=False表示提交后对象不会过期
async_session = async_sessionmaker(db_engine, class_=AsyncSession, expire_on_commit=False)
async def get_session() -> AsyncSession:
async with async_session() as session:
yield session
# 注意：此处不需要手动调用 await session.close()
# 上下文管理器会在退出时自动处理会话关闭和连接返回连接池 关键点： 当您使用async with async_session() as session:这样的上下文管理器时，session.close()方法会在with块结束时自动被调用。这意味着您不需要在yield session之后显式地添加await session.close()。手动调用它不仅是多余的，有时还可能导致意想不到的行为，因为连接池已经准备好接收连接。
配置连接池大小 (pool_size)
虽然默认的连接池大小（通常为5）适用于许多场景，但在高并发或特定性能要求的应用中，您可能需要调整连接池的大小。这可以通过在创建异步引擎时传递pool_size参数来实现。
pool_size参数定义了连接池中维护的连接数，这些连接即使在空闲时也会保持打开状态。
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine, async_sessionmaker
# 配置连接池大小为10
# pool_size参数应传递给 create_async_engine
db_engine = create_async_engine(
'',
echo=False,
future=True,
pool_size=10 # 设置连接池大小
)
async_session = async_sessionmaker(db_engine, class_=AsyncSession, expire_on_commit=False)
async def get_session() -> AsyncSession:
async with async_session() as session:
yield session 注意事项：
- pool_size的合理值： 过大的pool_size会占用更多数据库资源，而过小则可能导致连接等待时间增加。最佳值取决于您的应用负载、数据库服务器能力以及并发需求。通常建议通过性能测试来确定最合适的pool_size。
- 其他连接池参数： 除了pool_size，create_async_engine还支持其他连接池相关参数，如max_overflow（当所有池内连接都被占用时，允许额外创建的连接数）和pool_timeout（获取连接的超时时间）。详细信息请参考SQLAlchemy官方文档。
总结
SQLAlchemy通过其智能的连接池机制，优化了数据库连接的创建和管理，从而提高了应用程序的性能和响应速度。当您看到PostgreSQL中有一些连接保持开放时，这通常是连接池在正常工作，而非连接泄露问题。
关键要点：
- 连接池是性能优化： SQLAlchemy默认使用连接池来复用数据库连接，减少开销。
- 上下文管理器自动关闭： 使用async with async_session() as session:时，session.close()会自动被调用，无需手动操作。
- 配置pool_size： 可以通过create_async_engine的pool_size参数调整连接池的大小，以适应不同的应用需求。
理解并正确配置这些机制，将有助于您构建更健壮、高效的SQLAlchemy异步应用程序。
