python - Flask-SQLAlchemy - session 如何与多个数据库一起工作？

Question

我正在做一个 Flask 项目，我正在使用 Flask-SQLAlchemy。
我需要使用多个现有的数据库。
我创建了“app”对象和 SQLAlchemy 对象:

from flask import Flask
from flask_sqlalchemy import SQLAlchemy


app = Flask(__name__)
db = SQLAlchemy(app)

在配置中，我设置了默认连接和附加绑定(bind):

SQLALCHEMY_DATABASE_URI = 'postgresql://pg_user:pg_pwd@pg_server/pg_db'
SQLALCHEMY_BINDS = {
    'oracle_bind': 'oracle://oracle_user:oracle_pwd@oracle_server/oracle_schema',
    'mssql_bind': 'mssql+pyodbc://msssql_user:mssql_pwd@mssql_server/mssql_schema?driver=FreeTDS'
}

然后我使用声明式系统创建了表模型，并在需要时设置__bind_key__ 参数来指示表所在的数据库。
例如:

class MyTable(db.Model):
    __bind_key__ = 'mssql_bind'
    __tablename__ = 'my_table'
    
    id = db.Column(db.Integer, nullable=False, primary_key=True)
    val = db.Column(db.String(50), nullable=False)

以这种方式，一切正常，当我进行查询时，它是在正确的数据库上进行的。

阅读 SQLAlchemy 文档和 Flask-SQLALchemy 文档我明白了这些事情(我把它们写下来以检查我是否理解正确):

• 您可以通过 session 处理交易。
• 在 SQLAlchemy 中，您可以将 session 与特定引擎绑定(bind)。
• Flask-SQLAlchemy 在请求开始时自动创建 session (scoped_session)，并在请求结束时销毁它
  

所以我可以这样做:

record = MyTable(1, 'some text')
db.session.add(record)
db.session.commit()

我不明白当我们在 Flask-SqlAlchemy 中使用多个数据库时，关于 session 会发生什么。

我验证了系统能够通过 __bind_key__ 参数在正确的数据库中正确绑定(bind)表，因此，我可以通过 db.session 在不同的数据库中插入数据，并在提交时保存所有内容。

但是，我无法理解 Flask-SQLAlchemy 是否创建了多个 session (每个引擎一个)或者是否以不同的方式管理事物。
在这两种情况下，如何引用特定数据库的 session /事务？
如果我使用 db.session.commit()，系统会在所有涉及的数据库上提交，但如果我只想提交一个数据库，我该怎么办？
我会做类似的事情:

db.session('mssql_bind').commit()

但我不知道该怎么做。

我还看到了一个 Flask-SQLAlchemy 实现，它应该可以简化这些情况的管理:

问题:https://github.com/mitsuhiko/flask-sqlalchemy/issues/107
实现:https://github.com/mitsuhiko/flask-sqlalchemy/pull/249

但我不知道如何使用它。

在 Flask-SQLAlchemy 中，我如何专门为每个引擎管理 session ？

Answer 1

Flask-SQLAlchemy 使用 customized session根据给定的 __bind_key__ attribute in mapped class 处理绑定(bind)路由.在幕后，它实际上将该键作为信息添加到创建的表中。换句话说，Flask 不会创建多个 session ，一个用于每个绑定(bind)，而是根据绑定(bind) key 路由到正确的可连接(引擎/连接)的单个 session 。请注意， Vanilla SQLAlchemy 有 similar functionality开箱即用。

In both cases, how is it possible refer to the session/transaction of a specific database? If I use db.session.commit() the system does the commit on all involved databases, but how can I do if I want to commit only for a single database?

在 session 中使用 session 拥有的连接来颠覆和提交对特定数据库的提交可能不是一个好主意。 session 是一个整体和keeps track of state对于对象实例，在需要时刷新对数据库的更改等。这意味着 session 处理的事务不仅是数据库事务，而且还是 session 自己的事务。所有这些都应该作为一个整体提交和回滚。

另一方面，您可以创建新的 SQLAlchemy(或 Flask-SQLAlchemy) session ，这可能是 join the ongoing transaction in one of the binds :

session = db.create_scoped_session(
    options=dict(bind=db.get_engine(app, 'oracle_bind'),
                 binds={}))

这就是拉取请求的内容。它允许使用现有的事务连接作为新 Flask-SQLAlchemy session 的 绑定(bind)。这在测试中非常有用，正如该拉取请求的基本原理所示。这样你就可以拥有一个“主”事务，例如可以回滚在测试中完成的所有事情。

请注意，如果存在 bind_key，SignallingSession 始终会查询 db.get_engine() 方法。这意味着示例 session 无法查询没有绑定(bind)键且不存在于您的 Oracle 数据库中的表，但仍可用于具有您的 mssql_bind 键的表。

另一方面，您链接到的问题确实列出了向特定绑定(bind)发出 SQL 的方法:

rows = db.session.execute(query, params,
                          bind=db.get_engine(app, 'oracle_bind'))

还列出了其他不太显式的方法，但显式优于隐式。