Python中Flask-SQLAlchemy使用db.session.commit()失败后事务会自动回滚吗?

Flask-SQLAlchemy (版本 2.3.0+) 事务提交,如果 db.session.commit() 失败,会自动回滚吗?

db.session.add(something)
db.session.commit()

有必要改成这样吗:

db.session.add(something)
try:
    db.session.commit()
except:
    db.session().rollback()


Python中Flask-SQLAlchemy使用db.session.commit()失败后事务会自动回滚吗?

12 回复

有必要


在Flask-SQLAlchemy中,如果你使用db.session.commit()提交事务时发生异常,事务不会自动回滚。你需要手动处理回滚,通常的做法是在异常捕获块中调用db.session.rollback()

这里有个典型的使用模式:

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///test.db'
db = SQLAlchemy(app)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True, nullable=False)

try:
    # 一些数据库操作
    new_user = User(username='test_user')
    db.session.add(new_user)
    
    # 尝试提交
    db.session.commit()
except Exception as e:
    # 发生异常时回滚
    db.session.rollback()
    print(f"操作失败,已回滚: {e}")
finally:
    # 确保session被正确关闭或重置
    db.session.remove()

关键点:

  1. commit()失败后,事务会保持在"脏"状态,必须显式调用rollback()来清理
  2. 如果不回滚,后续的数据库操作可能会失败
  3. 最佳实践是在try-except块中处理提交和回滚

简单说就是:Flask-SQLAlchemy不会帮你自动回滚,得自己写回滚代码。

有必要

你不会滚数据库也会超时, 所以看你的需求

用 with 语法不好么?

请使用 with 语法处理

SQLAlchemy 回滚之后如何再发起重试,保证业务能够实现??

#5
#4
说用 with 的人,知道 with 是干嘛用的吗?这里不是 session.begin(),真是张口就来啊。

哦!
我是建议用 with 来改写着部分代码,不要用 try-catch 模式。还是你觉得这样改写有什么问题?

不 rollback 你下次写的时候直接报错。

这两天刚好也在研究这个问题
首先 session.commit()失败后,数据库层面的事务会自动回滚,整个事务内的操作都不会写到磁盘中(在这个请求内调用 session 执行的所有查询写入操作都在这个事务内)。
但是 sqlalchemy 层面也维护了一套事务的逻辑,当前 session 开启的事务被回滚之后,如果你需要继续使用这个 session 发起新的事务(执行查询或其他操作),必须调用 session.rollback(这里我的理解是对这个 session 进行回滚)。如果你不再需要使用这个 session,可以直接 close()关闭 session.
因为 flask-sqlalchemy 会在每个请求上下文内自动创建销毁 session,所以结论是如果你需要在当前请求内继续使用 session,必须执行 session.rollback();如果确定不再需要这个 session 执行任何数据库操作了,可以不用 rollback,让它在请求结束时自动销毁。
当然,大部分情况下我们会加入 session.rollback 封装起来进行复用。
没看源码,根据文档和动手测试得出的结论,不一定准确

回到顶部