前言:
当一个程序需要同时执行多个任务时,例如下发邮件、发送验证码等,在实际操作中可以使用异步执行的方式,Flask工程可以借助Celery+Redis结合使用。
前置
需要先安装redis、celery、flask
1.pip install flask
2.pip install redis
3.pip install celery
内容
将所有内容加添加到一个py文件里面(文件名称: task.py)
import time
from celery import Celery, platforms
from flask import Flask, jsonify
app = Flask(__name__)
redis_host = "10.122.2.68"
# 配置
app.config['CELERY_BROKER_URL'] = f'redis://{redis_host}:6379/0'
app.config['CELERY_RESULT_BACKEND'] = f'redis://{redis_host}:6379/0'
app.config['CELERY_TIMEZONE'] = 'Asia/Shanghai'
platforms.C_FORCE_ROOT = True # 避免权限问题
celery = Celery(app.name, broker=app.config['CELERY_BROKER_URL'], include=[app.name])
# 将Flask中的配置直接传递给Celery
celery.conf.update(app.config)
@app.route('/long_task', methods=['GET', 'POST'])
def long_task():
task = long_task.apply_async()
return jsonify({'task_id': task.id})
@app.route('/status/<task_id>')
def task_status(task_id):
"""
获取任务状态
:param task_id:
:return:
"""
task = long_task.AsyncResult(task_id)
if task.state == 'PENDING':
response = {
'state': task.state,
'current': 0,
'total': 100,
}
elif task.state != 'FAILURE':
response = {
'state': task.state,
'current': task.info.get('current', 0),
'total': task.info.get('total', 100),
}
else:
response = {
'state': task.state,
'current': 100,
'total': 100,
}
return jsonify(response)
@celery.task(name='app.long_task', bind=True)
def long_task(self):
for i in range(100):
self.update_state(state='PROGRESS', meta={'current': i, 'total': 100})
time.sleep(1)
return {'current': 100, 'total': 100}
if __name__ == "__main__":
app.run(port=5000, debug=True)
执行
1: 先将redis服务启动起来
2: 启动worker: celery -A task.celery worker -l info -P eventlet
3: 启动后端服务,直接运行代码