分布式事务年末特惠

class Saga:
    def __init__(self):
        self.steps = []
        self.compensations = []

    def add_step(self, action, compensation):
        self.steps.append(action)
        self.compensations.append(compensation)

    def execute(self):
        for i, step in enumerate(self.steps):
            try:
                step()
            except Exception as e:
                for comp in reversed(self.compensations[:i]):
                    comp()
                raise e

# 使用示例
def create_order():
    print("Creating order...")
    # 实际业务逻辑

def reserve_inventory():
    print("Reserving inventory...")
    # 实际业务逻辑

def cancel_order():
    print("Canceling order...")
    # 实际业务逻辑

def release_inventory():
    print("Releasing inventory...")
    # 实际业务逻辑

saga = Saga()
saga.add_step(create_order, cancel_order)
saga.add_step(reserve_inventory, release_inventory)

try:
    saga.execute()
except Exception as e:
    print(f"Transaction failed: {e}")