快捷導(dǎo)航

Django初步使用Celery處理耗時(shí)任務(wù)和定時(shí)任務(wù)問題

更新時(shí)間：2022年12月01日 11:02:26 投稿：jingxian

這篇文章主要介紹了Django初步使用Celery處理耗時(shí)任務(wù)和定時(shí)任務(wù)問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

1、Celery方式的選擇

這里Celery的中間人，采用Redis。也可以用Django自身的mongodb等。Celery的中間人你可以理解為在Celery執(zhí)行過程中的數(shù)據(jù)支持。保存列隊(duì)記錄、執(zhí)行記錄等等。

需要安裝celery-with-redis，執(zhí)行命令

pip?install?celery-with-redis

該命令會(huì)自動(dòng)安裝redis(python庫操作redis的庫)、celery、kombu、billiard、amqp、vine和celery-with-redis相關(guān)庫。

注意，這里pip安裝的redis是python操作redis的庫，非redis數(shù)據(jù)庫。

redis數(shù)據(jù)庫需要獨(dú)立安裝，在cmd里輸入 pip3 install redis

先說說django-celery的方式吧。這種方式就是通過manage.py啟動(dòng)celery。通常先被提到的方案是不會(huì)采用。用pip安裝django-celery，在settings引用djcelery應(yīng)用。

再更新數(shù)據(jù)庫：

python manage.py makemigrations djcelery
 
python manage.py migrate djcelery

查看數(shù)據(jù)庫，會(huì)發(fā)現(xiàn)多了很多相關(guān)的表，顯得十分多余了就。

djcelery還有個(gè)用途是在admin后臺(tái)動(dòng)態(tài)添加定時(shí)任務(wù)。這個(gè)功能也是比較雞肋，維護(hù)不方便而且可能造成各種不可預(yù)知的問題。

所以建議直接使用Celery管理Django中的任務(wù)（第二種方式）。這種方式也是Celery官網(wǎng)推薦的方式。

2、Django簡單項(xiàng)目準(zhǔn)備

這里我也簡單做一個(gè)示例。

首先，確保celery和redis已經(jīng)安裝好了，并且已經(jīng)啟動(dòng)了Redis服務(wù)。

另外，有個(gè)已經(jīng)搭建好了Django項(xiàng)目。

作為示例，簡單project和簡單app如下：

左邊側(cè)邊欄是該django的目錄結(jié)構(gòu)，右邊是myapp中的Blog模型。

再進(jìn)入后臺(tái)隨便加了兩條數(shù)據(jù)：

為了測試，一切從簡。views.py寫了一個(gè)響應(yīng)方法：

#views.py
from django.shortcuts import render
 
from django.http import HttpResponse
 
 
from models import Blog
 
import json
 
 
def home(request):
 
    data = list(Blog.objects.values('caption'))
 
    return HttpResponse(json.dumps(data), content_type = 'application/json')

django項(xiàng)目的urls.py加了一條首頁的url路由設(shè)置：

#urls.py
 
from django.conf.urls import url
 
from django.contrib import admin
 
from myapp.
 
 
urlpatterns = [
 
    url(r'^admin/', admin.site.urls),
 
    url(r'^$', 'myapp.views.home', name='home')
 
]

運(yùn)行django項(xiàng)目：

python?manage.py?runserver

打開http://localhost:8000/，如下效果：

3、Django加入Celery

現(xiàn)打開首頁要執(zhí)行一個(gè)收集訪客數(shù)據(jù)，發(fā)送郵件等操作。這是一個(gè)耗時(shí)任務(wù)，若放在home處理方法中執(zhí)行，用戶打開首頁會(huì)很慢。用戶體驗(yàn)不好，很可能不會(huì)等到頁面打開。

通常這個(gè)耗時(shí)任務(wù)可以多線程處理或者異步處理。我們模擬一個(gè)耗時(shí)任務(wù)，丟給Celery異步處理。

先模擬耗時(shí)任務(wù)，打開views.py，修改如下：

#views.py
 
from django.shortcuts import render
 
from django.http import HttpResponse
 
 
from .models import Blog
 
import json
 
import time
 
 
def sendmail(email):
 
    print('start send email to %s' % email)
 
    time.sleep(5) #休息5秒
 
    print('success')
 
    return True
 
 
