多次回滚Python SQLite会话引发Docker容器崩溃，如何解决？

问题根源 #

你当前的代码存在两个核心问题：

1. 错误地将sessionmaker生成的会话工厂直接当作会话实例使用，全局共用同一个工厂对象执行回滚操作，并行请求时会导致会话状态混乱。
2. 没有为每个请求分配独立的数据库会话，多个请求共享同一会话连接，多次回滚操作会触发SQLite的锁冲突或连接状态异常，最终导致容器崩溃。

具体解决方案 #

1. 为每个请求创建独立会话（核心修复）

利用Flask的请求上下文对象g，在请求开始时创建专属会话，请求结束后统一处理回滚/提交并关闭会话：

from flask import g
from sqlalchemy.orm import sessionmaker

# 全局仅初始化一次会话工厂（绑定数据库引擎）
SessionFactory = sessionmaker(bind=self.db)

@app.before_request
def init_session():
    # 每个请求创建独立会话，存储到请求上下文的g对象中
    g.db_session = SessionFactory()

@app.after_request
def handle_session(response):
    try:
        # 根据响应状态码处理会话
        if response.status_code not in [200, 201, 202]:
            g.db_session.rollback()
        else:
            g.db_session.commit()
    except Exception as e:
        # 捕获会话处理时的异常，避免崩溃
        g.db_session.rollback()
    finally:
        # 无论成功失败，必须关闭会话释放连接
        g.db_session.close()
    return response

2. 修复视图函数中的会话使用方式

所有视图函数中必须使用g.db_session作为当前请求的会话实例，禁止使用全局会话对象：

@app.route('/api/example', methods=['POST'])
def example_api():
    try:
        # 使用请求专属会话操作数据库
        g.db_session.add(YourModel(...))
        # 业务逻辑处理
        return {"status": "ok"}, 200
    except Exception as e:
        # 主动抛出异常，交给after_request处理回滚
        return {"error": str(e)}, 500

3. 优化SQLite连接配置

SQLite默认的文件锁机制对并行写入不友好，可在初始化引擎时添加超时配置，避免锁等待导致的崩溃：

from sqlalchemy import create_engine

# 添加超时配置，允许等待锁30秒
engine = create_engine('sqlite:///your_db.db', connect_args={"timeout": 30})

关键注意事项 #

• 绝对禁止全局共用会话实例，必须保证每个请求的会话独立。
• after_request中必须用finally块关闭会话，避免连接泄漏。
• 视图函数中捕获业务异常时，不要自行回滚，统一交给after_request处理，避免重复操作。

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