您可以并行运行多个 Flask SQLAlchemy 查询吗？

Question

在我当前的设置中，如果我执行五个 100 毫秒的查询，它们总共需要 500 毫秒。有没有办法可以并行运行它们，只需要 100 毫秒？

我在 nginx/uwsgi 后面运行 Flask，但可以更改其中任何一个。

具体来说，我希望能够从这里转换代码：

result_1 = db.session.query(...).all()
result_2 = db.session.query(...).all()
result_3 = db.session.query(...).all()

对于这样的事情：

result_1, result_2, result_3 = run_in_parallel([
  db.session.query(...).all(),
  db.session.query(...).all(),
  db.session.query(...).all(),
])

有没有办法用 Flask 和 SQLAlchemy 做到这一点？

Answer 1

一般的并行性

一般来说，如果你想并行运行任务，你可以使用线程或进程。在 python 中，线程非常适合 I/O 绑定的任务（意味着它们花费的时间花在等待另一个资源上——等待你的数据库、磁盘或远程网络服务器），而进程非常适合任务受 CPU 限制的（数学和其他计算密集型任务）。

并发期货

在您的情况下，线程是理想的。Python 有一个线程模块，您可以查看它，但有一些需要解压缩的内容：安全使用线程通常意味着通过使用线程池和任务队列来限制可以运行的线程数。出于这个原因，我更喜欢concurrent.futures库，它提供包装器threading，为您提供易于使用的界面并为您处理很多复杂性。

使用 时concurrent.futures，您将创建一个执行程序，然后将任务与参数列表一起提交给它。而不是像这样调用函数：

# get 4 to the power of 5
result = pow(4, 5)
print(result)

您提交函数及其参数：

您通常会像这样使用 concurrent.futures：

from concurrent.futures import ThreadPoolExecutor
executor = ThreadPoolExecutor()
future = executor.submit(pow, 4, 5)
print(future.result())

注意我们不是通过 using 调用函数pow()，而是提交pow执行器将在线程内调用的函数对象。

为了更轻松地将concurrent.futures库与 Flask一起使用，您可以使用flask-executor，它的工作方式与任何其他 Flask 扩展一样。它也处理的边缘情况下，你的后台任务需要访问瓶的上下文当地人（如app，session，g或request对象）一个后台任务内。完全披露：我编写并维护了这个库。

（有趣的事实：concurrent.futures使用相同的 API 包装线程和多处理 - 因此，如果您发现自己将来需要对 CPU 绑定任务进行多处理，您可以以相同的方式使用相同的库来实现您的目标）

把这一切放在一起

以下是flask-executor用于并行运行 SQLAlchemy 任务的内容：

from flask_executor import Executor
# ... define your `app` and `db` objects

executor = Executor(app)   

# run the same query three times in parallel and collect all the results
futures = []
for i in range(3):
    # note the lack of () after ".all", as we're passing the function object, not calling it ourselves
    future = executor.submit(db.session.query(MyModel).all) 
    futures.append(future)

for future in futures:
    print(future.result())

Boom，您现在已经并行运行了多个 Flask SQLAlchemy 查询。