在Python循环中执行DB2的SQL MERGE语句未生效的问题排查

看起来你遇到的问题是在Python循环里用SQLAlchemy执行DB2的MERGE语句时，修改没有生效——单独运行SQL完全正常，但嵌入Python循环后，redeemer列始终是0。我来帮你梳理几个最可能的原因和对应的解决办法：

1. 事务提交不及时（最核心的原因） #

SQLAlchemy的Connection对象默认是在事务上下文里运行的，每次execute执行的语句不会自动提交到数据库。也就是说，你的MERGE语句修改的数据只是在当前事务的临时状态中，没有真正写入数据库：

• 如果你用来验证的sum查询是在同一个未提交的事务里，可能因为DB2默认的READ COMMITTED隔离级别，看不到未提交的修改；
• 后续的UPDATE重置语句会在同一个事务里覆盖MERGE的结果，最后整个事务结束时，只有UPDATE的修改被保留下来。

解决办法：手动提交事务
每次执行完MERGE后，调用connection.commit()把修改提交到数据库，再进行后续操作：

for i in categories:
    # 执行MERGE更新
    connection.execute(
        f'''
        MERGE INTO TABLE_1 A
        USING
            (
            SELECT  ACCOUNT
            FROM    REDEEMERS
            WHERE   CATEGORY = '{i}'
            ) B
        ON      A.ACCOUNT = B.ACCOUNT
        WHEN MATCHED THEN UPDATE
        SET     A.REDEEMER = 1
        ;'''
    )
    connection.commit()  # 关键：提交MERGE的修改到数据库

    # 这里执行你的验证或业务逻辑，现在就能看到redeemer被更新为1的结果了

    # 重置redeemer为0，同样需要提交
    connection.execute('''UPDATE TABLE_1 SET REDEEMER = 0;''')
    connection.commit()

或者你也可以在创建连接时直接开启自动提交模式，省去手动commit的步骤：

# 创建连接时设置autocommit=True
connection = engine.connect().execution_options(autocommit=True)

2. Session使用方式错误 #

你尝试用Session管理事务，但代码存在明显的问题：你在with Session(connection) as session块里依然用connection.execute，而不是session.execute，而且session.commit()写在了with块外面——with块结束后Session已经被自动关闭，这时候commit完全无效。

正确的Session用法：

from sqlalchemy.orm import Session

categories = ["x", "y", "z"]

for i in categories:
    with Session(engine) as session:
        # 用session.execute执行SQL，推荐用参数绑定替代f-string
        session.execute(
            '''
            MERGE INTO TABLE_1 A
            USING
                (
                SELECT  ACCOUNT
                FROM    REDEEMERS
                WHERE   CATEGORY = :category
                ) B
            ON      A.ACCOUNT = B.ACCOUNT
            WHEN MATCHED THEN UPDATE
            SET     A.REDEEMER = 1
            ;''',
            {"category": i}  # 参数绑定：安全且避免字符串格式化问题
        )
        session.commit()  # 在with块内提交事务

        # 执行验证操作，比如查询sum
        sum_redeemer = session.execute('''SELECT SUM(redeemer) FROM TABLE_1''').scalar()
        print(f"当前分类 {i} 的redeemer总和：{sum_redeemer}")

        # 重置redeemer列
        session.execute('''UPDATE TABLE_1 SET REDEEMER = 0;''')
        session.commit()

这里额外提一句：用参数绑定（:category+字典传参）替代f-string拼接SQL，不仅能彻底避免SQL注入风险，还能避免字符串格式化时可能出现的引号、特殊字符问题，是更规范的数据库操作写法。

3. 其他可能性（概率较低） #

• 权限/约束问题：虽然你单独运行SQL正常，但Python连接数据库的用户是否有足够的MERGE权限？或者TABLE_1的account列是否有主键/唯一约束？如果account不唯一，MERGE可能出现匹配多条记录的异常，但这个可能性很低，因为你单独执行SQL是正常的。
• 事务隔离级别：如果数据库的隔离级别设置为REPEATABLE READ，可能会导致在同一个事务里看不到MERGE的修改，这种情况下只要确保每次MERGE后提交事务即可解决。

你可以先尝试第一种解决办法（手动commit），这是最常见的原因，应该能快速解决你的问题。

备注：内容来源于stack exchange，提问作者SRJCoding