SQLAlchemy中如何为PostgreSQL新会话触发执行运行时配置语句？

利用SQLAlchemy连接事件一次性配置连接 #

PostgreSQL的SET enable_bitmapscan TO off;是连接级别的配置，一旦在某个连接上执行，只要连接不被销毁，该配置会一直生效。SQLAlchemy的连接池会复用连接，所以我们只需要在新连接创建完成时执行一次该语句即可，无需每次创建会话都重复操作。

具体实现可以通过SQLAlchemy的event.listen监听Pool的connect事件：

from sqlalchemy import create_engine
from sqlalchemy import event

engine = create_engine("postgresql://user:password@host/dbname")

@event.listens_for(engine, "connect")
def set_disable_bitmapscan(dbapi_connection, connection_record):
    cursor = dbapi_connection.cursor()
    cursor.execute("SET enable_bitmapscan TO off;")
    cursor.close()

这段代码会在每个新的数据库连接建立时自动执行SET语句，连接池复用已有连接时不会重复触发，完美平衡了配置生效范围和性能开销。

备选：会话级初始化（若需更细粒度控制） #

如果确实需要针对每个会话单独配置（比如部分会话需要开启位图扫描），可以监听会话的after_begin事件，在事务开始前执行语句：

from sqlalchemy.orm import sessionmaker
from sqlalchemy import event

Session = sessionmaker(bind=engine)

@event.listens_for(Session, "after_begin")
def set_disable_bitmapscan_on_session(session, transaction, connection):
    connection.execute("SET enable_bitmapscan TO off;")

不过这种方式会在每次会话启动事务时执行语句，若连接被复用会重复操作，仅在特殊场景下使用。

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