mysql实现分布式事务管理

import mysql.connector
from mysql.connector import Error

def two_phase_commit(conn1, conn2, query1, query2):
    try:
        # 第一阶段：预提交
        cursor1 = conn1.cursor()
        cursor2 = conn2.cursor()
        cursor1.execute("PREPARE TRANSACTION '%s'" % query1)
        cursor2.execute("PREPARE TRANSACTION '%s'" % query2)

        # 第二阶段：提交
        conn1.commit()
        conn2.commit()
        print("Transaction committed successfully.")
    except Error as e:
        # 回滚
        conn1.rollback()
        conn2.rollback()
        print("Transaction failed and rolled back:", e)

# 连接数据库
conn1 = mysql.connector.connect(user='user1', password='password1', host='host1', database='db1')
conn2 = mysql.connector.connect(user='user2', password='password2', host='host2', database='db2')

# 执行事务
query1 = "UPDATE table1 SET column1 = 'value1' WHERE id = 1"
query2 = "UPDATE table2 SET column2 = 'value2' WHERE id = 1"
two_phase_commit(conn1, conn2, query1, query2)

# 关闭连接
conn1.close()
conn2.close()