mysql支持分布式事务

import pymysql
import time

def prepare_transaction(conn):
    cursor = conn.cursor()
    cursor.execute("PREPARE TRANSACTION 'tx1'")
    cursor.close()

def commit_transaction(conn):
    cursor = conn.cursor()
    cursor.execute("COMMIT PREPARED 'tx1'")
    cursor.close()

def rollback_transaction(conn):
    cursor = conn.cursor()
    cursor.execute("ROLLBACK PREPARED 'tx1'")
    cursor.close()

# 连接数据库
conn1 = pymysql.connect(host='db1_host', user='user', password='password', db='db1')
conn2 = pymysql.connect(host='db2_host', user='user', password='password', db='db2')

try:
    # 开始事务
    conn1.begin()
    conn2.begin()

    # 执行SQL操作
    cursor1 = conn1.cursor()
    cursor1.execute("UPDATE table1 SET column1 = 'value1' WHERE id = 1")
    cursor1.close()

    cursor2 = conn2.cursor()
    cursor2.execute("UPDATE table2 SET column2 = 'value2' WHERE id = 2")
    cursor2.close()

    # 准备提交
    prepare_transaction(conn1)
    prepare_transaction(conn2)

    # 提交事务
    commit_transaction(conn1)
    commit_transaction(conn2)

except Exception as e:
    print(f"Transaction failed: {e}")
    rollback_transaction(conn1)
    rollback_transaction(conn2)

finally:
    conn1.close()
    conn2.close()