如何用SQLAlchemy关闭PostgreSQL会话？连接残留问题排查与解决

是否会造成问题？ #

少量留存的连接不一定有问题——SQLAlchemy异步引擎默认会维护连接池，这些空闲连接是用来复用的，能提升后续请求的连接效率。但如果连接数持续增长，超过PostgreSQL配置的max_connections阈值，就会直接导致新请求无法建立连接，抛出"too many connections"错误，影响业务正常运行。

问题原因 #

你的代码里，async with上下文管理器会自动关闭会话并将连接归还到连接池，手动调用session.close()属于重复操作，一般不会引发泄漏。真正可能的诱因：

• 连接池默认配置限制：create_async_engine默认用QueuePool，核心池大小pool_size=5，临时溢出连接max_overflow=10，最多会有15个连接。如果业务并发量超过这个数，或者会话没被正确释放，就会导致连接数超标。
• 会话未被正确消费：如果调用get_session的地方没有正确处理生成器（比如手动调用时没走完上下文），会导致会话无法关闭，连接一直被占用。

解决办法 #

1. 调整连接池配置
   显式指定连接池参数，按需限制连接数，同时避免空闲连接失效：

   db_engine = create_async_engine(
       '<DATABASE_URL>',
       echo=False,
       future=True,
       pool_size=5,  # 核心空闲连接数，根据业务并发调整
       max_overflow=5,  # 允许临时创建的额外连接数
       pool_recycle=3600,  # 1小时后回收空闲连接，避免数据库端主动断开
       pool_pre_ping=True  # 获取连接前先ping，确保连接可用
   )
   

2. 简化会话生成函数
   async with会自动管理会话生命周期，无需手动调用session.close()，修改后的代码更简洁安全：

   async def get_session() -> AsyncSession:
       async with async_session() as session:
           yield session
   

3. 排查连接泄漏场景

   • 如果是FastAPI这类框架，确保依赖注入的get_session被正确使用，每个请求的会话都会被框架自动释放。
   • 手动调用时，要确保完整消费生成器，比如用async for遍历：

     async def do_db_operation():
         async for session in get_session():
             # 执行查询、提交等操作
             await session.execute(...)
     

4. 监控数据库连接状态
   在PostgreSQL中执行以下命令，查看当前连接数和状态，区分空闲连接和活跃连接，判断是否真的存在泄漏：

   SELECT count(*) AS total_connections,
          sum(CASE WHEN state = 'idle' THEN 1 ELSE 0 END) AS idle_connections
   FROM pg_stat_activity;
   

内容的提问来源于stack exchange，提问作者tobias