使用带有异步 Flask 视图的 Flask-SQLAlchemy 时，“事务中空闲”PostgreSQL 查询

Question

我有一个遗留的 Flask 项目，它现在有一些异步视图，它们也使用 Flask-SQLAlchemy 进行同步数据库调用。 我的期望是这些将阻止调用，但它们应该可以正常工作。 但是，我注意到当在异步视图中使用 Flask-SQLAlchemy session 时，事务在数据库中被卡为“事务中的空闲”。

可以通过运行以下示例代码来重现此问题：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

app.config["SQLALCHEMY_DATABASE_URI"] = "postgresql://127.0.0.1:5432/temp"
app.config["SQLALCHEMY_ECHO"] = True
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False

db = SQLAlchemy(app)

class SomeModel(db.Model):
    """Simplest possible model."""
    id = db.Column(db.Integer, primary_key=True)

@app.route("/async")
async def hello_async():
    """Sample asynchronous view."""
    SomeModel.query.all()
    return "Hello"

@app.route("/sync")
def hello_sync():
    """Sample synchronous view."""
    SomeModel.query.all()
    return "Hello"

if __name__ == "__main__":
    db.create_all()
    app.run()

加载/sync端点后，终端 output 如下所示：

2022-09-26 09:28:21,300 INFO sqlalchemy.engine.Engine BEGIN (implicit)
2022-09-26 09:28:21,300 INFO sqlalchemy.engine.Engine SELECT some_model.id AS some_model_id
FROM some_model
2022-09-26 09:28:21,300 INFO sqlalchemy.engine.Engine [cached since 15.05s ago] {}
2022-09-26 09:28:21,303 INFO sqlalchemy.engine.Engine ROLLBACK
127.0.0.1 - - [26/Sep/2022 09:28:21] "GET /sync HTTP/1.1" 200 -

但是，当访问 /async 端点时，我看到：

2022-09-26 09:28:46,277 INFO sqlalchemy.engine.Engine BEGIN (implicit)
2022-09-26 09:28:46,277 INFO sqlalchemy.engine.Engine SELECT some_model.id AS some_model_id
FROM some_model
2022-09-26 09:28:46,277 INFO sqlalchemy.engine.Engine [cached since 40.03s ago] {}
127.0.0.1 - - [26/Sep/2022 09:28:46] "GET /async HTTP/1.1" 200 -

请注意，第二组日志缺少同步版本中发出的INFO sqlalchemy.engine.Engine ROLLBACK 。 因此，如果我查询 Postgres：

select query, state from pg_stat_activity where state = 'idle in transaction'

对于我向异步端点发出的每个请求，我都会看到一个空闲查询。 如果我向/async端点发出附加请求，这些查询最终会使 SQLAlchemy 连接池饱和，导致TimeoutError: QueuePool limit错误，应用程序将达到 500。

查看 Flask-SQLAlchemy 源代码，我可以看到ROLLBACK通常是由在@app.teardown_appcontext块中的这行代码上调用self.session.remove()发出的。 我可以确认在每个异步视图的末尾也调用了这一行，但它不会发出任何 SQL 或结束 session。

我的问题是：有没有办法在async Flask 视图中使用我现有的同步 Flask-SQLAlchemy session 并让它正确关闭 session？

Answer 1

对于后代，这是 Flask-SQLAlchemy 中的一个错误，已在 3.0.0 版中修复