def home(request):
 
    #耗時(shí)任務(wù)，發(fā)送郵件
 
    sendmail('test@test.com')
 
 
    #其他行為
 
    data = list(Blog.objects.values('caption'))
 
    return HttpResponse(json.dumps(data), content_type = 'application/json')

如此一來，至少需要再多等待5秒，才可以打開網(wǎng)頁。

打開settings.py所在的文件夾，新建celery.py文件。

加入如下代碼(注意，因?yàn)閏elery-with-django版本限制，我安裝的celery版本為3.1.25?？赡躢elery4.x的版本代碼不同)：

#celery.py
 
from __future__ import absolute_import, unicode_literals
 
 
from celery import Celery
 
from django.conf import settings
 
import os
 
 
#獲取當(dāng)前文件夾名，即為該Django的項(xiàng)目名
 
project_name = os.path.split(os.path.abspath('.'))[-1]
 
project_settings = '{}.settings'.format(project_name)
 
 
#設(shè)置環(huán)境變量
 
os.environ.setdefault('DJANGO_SETTINGS_MODULE', project_settings)
 
 
#實(shí)例化Celery
 
app = Celery(project_name)
 
 
#使用django的settings文件配置celery
 
app.config_from_object('django.conf:settings')
 
 
#Celery加載所有注冊(cè)的應(yīng)用
 
app.autodiscover_tasks(lambda: settings.INSTALLED_APPS)

這個(gè)文件還沒被加載，接著打開settings.py同個(gè)目錄下的__init__.py文件。讓運(yùn)行該Django項(xiàng)目的時(shí)候，加載該文件配置Celery。

修改代碼如下：

#__init__.py
 
from __future__ import absolute_import, unicode_literals
 
 
#引入celery實(shí)例對(duì)象
 
from .celery import app as celery_app

還需在settings.py中設(shè)置celery，尤其是中間人的設(shè)置。若不設(shè)置中間人，會(huì)提示無法連接中間人的錯(cuò)誤。

在settings.py文件中添加如下設(shè)置：

#celery settings
 
#celery中間人 redis://redis服務(wù)所在的ip地址:端口/數(shù)據(jù)庫號(hào)
 
BROKER_URL = 'redis://localhost:6379/0'
 
#celery結(jié)果返回，可用于跟蹤結(jié)果
 
CELERY_RESULT_BACKEND = 'redis://localhost:6379/0'
 
 
#celery內(nèi)容等消息的格式設(shè)置
 
CELERY_ACCEPT_CONTENT = ['application/json',]
 
CELERY_TASK_SERIALIZER = 'json'
 
CELERY_RESULT_SERIALIZER = 'json'
 
 
#celery時(shí)區(qū)設(shè)置，使用settings中TIME_ZONE同樣的時(shí)區(qū)
 
CELERY_TIMEZONE = TIME_ZONE

4、把耗時(shí)任務(wù)丟給celery處理

上面views.py中有個(gè)耗時(shí)任務(wù)sendmail。在myapp應(yīng)用中新建文件tasks.py，將sendmail方法剪切到該文件中并用定義為celery任務(wù)。

tasks.py文件如下代碼：

#tasks.py
 
from celery.decorators import task 
 
import time
 
 
@task
def sendmail(email):
 
    print('start send email to %s' % email)
 
    time.sleep(5) #休息5秒
 
    print('success')
 
    return True

在原有的方法上加上celery裝飾器task(或者也可以通過前面添加的celery_app給sendmail方法加裝飾器)：

#tasks.py
#myproject是當(dāng)前django的項(xiàng)目名
 
from myproject import celery_app
import time
 
@celery_app.task
def sendmail(email):
 
    print('start send email to %s' % email)
 
    time.sleep(5) #休息5秒
 
    print('success')
 
    return True

另外原先的views.py修改如下：

#views.py
 
from django.shortcuts import render
 
from django.http import HttpResponse
 
 
from .models import Blog
 
from .tasks import sendmail #引用tasks.py文件的中sendmail方法
 
import json
 
 
def home(request):
 
    #耗時(shí)任務(wù)，發(fā)送郵件（用delay執(zhí)行方法）
 
    sendmail.delay('test@test.com')
 
 
    #其他行為
 
    data = list(Blog.objects.values('caption'))
 
    return HttpResponse(json.dumps(data), content_type = 'application/json')