Celery 和 Django 简单示例

Question

让我们举一个简单的 Django 示例。

app/models.py

from django.db import models
from django.contrib.auth.models import User

class UserProfile(models.Model):
    user = models.OneToOneField(User)
    token = models.CharField(max_length=32)

app/views.py

from django.http import HttpResponse
from django.views.decorators.csrf import csrf_exempt
from forms import RegisterForm
from utils.utilities import create_user

@csrf_exempt
def register_view(request):
    if request.method == 'POST':
        form = RegisterForm(request.POST)
        if form.is_valid():
            create_user(form.cleaned_data)
            return HttpResponse('success')

utils/utilities.py

def create_user(data):
    user = User.objects.create_user(username=data['username'], email=None, password=data['password'])
    user.save()
    profile = UserProfile()
    profile.user = user
    profile.token = generate_token()
    profile.save()

有人可以在这个例子中提供 Celery 的实现吗？想象一下这是一个每秒有数百个请求的大型项目。

Answer 1

假设您同时安装了 Python 的 celery 和 django-celery ，请在您的应用下创建以下 tasks.py 文件：

utils/tasks.py

from celery import task
# other imports

@task()
def create_user(data):
    user = User.objects.create_user(
        username=data['username'], email=None, password=data['password']
    )
    user.save()
    profile = UserProfile()
    profile.user = user
    profile.token = generate_token()
    profile.save()

    return None

删除上面示例中的 utils/utilities.py 文件。

在views.py 的代码中，将create_user 调用从：

create_user(form.cleaned_data)

到：

create_user.delay(form.cleaned_data)

基本上create_user现在是一个芹菜任务；如果你安装了正确的 Python 包（如上所述），代码方面（你要求的实现）就是这样。 delay 异步执行您的函数 - 即返回 HTTP 响应而不等待异步任务完成。

您可以在本地使用 python manage.py celeryd 运行 celery 守护进程。

在生产中，您必须使用 upstart、supervisor 或任何其他工具来设置 celery 流程本身来控制此类流程的生命周期。

更多详细信息记录在here。