[Python知识库]Celery+Django异步处理请求

Celery目录

celery作为一个应用，我们有下面的目录结构：

main.py

存放启动一个celery任务的主要流程，包括：

1. 创建Celery的主程序对象
2. 将jango和celery结合，需要先启动django
3. 加载celery配置
4. 注册celery任务
5. 最后通过终端命令运行

from celery import Celery

# 创建Celery的主程序对象
app = Celery("luffy")

# 将jango和celery结合，需要先启动django
import os
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'luffyapi.settings.dev')
import django
django.setup()

# 加载配置
app.config_from_object("mycelery.config")

# 注册celery任务
app.autodiscover_tasks(["mycelery.sms"])

# 通过终端来启动celery
# celery -A mycelery.main worker --loglevel=info

config.py

如注释，主要是设置任务队列的url地址、结果队列地址，我们这里用redis来存放：

# 设置任务队列的url地址
broker_url = "redis://127.0.0.1:6379/14"
# 设置结果队列
result_backend = "redis://127.0.0.1:6379/15"

tasks.py

必须这么命名。通过装饰器来申明这是一个要运行的task，并给这个task起个名字。这里封装好要异步处理的代码:

from mycelery.main import app
from luffyapi.libs.yuntongxun.sms import CCP
from luffyapi.settings import constants

import logging
log = logging.getLogger("django")

@app.task(name="send_sms")
def send_sms(mobile, sms_code):
    """发送短信"""
    ccp = CCP()
    ret = ccp.send_template_sms(mobile, [sms_code, constants.SMS_EXPIRE_TIME//60], constants.SMS_TEMPLATE_ID)
    if not ret:
        # log.error(ret)
        log.error("用户注册短信发送失败！手机号：%s" % mobile)
        # return {"message" : "发送短信失败"}

views.py

将views.py中实现的逻辑代码换成下面酱紫，便可以异步执行了。

from mycelery.sms.tasks import send_sms
# send_sms本身没有delay()方法的,是通过装饰器来传到send_sms中的
send_sms.delay(mobile, sms_code)

启动celery

在终端执行：

celery -A mycelery.main worker --loglevel=info

这就实现了发送验证码的异步请求。

创作挑战赛

新人创作奖励来咯，坚持创作打卡瓜分现金大奖