Django Celery异步任务队列的实现


Posted in Python onJuly 24, 2019

背景

在开发中,我们常常会遇到一些耗时任务,举个例子:

上传并解析一个 1w 条数据的 Excel 文件,最后持久化至数据库。

在我的程序中,这个任务耗时大约 6s,对于用户来说,6s 的等待已经是个灾难了。

比较好的处理方式是:

  1. 接收这个任务的请求
  2. 将这个任务添加到队列中
  3. 立即返回「操作成功,正在后台处理」的字样
  4. 后台消费这个队列,执行这个任务

我们按照这个思路,借助 Celery 进行实现。

实现

本文所使用的环境如下:

  • Python 3.6.7
  • RabbitMQ 3.8
  • Celery 4.3

使用 Docker 安装 RabbitMQ

Celery 依赖一个消息后端,可选方案有 RabbitMQ, Redis 等,本文选用 RabbitMQ 。

同时为了安装方便,RabbitMQ 我直接使用 Docker 安装:

docker run -d --name anno-rabbit -p 5672:5672 rabbitmq:3

启动成功后,即可通过 amqp://localhost 访问该消息队列。

安装并配置 Celery

Celery 是 Python 实现的工具,安装可以直接通过 Pip 完成:

pip install celery

同时假设当前我的项目文件夹为 proj ,项目名为 myproj ,应用名为 myapp

安装完成后,在 proj/myproj/ 路径下创建一个 celery.py 文件,用来初始化 Celery 实例:

proj/myproj/celery.py

from __future__ import absolute_import, unicode_literals
import os
from celery import Celery, platforms

# set the default Django settings module for the 'celery' program.
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'myproj.settings')

app = Celery('myproj',
       broker='amqp://localhost//',
       backend='amqp://localhost//')

# Using a string here means the worker don't have to serialize
# the configuration object to child processes.s
# - namespace='CELERY' means all celery-related configuration keys
#  should have a `CELERY_` prefix.
app.config_from_object('django.conf:settings', namespace='CELERY')

# Load task modules from all registered Django app configs.
app.autodiscover_tasks()

然后在 proj/myproj/__init__.py 中添加对 Celery 对象的引用,确保 Django 启动后能够初始化 Celery:

proj/myproj/__init__.py

from __future__ import absolute_import, unicode_literals

# This will make sure the app is always imported when
# Django starts so that shared_task will use this app.
from .celery import app as celery_app

__all__ = ('celery_app',)

无其他特殊配置的话,Celery 的基本配置就是这些。

编写一个耗时任务

为了模拟一个耗时任务,我们直接创建一个方法,使其「睡」10s ,并将其设置为 Celery 的任务:

proj/myapp/tasks.py

import time
from myproj.celery import app as celery_app

@celery_app.task
def waste_time():
  time.sleep(10)
  return "Run function 'waste_time' finished."

启动 Celery Worker

Celery 配置完成,并且任务创建成功后,我们以异步任务的模式启动 Celery :

celery -A myproj worker -l info

注意到我强调了异步模式,是因为 Celery 除了支持异步任务,还支持定时任务,因此启动时候要指明。

同时要注意,Celery 一旦启动,对 Task(此处为 waste_time) 的修改必须重启 Celery 才会生效。

任务调用

在请求处理的逻辑代码中,调用上面创建好的任务:

proj/myapp/views.py

from django.http import JsonResponse
from django.views.decorators.http import require_http_methods
from .tasks import waste_time

@require_http_methods(["POST"])
def upload_files(request):
  waste_time.delay()
  # Status code 202: Accepted, 表示异步任务已接受,可能还在处理中
  return JsonResponse({"results": "操作成功,正在上传,请稍候..."}, status=202)

调用 waste_time.delay() 方法后, waste_time 会被加入到任务队列中,等待空闲的 Celery Worker 调用。

效果

当我们发送请求时,这个接口会直接返回 {"results": "操作成功,正在上传,请稍候..."} 的响应内容而非卡住十秒,用户体验要好许多。

总结

用 Celery 处理这种异步任务是 Python 常用的方法,虽然实际执行成功耗时不变甚至有所增加(如 Worker 繁忙导致处理滞后),但是对于用户体验来说更容易接受,点击上传大文件后可以继续处理其他事务,而不需要在页面等待。
Celery 还有更多用法本文未介绍到,其文档已经非常详尽,有需要可直接参考。

