Python + Flask 执行异步任务

在使用Flask开发python的web应用时,可能会遇到需要较长时间处理的任务,此时就需要使用异步的方式来实现,让长时间任务在后台运行,先将本次请求的响应状态返回给前端,不然前端界面"卡顿",当异步任务处理好后,如果需要返回状态,再将状态返回。

下面介绍两种常用的方式

(1)使用线程的方式

当要执行耗时任务时,直接开启一个新的线程来执行任务,这种方式最为简单快速。

通过ThreadPoolExecutor来实现

asyn_request_demo.py

# -*- coding: UTF-8 -*-
"""
# rs
"""
from logzero import logger
import time
from flask import Flask
from flask_restful import Api
from flask import jsonify
from flask import make_response
from concurrent.futures import ThreadPoolExecutor

executor = ThreadPoolExecutor()
# executor = ThreadPoolExecutor(10) 里面的数字是线程池所能同时进行的最大数量 
 
app = Flask(__name__)
api = Api(app)
 

def run(name, age):
    """
    # 异步任务
    """
    time.sleep(3)
    logger.info("name: %s, age: %s." % (name, age))
    logger.info("耗时任务执行结束")
 
 
@app.route('/test')
def test():
    # 传递多个参数
    args = ["rongsong", "28"]
    # 交给线程去处理耗时任务, 异步执行
    executor.submit(lambda p: run(*p), args)
    
    # 直接返回结果
    result = {
        "code": "200",
        "message": "success",
        "data": "green"
    }
    return make_response(jsonify(result))

 
if __name__ == "__main__":
    # 启动一个HTTP服务
    app.run(host="0.0.0.0", port=8800, debug=True)
  • 使用pip3安装相应的库(不再赘述)。
  • 使用python3  asyn_request_demo.py运行启动即可(启动端口8800,可修改)。
  • 然后访问http://{执行机器所在IP}:8800/test观察。
  • 预期可以看到:接口直接返回result结果(同步),后台日志等待5s左右打印日志"耗时任务执行结束"(异步)。

当要执行一些比较简单的耗时任务时可以使用这种方式,如发邮件、发送短信验证码等。

当这种方式有个很明显的问题,就是我们无法得到任务的执行状态

如果想要随时得到任务的执行状态,就需要设计一些逻辑,比如奖任务执行状态存储到Redis中,通过唯一的任务id进行标识,然后再写一个接口通过任务id去获取任务的状态,然后让前端定时的去请求该接口,从而获得任务状态信息。

是不是有点复杂,如果让我们自己实现的话,哈哈哈,别慌,Celery框架刚好实现了这样的逻辑,比较成熟,一般我们遇到问题时都可以先看看有没有现成的框架和方法,毕竟站在巨人的肩膀上我们可以看的更远。

(2)使用Celery的方式

Celery是定时任务处理和调度的分布式任务队列,常用于web异步任务、定时任务等。

你可能感兴趣的:(#,Python技术学习,python,flask,异步接口)