Django Celery异步任务管理简介

2023-04-11 00:00:00 django 管理 简介

Django Celery是一个基于Python的异步任务管理框架，它通过将任务缓存到消息队列中，从而使得我们能够将任务异步地执行，减少Web应用的延迟，提高程序的并行性。

下面简要介绍Django Celery的使用方法：

  1. 安装Celery

使用pip安装Celery：

pip install celery
  1. 配置Celery

在Django项目中添加一个'celery.py'文件，并在其中添加以下内容：

import os
from celery import Celery

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'your_project_name.settings')

app = Celery('your_project_name')

app.config_from_object('django.conf:settings', namespace='CELERY')

app.autodiscover_tasks()

这段代码会从Django项目中读取配置信息，并根据配置信息初始化一个Celery客户端。

  1. 编写任务

在Django项目的任意一个app中添加一个'tasks.py'文件，并在其中定义一个任务：

from celery import shared_task


@shared_task
def test_task(name):
    return 'Hello, {}'.format(name)

这个任务可以接受一个名字参数，返回一个字符串。

  1. 启动Celery

在终端中输入以下命令，启动Celery：

celery -A your_project_name worker --loglevel=info

这条命令会启动一个Celery worker进程，这个进程会不断地从队列中取出任务，并执行对应的任务函数。

  1. 调用任务

在任何位置都可以调用我们之前定义的任务，例如在Django的视图函数中：

from your_project_name.tasks import test_task


def test_view(request):
    result = test_task.delay('pidancode.com')
    return HttpResponse(result.id)

这个视图函数调用了之前定义的test_task任务，并向它传递了一个'name'参数。调用的结果是一个异步任务的id，这个id可以用来查询任务的状态。

  1. 查询任务状态

我们可以通过之前异步任务的id来查询任务的状态，例如在Django的其他视图函数中：

from your_project_name.tasks import test_task


def test_task_result(request, task_id):
    result = test_task.AsyncResult(task_id)

    if result.successful():
        return HttpResponse(result.result)
    else:
        return HttpResponse('Task not finished yet')

这个视图函数接受了一个task_id参数，然后使用AsyncResult函数查询任务的状态。如果任务已经完成，那么返回任务的结果；否则返回任务未完成的提示信息。

以上就是Django Celery的简单介绍，希望能对大家有所帮助。

相关文章