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开发相关所有的依赖包。安装完成我们可以看到:

安装celery后的截图.png

这里我们可以看到安装了billiard、pytz、vine、amqp、redis、celery等

redis-server安装

既然是用redis做中间人,当然需要安装redis了、我们直接运行:

sudo apt-getin install redis-server

运行成功以后可以,redis-server直接就作为服务启动了, 我们可以通过:

ps aux|grep redis

命令来查看redis是否启动如下图:


查看redis服务器启动情况.png

这里我们可以看到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中的异步函数,如下:


celery启动截图.png

接下来我们分析一下上面的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的输出:

celery接收到celery的截图.png

在最后面我们可以清楚的看到调用了add函数, add函数的执行结果会返回到redis中
这里delay函数是将函数执行异步放入到redis中交给celery执行, 这样delay之后就会有个问题就是如果我们需要理解得到结果怎么办呢?

我们可以直接调用:

add.delay(1,2).get()

这样就变成同步的了,等到返回结果才会去执行下一步

celery添加异步任务

celery的使用非常简单

这里我们可以看到需要将一个函数变为异步函数非常简单, 只需要添加@app.task装饰器就够了。 是不是非常简单啊。

  1. 配置celery连接redis

    app.conf.result_backend = ‘redis://localhost:6379/0

  2. 配置任务执行结果保存地址

    app.conf.result_backend = ‘redis://localhost:6379/0

    前面我们讲到过celery是从中间人取出函数并执行,但是保存结果也需要保存到中间人, 这里实际上取任务的地方和保存结果的中间人实际上可以不一样, 所有这里就提供了中间结果执行的保存地址

集成celery到django中

这里以我的一门django搭建在线教育平台的课程为例来讲解,大家如果有兴趣可以去关注一下,课程强力django+杀手级xadmin

首先我们来看一下完整的系统结构图:


系统结构.png

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)

结构如图所示:


添加celerypy文件后.png

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来定义需要的异步任务。

转自bobby博客,自用

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 196,099评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,473评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,229评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,570评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,427评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,335评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,737评论 3 386
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,392评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,693评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,730评论 2 312
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,512评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,349评论 3 314
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,750评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,017评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,290评论 1 251
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,706评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,904评论 2 335