Celery在Django项目中集成

发布时间 2023-08-05 11:06:27作者: hkwJsxl

使用celery第一件要做的最为重要的事情是需要先创建一个Celery实例对象,我们一般叫做celery应用对象,或者更简单直接叫做一个app。app应用对象是我们使用celery所有功能的入口,比如启动celery、创建任务,管理任务,执行任务等.

celery框架有2种使用方式,一种是单独一个项目目录,另一种就是Celery集成到web项目框架中。

Celery作为单个项目运行

项目目录

-- celeryapp
  -- sms  # 异步任务目录,这里拿发送短信来举例,一个类型的任务就一个目录
    - __init__.py
    - ronglianyunapi.py  # 短信发送文件
    - tasks.py  # 任务的文件,文件名必须是tasks.py!!!每一个任务就是一个被装饰的函数,写在任务文件中
  - __init__.py
  - main.py  # 入口程序
  - settings.py  # 配置文件

main.py,代码:

from celery import Celery

# 实例化celery应用,参数一般为项目应用名
app = Celery("lg_shop")

# 通过app实例对象加载配置文件(参数为导包路径)
app.config_from_object("celeryapp.settings")

# 注册任务, 自动搜索并加载任务
# 参数必须必须是一个列表,里面的每一个任务都是任务的路径名称
# app.autodiscover_tasks(["任务1","任务2",....])
app.autodiscover_tasks(["celeryapp.sms"])

# 启动Celery的终端命令
# celery -A celeryapp.main worker --loglevel=info  # windows不好使,能接收到任务,但不执行
# celery -A celeryapp.main worker --loglevel=info --pool=solo  # windows加上这个参数
# celery -A celeryapp.main worker --loglevel=info -P eventlet -c 10  # 或者使用eventlet,-c是协程的数量,生产环境可以用1000

配置文件settings.py,代码:

REDIS_PASSWORD = "root123456"
REDIS_HOST = "127.0.0.1"
REDIS_PORT = 6379

# 任务队列的链接地址
broker_url = 'redis://:%s@%s:%s/14' % (
    REDIS_PASSWORD, REDIS_HOST, REDIS_PORT
)
# 结果队列的链接地址
result_backend = 'redis://:%s@%s:%s/15' % (
    REDIS_PASSWORD, REDIS_HOST, REDIS_PORT
)

创建任务文件sms/tasks.py,任务文件名必须固定为tasks.py,并创建任务,代码:

from ..main import app
from . import sms_sington


@app.task(name="send_sms_code")
def send_sms_code(tid, mobile, datas):
    response = sms_sington.send_sms(tid=tid, mobile=mobile, datas=datas)
    return response

@app.task(name="send_sms1")
def send_sms1():
    """没有任何参数的异步任务"""
    print('任务:send_sms1执行了...')


@app.task(name="send_sms2")
def send_sms2(mobile, code):
    """有参数的异步任务"""
    print(f'任务:send_sms2执行了...mobile={mobile}, code={code}')


@app.task(name="send_sms3")
def send_sms3():
    """有结果的异步任务"""
    print('任务:send_sms3执行了...')
    return 100


@app.task(name="send_sms4")
def send_sms4(x, y):
    """有结果的异步任务"""
    print('任务:send_sms4执行了...')
    return x + y

发送短信api,sms/ronglianyunapi.py

import os
import json

from ronglian_sms_sdk import SmsSDK

RONGLIANYUN = {
    "accId": '2c94811c86c00e9b0186f2873a040afa',
    "accToken": os.environ.get("RONGLIANYUNACCTOKEN"),
    "appId": '2c94811c86c00e9b0186f2873b0d0b01',
    "reg_tid": 1,  # 注册短信验证码的模板ID
    "sms_expire": 60,  # 短信有效期,单位:秒(s)
    "sms_interval": 60,  # 短信发送的冷却时间,单位:秒(s)
}


