django通过celery添加异步任务
异步任务的重要性
大家在做web项目的时候经常会遇到一些耗时的操作, 比如: 发送邮件、发送短信、生成pdf。这些操作在某些情况下需要立即返回结果给用户,但是可以在后台异步执行。
比如用户邮箱注册的时候, 在发送邮件的时候可以先把”已经发送激活邮件到邮箱”返回给用户, 同时把邮件发送任务提交到异步处理线程中。
现在介绍一款python写的专门用于处理异步任务的框架—celery。当然celery能完成的功能远不止异步任务, 还有一个很常用的功能—定时任务
celery的功能还包括:定义工作流、监控、任务流控制、资源泄露保护以及自定义用户组件等。
celery介绍
说明:
最新版本的celery支持的python版本必须大于2.7.6,
如果是python2.7.6及以下版本的时候import celery是会报错滴。
celery是通过将代码序列然后传输到中间通信组件,这些组件可以采用任何方式实现, 这里最常用的两种是rabbitmq和redis, 然后celery的后台线程不停的从rabbitmq或者redis中读取这些任务并执行然后返回结果到这些组件,这样就实现了一个异步的功能。
Celery 用redis或者rabbitmq做消息通信,这里redis或者rabbitmq被称为中间人(Broker)Celery 系统可包含多个线程和中间人,以此获得高可用性和横向扩展能力。
Celery 虽然是用 Python 编写的,但协议可以用任何语言实现。迄今,已有 Ruby 实现的 RCelery 、node.js 实现的 node-celery 以及一个 PHP 客户端 ,语言互通也可以通过 using webhooks 实现。
django 介绍
django作为python最主流也是资格最老的的web开发系统,是一个全栈的开发框架,几乎web开发系统中会用到的所有功能django都有,即使没有也可以在网站找到对应的开源解决方案,在stackoverflow上的问答也是最多的。基本上学习懂了django以后学习其他如flask、tornado都会觉得手到擒来。
本文中我们就介绍一下如何将celery集成到django中来完成django耗时任务的异步执行和定时任务计划。
我们将采用redis来做为中间人
celery 安装和使用
celery安装
pip install -U celery[redis]
该命令会安装celery以及redis开发相关所有的依赖包。安装完成我们可以看到:
这里我们可以看到安装了billiard、pytz、vine、amqp、redis、celery等
redis-server安装
既然是用redis做中间人,当然需要安装redis了、我们直接运行:
sudo apt-getin install redis-server
运行成功以后可以,redis-server直接就作为服务启动了, 我们可以通过:
ps aux|grep redis
命令来查看redis是否启动如下图:
这里我们可以看到redis已经在6379端口监听了
启动celery的worker
前面介绍了celery的处理流程, 既然我们已经启动了redis, 当然我们需要启动一个随时监听异步处理函数的worker了。 这里我们直接启动celery的worker就行了
首先我们来新建一个tasks.py 文件, 内容如下:
from celery import Celery
app = Celery('hello', broker='redis://localhost:6379/0')
@app.task
def hello():
return 'hello world'
然后我们允许下面的命令启动celery的worker
celery -A tasks worker --loglevel=info
注:
这里tasks表示的是上面创建的文件的名字, 比如如果我们的py文件为tasks.py, 则直接 celery -A tasks worker --loglevel=info。 如果我们的py文件为celery-tasks这命令应该修改为:
celery -A celery-tasks worker --loglevel=info
启动后我们就可以看到celery已经启动了线程时刻监听redis中的异步函数,如下:
接下来我们分析一下上面的tasks.py文件:
1. 首先直接初始化Celery对象, 并指明使用的redis的连接地址
2. 直接用celery对象的task装饰任何我们需要异步的函数
简单两步就完成了celery的异步函数
直接执行异步函数
这一步里面我们直接新建test.py文件, 内容如下:
from tasks import add
add.delay(1,2)
注意这里对add函数的调用采用的是delay函数而不是直接采用add(1,2),因为这样调用就和普通函数调用没有区别了。所以这里一定要注意。运行test.py文件后我们可以看到celery的输出:
在最后面我们可以清楚的看到调用了add函数, add函数的执行结果会返回到redis中
这里delay函数是将函数执行异步放入到redis中交给celery执行, 这样delay之后就会有个问题就是如果我们需要理解得到结果怎么办呢?
我们可以直接调用:
add.delay(1,2).get()
这样就变成同步的了,等到返回结果才会去执行下一步
celery添加异步任务
celery的使用非常简单
这里我们可以看到需要将一个函数变为异步函数非常简单, 只需要添加@app.task装饰器就够了。 是不是非常简单啊。
配置celery连接redis
app.conf.result_backend = ‘redis://localhost:6379/0’
配置任务执行结果保存地址
app.conf.result_backend = ‘redis://localhost:6379/0’
前面我们讲到过celery是从中间人取出函数并执行,但是保存结果也需要保存到中间人, 这里实际上取任务的地方和保存结果的中间人实际上可以不一样, 所有这里就提供了中间结果执行的保存地址
集成celery到django中
这里以我的一门django搭建在线教育平台的课程为例来讲解,大家如果有兴趣可以去关注一下,课程强力django+杀手级xadmin
首先我们来看一下完整的系统结构图:
1.修改django项目的MxOnline/settings.py文件, 加上:
###配置Broker
BROKER_URL = 'redis://127.0.0.1:6379/0'
BROKER_TRANSPORT = 'redis'
2.在MxOline下面新建celery.py文件
from __future__ import absolute_import
import os
import django
from celery import Celery
from django.conf import settings
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'MxOnline.settings')
django.setup()
app = Celery('MxOnline')
app.config_from_object('django.conf:settings')
app.autodiscover_tasks(lambda: settings.INSTALLED_APPS)
结构如图所示:
3.在对应的app下面新建tasks.py文件, 这里我们在users这个app下面新建, 如图所示:
文件源码如下:
from MxOnline.celery import app
@app.task
def send_register_email(email, send_type="register"):
email_record = EmailVerifyRecord()
if send_type == "update_email":
code = random_str(4)
else:
code = random_str(16)
email_record.code = code
email_record.email = email
email_record.send_type = send_type
email_record.save()
email_title = ""
email_body = ""
if send_type == "register":
email_title = "慕学在线网注册激活链接"
email_body = "请点击下面的链接激活你的账号: http://www.imooc.com/active/{0}".format(code)
send_status = send_mail(email_title, email_body, EMAIL_FROM, [email])
if send_status:
pass
elif send_type == "forget":
email_title = "慕学在线网注册密码重置链接"
email_body = "请点击下面的链接重置密码: http://www.imooc.com/reset/{0}".format(code)
send_status = send_mail(email_title, email_body, EMAIL_FROM, [email])
if send_status:
pass
elif send_type == "update_email":
email_title = "慕学在线邮箱修改验证码"
email_body = "你的邮箱验证码为: {0}".format(code)
send_status = send_mail(email_title, email_body, EMAIL_FROM, [email])
if send_status:
pass
4. 编辑views.py文件完成邮件发送异步调用:
#coding:utf-8
from django.shortcuts import render
from django.http import HttpResponse
from .tasks import send_register_email
def index(request):
send_register_email.delay()
return HttpResponse(u"邮件发送成功, 请查收")
5. 进入MxOnline目录运行:
celery -A demo worker -l debug
以此来启动celery的worker服务
关于django是如何实现邮件发送以及如何配置邮件的发送方配置,在课程强力django+杀手级xadmin中我会详细讲解,另外,课程中还会讲解django实现cookie和session的登录原理,当然django的强大绝不止于此,尤其是结合xadmin,你几乎可以做到以最快的速度搭建完整的前后台,如果你想全面掌握django,谋求一份python web 开发的工作,那就跟我一起来做一次实战:用django开发一个完整的在线教育网站,在实战中学习是掌握技术的捷径,你不仅会理解真实开发与纯理论的差别,更会了解如何避免开发中的“坑”,同时,这套完整的在线教育网站的全套代码都会开放给你,你甚至可以直接拿来用在工作中,所以如果这些是你想要的,那么我在慕课网等着你
至此,大功告成了!我们可以在我们定义的任何apps中添加tasks来定义需要的异步任务。