mysql分布式事务处理

import pymysql
from pymysql.constants import CLIENT

def prepare_transaction(conn):
    with conn.cursor() as cursor:
        cursor.execute("PREPARE TRANSACTION 'my_transaction'")
    conn.commit()

def commit_transaction(conn):
    with conn.cursor() as cursor:
        cursor.execute("COMMIT PREPARED 'my_transaction'")
    conn.commit()

def rollback_transaction(conn):
    with conn.cursor() as cursor:
        cursor.execute("ROLLBACK PREPARED 'my_transaction'")
    conn.commit()

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

try:
    # 准备事务
    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()