参考

http://docs.celeryproject.org/en/latest/django/first-steps-with-django.html

https://hub.docker.com/_/rabbitmq

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python中查看变量内存地址的方法
May 05 Python
python实现爬虫下载美女图片
Jul 14 Python
Python中实现switch功能实例解析
Jan 11 Python
urllib和BeautifulSoup爬取维基百科的词条简单实例
Jan 17 Python
使用python 3实现发送邮件功能
Jun 15 Python
详解Django+Uwsgi+Nginx 实现生产环境部署
Nov 06 Python
Python使用sqlalchemy模块连接数据库操作示例
Mar 13 Python
Python面向对象实现一个对象调用另一个对象操作示例
Apr 08 Python
Python图片处理模块PIL操作方法(pillow)
Apr 07 Python
Django框架配置mysql数据库实现过程
Apr 22 Python
python中scipy.stats产生随机数实例讲解
Feb 19 Python
Python实战之OpenCV实现猫脸检测
Jun 26 Python
python如何统计代码运行的时长
Jul 24 #Python
Django时区详解
Jul 24 #Python
详解Django定时任务模块设计与实践
Jul 24 #Python
Python3中urlencode和urldecode的用法详解
Jul 23 #Python
对python3中的RE(正则表达式)-详细总结
Jul 23 #Python
python正则表达式匹配不包含某几个字符的字符串方法
Jul 23 #Python
python使用百度文字识别功能方法详解
Jul 23 #Python
You might like
基于php中使用excel的简单介绍
2013/08/02 PHP
php打包网站并在线压缩为zip
2016/02/13 PHP
php5.x禁用eval的操作方法
2018/10/19 PHP
Javascript Tab 导航插件 (23个)
2009/06/11 Javascript
JavaScript学习笔记(二) js对象
2011/10/25 Javascript
利用谷歌地图API获取点与点的距离的js代码
2012/10/11 Javascript
extjs ColumnChart设置不同的颜色实现代码
2013/05/17 Javascript
JavaScript排序算法之希尔排序的2个实例
2014/04/04 Javascript
js+CSS实现弹出居中背景半透明div层的方法
2015/02/26 Javascript
js实现点击按钮后给Div图层设置随机背景颜色的方法
2015/05/06 Javascript
深入理解JS函数的参数(arguments)的使用
2016/05/28 Javascript
利用React-router+Webpack快速构建react程序
2016/10/27 Javascript
动手写一个angular版本的Message组件的方法
2017/12/16 Javascript
微信小程序录音与播放录音功能
2017/12/25 Javascript
webstorm添加*.vue文件支持
2018/05/08 Javascript
npm 更改默认全局路径以及国内镜像的方法
2018/05/16 Javascript
npm配置国内镜像资源+淘宝镜像的方法
2018/09/07 Javascript
基于Koa(nodejs框架)对json文件进行增删改查的示例代码
2019/02/02 NodeJs
layui实现数据分页功能(ajax异步)
2019/07/27 Javascript
基于Python的身份证号码自动生成程序
2014/08/15 Python
python中通过预先编译正则表达式提高效率
2017/09/25 Python
在Pycharm中项目解释器与环境变量的设置方法
2018/10/29 Python
python3发送邮件需要经过代理服务器的示例代码
2019/07/25 Python
解决django 新增加用户信息出现错误的问题
2019/07/28 Python
win10下python2和python3共存问题解决方法
2019/12/23 Python
python pyenv多版本管理工具的使用
2019/12/23 Python
在PyCharm中实现添加快捷模块
2020/02/12 Python
python 自定义异常和主动抛出异常(raise)的操作
2020/12/11 Python
意大利一家专营包包和配饰的网上商店:Borse Last Minute
2019/08/26 全球购物
关于雷锋的演讲稿
2014/05/10 职场文书
司法所长先进事迹
2014/06/02 职场文书
班训口号大全
2014/06/18 职场文书
项目安全员岗位职责
2015/02/15 职场文书
学校运动会通讯稿
2015/07/18 职场文书
《攀登者》:“海拔8000米以上,你不能指望任何人”
2019/11/25 职场文书
JDBC连接的六步实例代码(与mysql连接)
2021/05/12 MySQL