Python+SQLAlchemy操作MySQL遇2006错误：连接断开且无法关闭

我用Python+MySQL+SQLAlchemy维护公司代码，流程是从输入库拉取数据，对比日志库后推送到Jira自定义字段。代码逻辑顺序为：查询输入库创建临时表 → 执行更新逻辑 → 执行新增逻辑 → 执行禁用逻辑 → 执行启用逻辑。目前临时表、日志库和Jira的操作都能正常完成，但到禁用逻辑时触发MySQL 2006错误（输入库连接断开），导致禁用、启用逻辑无法执行。

已尝试的方案 #

• 每5分钟发送心跳包：能避免错误，但程序结束后心跳仍在运行，说明连接未真正关闭
• 查询后及程序结束时调用session.close()：后续代码会自动重连，问题依旧
• 调整pool_timeout等连接池参数：无效果
• 确认SQL相关设置为默认值：程序运行时长不足30分钟，远低于默认8小时超时阈值
• 临时修复：报错后重启连接可完成逻辑，但希望彻底解决

针对性解决建议 #

1. 用上下文管理器显式控制连接生命周期 #

通过with语句管理session，确保查询完成后立即释放连接，避免闲置超时：

from sqlalchemy.orm import sessionmaker

def session_scope():
    Session = sessionmaker(bind=engine)
    session = Session()
    try:
        yield session
        session.commit()
    except:
        session.rollback()
        raise
    finally:
        session.close()

# 使用示例
with session_scope() as session:
    # 执行输入库查询、创建临时表操作
    data = session.query(YourModel).all()
    # 临时表创建逻辑

2. 调整连接池pool_recycle参数 #

部分云数据库或自建库会设置更短的闲置断开时间（比如30分钟），将pool_recycle设为比该值短的时长，强制SQLAlchemy主动回收闲置连接：

from sqlalchemy import create_engine

engine = create_engine(
    "mysql+pymysql://user:password@host/db_name",
    pool_recycle=1500  # 25分钟，单位秒
)

3. 按需禁用连接池 #

如果是单任务脚本，不需要连接池复用，可直接关闭池功能，每次操作创建新连接：

from sqlalchemy import create_engine
from sqlalchemy.pool import NullPool

engine = create_engine(
    "mysql+pymysql://user:password@host/db_name",
    poolclass=NullPool
)

4. 拆分长逻辑，提前释放连接 #

检查禁用/启用逻辑是否有长时间阻塞的操作（比如大量数据处理、Jira API调用），可先将输入库数据加载到内存，关闭连接后再执行后续逻辑，避免连接长时间闲置。

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