使用engine.begin()时出现MySQL连接丢失错误，求解决方案

先看你代码里的几个核心问题，这些是导致连接丢失的主要原因：

• 同时混用了自定义连接池和SQLAlchemy自带的连接池，两种连接管理逻辑冲突，容易导致连接状态混乱
• 变量名重复：先从自定义连接池拿了conn，后面with engine.begin() as conn又用了同一个变量名，导致原来的连接池连接被覆盖，finally里归还的可能不是正确的连接
• 异常捕获只针对psycopg2的错误，但你用的是MySQL+pymysql，根本捕获不到实际的连接错误，出问题也没法正确处理
• 每次请求都创建新的engine，SQLAlchemy的engine是重量级对象，频繁创建会导致连接管理失控

第一步：简化连接逻辑，去掉自定义连接池 #

SQLAlchemy的create_engine自带连接池管理，完全不需要自己额外搞连接池，直接用它的就行。修正后的代码大概是这样：

...
server = f"{_db.credentials.host}:{_db.credentials.port}"
database = _db.credentials.database
user = _db.credentials.user
password = _db.credentials.password

# 用f-string拼接更清晰
connection_url = f"mysql+pymysql://{user}:{password}@{server}/{database}"

# 把engine创建移到函数外部，全局只初始化一次，不要每次请求都创建
# engine = create_engine(connection_url, pool_pre_ping=True, pool_recycle=3600)

try:
    # 如果需要生成查询字符串，直接用engine的connect来生成，不需要自定义连接池的conn
    with engine.connect() as temp_conn:
        q = query.as_string(temp_conn)
    
    _logger.info(q)
    # 用engine直接执行查询，pd.read_sql_query可以直接用engine，不用手动开conn
    return pd.read_sql_query(q, engine)
except sqlalchemy.exc.OperationalError as ex:
    _logger.info(f"MySQL连接错误: {ex}")
    raise ex
except Exception as ex:
    _logger.info(f"其他数据库错误: {ex}")
    raise ex
...

第二步：给SQLAlchemy的engine配置连接保活参数 #

在创建engine的时候加上这两个参数，解决连接超时被MySQL服务器主动断开的问题：

• pool_pre_ping=True：每次从连接池拿连接前，先发送一个测试查询（比如SELECT 1），如果连接已经失效，会自动重新创建
• pool_recycle=3600：强制回收超过3600秒的连接，避免MySQL的wait_timeout设置导致连接失效（MySQL默认wait_timeout是8小时，你可以根据自己的数据库配置调整这个值，比如设为比wait_timeout小300秒）

创建engine的代码改成：

engine = create_engine(
    connection_url,
    pool_pre_ping=True,
    pool_recycle=3600,
    # 可以根据需求调整连接池大小
    pool_size=10,
    max_overflow=20
)

第三步：检查MySQL服务器的配置 #

如果上面的代码调整后还是有问题，去检查MySQL的这几个配置：

• wait_timeout和interactive_timeout：这两个参数控制连接空闲多久会被服务器断开，建议不要设置得太小，比如设为28800（8小时），如果你的服务有长时间空闲的情况，要确保SQLAlchemy的pool_recycle值比这个小
• max_connections：如果MySQL的最大连接数设得太小，也会导致拿不到连接，出现类似的错误，可以适当调大

额外注意点 #

• 不要在每次请求里创建engine，engine是全局单例的，初始化一次就行，频繁创建会导致大量无用连接，耗尽MySQL的连接资源
• 避免长时间占用连接：如果查询耗时很长，要确保MySQL的net_read_timeout和net_write_timeout足够大，避免服务器中途断开连接

内容的提问来源于stack exchange，提问作者David D'Silva