如何在Snowflake中以事务方式执行两次session.write_pandas操作？

这个问题我之前也碰到过！核心原因其实是write_pandas默认会自动提交事务——这就导致你第一个写入操作完成后，哪怕后面抛出异常，数据已经被持久化到Snowflake里了。而你用原生SQL执行insert的时候，是手动控制提交时机的，所以事务能按预期的原子性工作。

要解决这个问题，关键就是手动接管事务的控制权，关闭write_pandas的自动提交功能，然后统一管理提交和回滚：

具体解决方案步骤： #

1. 给两次write_pandas调用都加上auto_commit=False参数，让它们不会自动提交事务；
2. 用try-except块包裹两次写入操作，确保任何一步失败时都能回滚整个事务，只有全部成功时才提交。

示例代码： #

# 先提交当前可能存在的未完成事务，开启一个全新的事务上下文
session.sql("COMMIT").collect()

try:
    # 第一次写入：关闭自动提交
    session.write_pandas(
        df1, "table1", "db", "schema", 
        auto_create_table=True, auto_commit=False
    )
    # 第二次写入：同样关闭自动提交
    session.write_pandas(
        df2, "table2", "db", "schema", 
        auto_create_table=True, auto_commit=False
    )
    # 两次写入都成功，才手动提交事务
    session.sql("COMMIT").collect()
    print("两次写入都成功提交啦！")
except Exception as e:
    # 任何一步出错，立刻回滚整个事务
    session.sql("ROLLBACK").collect()
    print(f"事务执行失败，已回滚：{str(e)}")
    raise e  # 重新抛出异常，方便你排查具体问题

为什么这样能行？ #

当你把auto_commit=False传给write_pandas时，这个写入操作会被纳入当前的事务上下文，而不是立刻提交。直到你手动执行COMMIT，所有的操作才会被持久化到Snowflake；如果中间任何一步抛出异常，ROLLBACK会把两次写入操作全部撤销，保证两个表要么都更新成功，要么都不更新，完美实现事务的原子性。

对比你的原始代码 #

你之前的write_pandas调用没加auto_commit=False，默认每次调用都会自动执行COMMIT，所以第一个写入完成后事务就已经生效了，后面的异常根本没法回滚这一步——这就是为什么你会看到table1里已经有数据了。而原生SQL的insert操作默认不会自动提交，所以事务能被你手动控制。

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