PHP前端开发

使用Python Flask构建高效API应用并实现后台任务的方法

百变鹏仔 11小时前 #Python
文章标签 高效

在python中,有若干解决方案可以实现后台任务,比如celery或redis queue,都是实现任务队列系统的好方法。但使用这二者都比较麻烦。设想这样一个场景,我们用flask建立一个api,实现从一个终端调用后台任务,并用另一个终端停止后台任务。

使用Flask建立了一个简单的API,有两个主要的方法,一个用于启动后台任务,另一个用于停止。

为了管理任务的生命周期,我们使用Event Objects,这是一种简单的线程间通信机制。

如下所示是所需导入的库、线程事件声明和后台任务方法:

from time import sleepfrom flask import Flaskfrom flask_cors import CORSimport threadingthread_event = threading.Event()def backgroundTask():    while thread_event.is_set():        print('Background task running!')        sleep(5)

这里的关键是is_set()方法,它将返回内部线程事件标志的值:true或false。

立即学习“Python免费学习笔记(深入)”;

首先,使用set()方法把标志设置为true,它将启动一个线程,并连续运行backgroundTask方法。

@app.route("/start", methods=["POST"])def startBackgroundTask():    try:        thread_event.set()        thread = threading.Thread(target=backgroundTask)        thread.start()        return "Background task started!"    except Exception as error:        return str(error)

如果要停止任务,调用clear()方法将标志设置为false,以停止正在运行的线程。

@app.route("/stop", methods=["POST"])def stopBackgroundTask():    try:        thread_event.clear()        return "Background task stopped!"    except Exception as error:        return str(error)