class SMSSington:
    def __new__(cls, *args, **kwargs):
        if not hasattr(cls, "_instance"):
            cls._instance = super().__new__(cls, *args, **kwargs)
            ronglianyun = RONGLIANYUN
            cls._instance.sms = SmsSDK(ronglianyun.get("accId"), ronglianyun.get("accToken"), ronglianyun.get("appId"))
        return cls._instance

    def send_sms(self, mobile, datas, tid=1):
        """
        发送短信
        @params tid: 模板ID,默认测试使用1
        @params mobile: 接收短信的手机号,多个手机号使用逗号隔开
                单个号码: mobile="13312345678"
                多个号码: mobile="13312345678,13312345679,...."
        @params datas: 短信模板的参数列表(测试只能发四位验证码)
                例如短信模板为: 【云通讯】您的验证码是{1},请于{2}分钟内正确输入。
                则datas=("1234",5,)
        """
        sdk = self._instance.sms
        resp = sdk.sendMessage(tid, mobile, datas)
        response = json.loads(resp)
        return response

sms/__init__.py,这个文件的作用是减少tasks.py里面的导包路径

from .ronglianyunapi import SMSSington

sms_sington = SMSSington()

接下来,我们运行celery。

# 项目根目录下运行

# 启动Celery的终端命令(普通的运行方式)
# celery -A celeryapp.main worker --loglevel=info  # windows不好使,能接收到任务,但不执行
# celery -A celeryapp.main worker --loglevel=info --pool=solo  # windows加上这个参数
# celery -A celeryapp.main worker --loglevel=info -P eventlet -c 10  # 或者使用eventlet,-c是协程的数量,生产环境可以用1000

# 启动多工作进程,以守护进程的模式运行[一个工作进程就是4个子进程]
# windows大概率报错
# 注意:pidfile和logfile必须以绝对路径来声明
celery multi start worker -A celeryapp.main -l info -n worker1
celery multi start worker -A celeryapp.main -l info -n worker2

# 关闭运行的工作进程
celery multi stop worker -A celeryapp.main -l info -n worker1
celery multi stop worker -A celeryapp.main -l info -n worker2

调用上面的异步任务,拿django的shell进行举例:

# 终端调试
python manage.py shell

# 调用celery执行异步任务
from celeryapp.sms.tasks import send_sms1,send_sms2,send_sms3,send_sms4
mobile = "18533538210"
code = "6666"

# delay 表示马上按顺序来执行异步任务,在celrey的worker工作进程有空闲的就立刻执行
# 可以通过delay异步调用任务,可以没有参数
ret1 = send_sms1.delay()
# 可以通过delay传递异步任务的参数,可以按位置传递参数,也可以使用命名参数
# ret2 = send_sms.delay(mobile=mobile,code=code)
ret2 = send_sms2.delay(mobile,code)

# apply_async 让任务在后面指定时间后执行,时间单位:秒/s
# 任务名.apply_async(args=(参数1,参数2), countdown=定时时间)
ret4 = send_sms4.apply_async(kwargs={"x":10,"y":20},countdown=30)

# 根据返回结果,不管delay,还是apply_async的返回结果都一样的。
ret4.id      # 返回一个UUID格式的任务唯一标志符,5b2becb8-c1ba-4e33-8593-9c6fa7ff8c78
ret4.status  # 查看当前任务的状态 SUCCESS表示成功! PENDING任务等待
ret4.get()   # 获取任务执行的结果[如果任务函数中没有return,则没有结果,如果结果没有出现则会导致阻塞]

if ret4.status == "SUCCESS":
    print(ret4.get())

在django的视图里面,我们调用Celery来异步执行任务(这里容联云发送短信举例)。

***/views.py

