Flask中简单实现异步任务，非常实用的小技巧

Posted 2021-04-12 会呼吸的Coder

众所周知Flask默认不能很好的完成异步任务，那么我们如何在请求中快速返回状态并异步执行比较复杂耗时的任务呢？

话不多说看实现。

用到了ThreadPoolExecutor类

初始化线程池（这里初始化了10个线程）

from flask import Flask
from concurrent.futures import ThreadPoolExecutor

executor = ThreadPoolExecutor(10)
app = Flask(__name__)

然后将封装一个任务函数，在路由函数中调用任务函数放入到线程池中即可

@app.route('/api/gen', methods=['POST'])
def gen():
    data_list = request.get_json().get('data')
    print(data_list )
    executor.submit(gen_task, (data_list))  # data_list为gen_task函数的参数
    return {}


def gen_task(data_list):
    time.sleep(3)
    print('start gen task...')
    for item in data_list:
        gen_records(item['code'], item['num'])
    print('finish.')

这种方法对原代码的改动非常小，却非常有效。