SQLAlchemy事务管理：多提交场景下全局回滚实现方案问询

原代码的问题分析 #

你当前的代码存在核心逻辑错误：with session.begin()已经帮你开启了一个顶级事务，但手动调用session.commit()会直接提交并终止这个顶级事务，后续的DB操作会自动开启新的独立事务。这就导致两次commit是完全分离的——如果第二次操作出错，第一次的提交已经生效，根本无法回滚。

正确的代码调整方案 #

要实现「一个事务内执行多步操作，任意一步出错则全部回滚」的需求，不需要手动调用session.commit()，而是利用SQLAlchemy的**保存点（Savepoint）**来管理事务内的子操作，同时依赖with session.begin()的上下文管理器自动处理全局的提交和回滚。

调整后的完整代码如下：

engine = create_engine(url="postgresql+psycopg2://postgres:0000@localhost:5432/db")
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

def get_database_session():
    try:
        session = SessionLocal()
        yield session
    finally:
        session.close()

app = FastAPI()

@app.post("/api/v1/users")
def add_user(data: AppUserCreateModel, session: Session = Depends(get_database_session)):
    # 开启顶级事务，上下文自动处理最终的提交/回滚
    with session.begin():
        # 第一步DB操作：用保存点管理
        sp1 = session.begin_nested()
        try:
            # 执行你的第一个DB操作（比如插入用户基础信息）
            # db call 1
            sp1.commit()  # 仅提交保存点，不结束顶级事务
        except Exception:
            sp1.rollback()  # 回滚到该保存点前的状态
            raise  # 抛出异常触发顶级事务全局回滚

        # 第二步DB操作：同样用保存点管理
        sp2 = session.begin_nested()
        try:
            # 执行第二个DB操作（比如关联用户角色）
            # db call 2
            sp2.commit()
        except Exception:
            sp2.rollback()
            raise
    # 上下文正常结束时自动提交顶级事务；若中间抛出异常则自动回滚所有操作

关键逻辑说明 #

1. 顶级事务的自动管理：with session.begin()会在进入上下文时开启事务，退出时（无异常）自动提交，遇到异常则自动回滚整个事务，完全替代手动commit/rollback。
2. 保存点的作用：session.begin_nested()创建的保存点，允许你在顶级事务内划分子操作单元。每个保存点的commit只是确认该子操作的有效性，但不会提交整个顶级事务；如果某个子操作出错，回滚到对应保存点后，抛出异常会触发顶级事务的全局回滚，确保所有已执行的子操作都被撤销。
3. 禁止手动提交顶级事务：绝对不要在with session.begin()的上下文内调用session.commit()，这会直接终止顶级事务，破坏全局回滚的逻辑。

内容的提问来源于stack exchange，提问作者Michael Xia