# 异步发送短信
from celeryapp.sms.tasks import send_sms_code
# 原来的
response = send_sms_code(tid=settings.RONGLIANYUN.get("reg_tid"), mobile=mobile,
                                       datas=(sms_code, SMS_CODE_EXPIRES // 60))
# 改成
response = send_sms_code.delay(tid=settings.RONGLIANYUN.get("reg_tid"), mobile=mobile,
                                       datas=(sms_code, SMS_CODE_EXPIRES // 60))

上面就是使用celery并执行异步任务的第一种方式,适合在一些无法直接集成celery到项目中的场景,或是想celery单独跑在一台服务器上的场景。

Celery作为第三方模块在Django中集成

这里还是拿django来举例,目录结构调整如下:

-- lg_shop/
  -- lg_shop/
    -- apps/
      -- verify/
        - tasks.py  # 分散在各个子应用下的异步任务模块
    -- libs/
      -- sms/
        - __init__.py
        - ronglianyunapi.py
    - __init__.py
  - __init__.py
  - celery_app.py  # celery入口程序,相当于上一种用法的main.py

lg_shop/celery_app.py,主应用目录下创建cerley入口程序,创建celery对象并加载配置和异步任务,代码:

import os
from celery import Celery

# 必须在实例化celery应用对象之前执行(Django项目manage.py里面的)
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'lg_shop.settings.dev')

# 实例化celery应用,参数一般为项目应用名
app = Celery("lg_shop")
# 指定任务的队列名称
app.conf.task_default_queue = 'Celery'
# 可以把配置写在django的项目配置中
# 设置django中配置信息以 "CELERY_"开头为celery的配置信息
app.config_from_object('django.conf:settings', namespace='CELERY')
# 注册任务,自动搜索并加载任务
app.autodiscover_tasks()

# 启动Celery的终端命令
# celery -A lg_shop worker --loglevel=info  # windows不好使,能接收到任务,但不执行
# celery -A lg_shop worker --loglevel=info --pool=solo  # windows加上这个参数
# celery -A lg_shop worker --loglevel=info -P eventlet -c 10  # 或者使用eventlet,-c是协程的数量,生产环境可以用1000

settings/dev.py,django配置中新增celery相关配置信息,代码:

# celery相关
# Celery异步任务队列框架的配置项[注意:django的配置项必须大写,所以这里的所有配置项必须全部大写]
# 任务队列的链接地址
CELERY_BROKER_URL = 'redis://:%s@%s:%s/14' % (
    REDIS_PASSWORD, REDIS_HOST, REDIS_PORT
)
# 结果队列的链接地址
CELERY_RESULT_BACKEND = 'redis://:%s@%s:%s/15' % (
    REDIS_PASSWORD, REDIS_HOST, REDIS_PORT
)
# 时区,与django的时区同步
CELERY_TIMEZONE = TIME_ZONE
# 防止死锁
CELERY_FORCE_EXECV = True
# 设置并发的worker数量
CELERYD_CONCURRENCY = 200
# celery的任务结果内容格式
CELERY_ACCEPT_CONTENT = ['json', 'pickle']
# 设置失败允许重试[这个慎用,如果失败任务无法再次执行成功,会产生指数级别的失败记录]
# CELERY_ACKS_LATE = True
# 每个worker工作进程最多执行500个任务被销毁,可以防止内存泄漏,500是举例,根据自己的服务器的性能可以调整数值
# CELERYD_MAX_TASKS_PER_CHILD = 500
# 单个任务的最大运行时间,超时会被杀死[慎用,有大文件操作、长时间上传、下载任务时,需要关闭这个选项,或者设置更长时间]
# CELERYD_TIME_LIMIT = 10 * 60
# 任务发出后,经过一段时间还未收到acknowledge, 就将任务重新交给其他worker执行
CELERY_DISABLE_RATE_LIMITS = True

# 之前定时任务(定时一次调用),使用了apply_async({}, countdown=30);
# 设置定时任务(定时多次调用)的调用列表,需要单独运行SCHEDULE命令才能让celery执行定时任务:
# celery -A celeryapp.main beat,当然worker还是要启动的

# from celery.schedules import crontab
# CELERY_BEAT_SCHEDULE = {
#     "send_sms1": {  # 定时任务的注册标记符[必须唯一的]
#         "task": "send_sms1",  # 定时任务的任务名称
#         "schedule": 10,  # 定时任务的调用时间,10表示每隔10秒调用一次add任务
#         # "schedule": crontab(hour=7, minute=30, day_of_week=1),  # 定时任务的调用时间,每周一早上7点30分调用一次add任务
#     }
# }

lg_shop/__init__.py,主应用下初始化,代码:

import pymysql

pymysql.install_as_MySQLdb()
from .celery_app import app as celery_app

__all__ = ['celery_app']

***/tasks.py,这个任务文件要写在子应用下,app.autodiscover_tasks会自动调用。

from celery import shared_task

from sms import sms_sington


@shared_task(name="send_sms_code")
def send_sms_code(tid, mobile, datas):
    response = sms_sington.send_sms(tid=tid, mobile=mobile, datas=datas)
    return response

libs/sms/__init__.py

from .ronglianyunapi import SMSSington

sms_sington = SMSSington()

libs/sms/ronglianyunapi.py

import json

from django.conf import settings

from ronglian_sms_sdk import SmsSDK

from logger import log
from response_code import RETCODE


class SMSSington:
    def __new__(cls, *args, **kwargs):
        if not hasattr(cls, "_instance"):
            cls._instance = super().__new__(cls, *args, **kwargs)
            ronglianyun = settings.RONGLIANYUN
            cls._instance.sms = SmsSDK(ronglianyun.get("accId"), ronglianyun.get("accToken"), ronglianyun.get("appId"))
        return cls._instance

    def send_sms(self, mobile, datas, tid=1):
        """
        发送短信
        @params tid: 模板ID,默认测试使用1
        @params mobile: 接收短信的手机号,多个手机号使用逗号隔开
                单个号码: mobile="13312345678"
                多个号码: mobile="13312345678,13312345679,...."
        @params datas: 短信模板的参数列表(测试只能发四位验证码)
                例如短信模板为: 【云通讯】您的验证码是{1},请于{2}分钟内正确输入。
                则datas=("1234",5,)
        """
        sdk = self._instance.sms
        resp = sdk.sendMessage(tid, mobile, datas)
        response = json.loads(resp)
        if response.get("statusCode") != "000000":
            msg = {"code": RETCODE.SMSCODESENDRR, "msg": f"{mobile}验证码发送失败-{response.get('statusMsg')}"}
            log.error(msg)
        return response

django中的用户发送短信,就可以改成异步发送短信了。

视图中调用异步发送短信的任务,代码:

# 异步发送短信
from 子应用.tasks import send_sms_code
# 原来的
response = send_sms_code(tid=settings.RONGLIANYUN.get("reg_tid"), mobile=mobile,
                                       datas=(sms_code, SMS_CODE_EXPIRES // 60))
# 改成
response = send_sms_code.delay(tid=settings.RONGLIANYUN.get("reg_tid"), mobile=mobile,
                                       datas=(sms_code, SMS_CODE_EXPIRES // 60))

终端下先启动celery,在django项目根目录下启动。

# 1. 普通运行模式,关闭终端以后,celery就会停止运行
# celery -A 项目名 worker -l info
celery -A lg_shop worker -l info --pool=solo
# 2. 启动多worker进程模式,以守护进程的方式运行,不需要在意终端。但是这种运行模型,一旦停止,需要手动启动。
# windows大概率报错
celery multi start worker -A lg_shop -l info -n lg_shop_worker
# 3. 停止多worker进程模式
celery multi stop worker -A lg_shop -l info -n lg_shop_worker

关于celery中异步任务发布的2个方法的参数如下:

异步任务名.delay(*arg, **kwargs)
异步任务名.apply_async((arg,), {'kwarg': value}, countdown=60, expires=120)

定时任务的调用器启动,可以在运行了worker以后,使用以下命令:

celery -A lg_shop beat

beat调度器关闭了,则定时任务无法执行,如果worker工作进程关闭了,则celery关闭,保存在消息队列中的任务就会囤积在那里。

代码示例

https://github.com/HkwJsxl/lg_shop