mysql 分布式transaction

import psycopg2
from psycopg2 import sql

class Coordinator:
    def __init__(self, db1_conn, db2_conn):
        self.db1_conn = db1_conn
        self.db2_conn = db2_conn

    def prepare(self):
        with self.db1_conn.cursor() as cursor:
            cursor.execute("PREPARE TRANSACTION 'tx1'")
        with self.db2_conn.cursor() as cursor:
            cursor.execute("PREPARE TRANSACTION 'tx2'")

    def commit(self):
        with self.db1_conn.cursor() as cursor:
            cursor.execute("COMMIT PREPARED 'tx1'")
        with self.db2_conn.cursor() as cursor:
            cursor.execute("COMMIT PREPARED 'tx2'")

    def rollback(self):
        with self.db1_conn.cursor() as cursor:
            cursor.execute("ROLLBACK PREPARED 'tx1'")
        with self.db2_conn.cursor() as cursor:
            cursor.execute("ROLLBACK PREPARED 'tx2'")

# 示例使用
db1_conn = psycopg2.connect(database="db1", user="user1", password="pass1", host="host1", port="5432")
db2_conn = psycopg2.connect(database="db2", user="user2", password="pass2", host="host2", port="5432")

coordinator = Coordinator(db1_conn, db2_conn)
try:
    coordinator.prepare()
    coordinator.commit()
except Exception as e:
    coordinator.rollback()