基於Django與Celery實現非同步佇列任務

發表於2015-08-24

在運營系統中經常用到非同步方式來處理我們的任務,比如將業務上線流程串成任務再寫入佇列,通過後臺作業節點去排程執行。比較典型的案例為騰訊的藍鯨、織雲、雲智慧等平臺。本譯文結合Django+Celery+Redis實現一個定期從Flickr 獲取圖片並展示的簡單案例,方便大家理解實現非同步對列任務的過程。

剛接觸django的時候,我經歷過的最讓人沮喪的事情是需要定期執行一段程式碼。我寫了一個需要每天上午12點執行一個動作的不錯的函式。很簡單是不是?錯了。事實證明,這對我來說是一個巨大的困難點,因為,那時我使用Cpane型別的虛擬主機管理系統,它能專門提供一個很友好,很方便的圖形使用者介面來設定cron作業。

經過反覆研究,我發現了一個很好的解決方案 – Celery,一個用於在後臺執行任務的強大的非同步作業佇列。但是,這也導致了其它的問題,因為我無法找到一系列簡單的指令將celery整合到Django專案中。

當然,我最終還是設法成功搞定了它 – 這正是本文將介紹的內容:如何將celery整合到一個Django專案,建立週期性任務。

該專案利用Python3.4,Django的1.8.2,celery3.1.18和Redis3.0.2.

 

一、概述

由於大篇幅的文字,為了您的方便,請參閱下表中的每一步的簡要資訊,並獲取相關的程式碼。

步驟                 概要                                     Git標籤
樣板                 樣板下載                                 V1
建立                 整合Celery和Django            V2
Celery任務      新增基本的Celery任務         V3
週期性任務      新增週期性任務                     V4
本地執行         本地執行我們的應用程式       V5
遠端執行         遠端執行我們的應用程式       V5

 

二、什麼是Celery

“Celery是一個非同步任務佇列/基於分散式訊息傳遞的作業佇列。它側重於實時操作,但對排程的支援也很好。”本文,我們將重點講解週期性執行任務的排程特點。

為什麼這一點有用呢?

回想一下你不得不在將來執行某一特定任務的經歷。也許你需要每隔一小時訪問一個API。或者,也許你需要在這一天結束時傳送一批電子郵件。不論任務大小,Celery都可以使得排程週期性任務變的很容易。

你永遠不希望終端使用者等待那些不必要的頁面載入或動作執行完成。如果你的應用程式工作流的一部分是一個需要很長時間的程式,當資源可用時,你就可以使用Celery在後臺執行這段程式,從而使你的應用程式可以繼續響應客戶端的請求。這樣可以使任務在應用程式的環境之外執行。

 

三、構建專案

在深入瞭解Celery之前,先從Github庫中獲取開始專案。確保啟用一個虛擬的環境,安裝必要的軟體,並執行遷移。然後啟動伺服器,通過你的瀏覽器導航到http://localhost:8000/。你應當能看到‘恭喜你的第一個Django頁面’。完成後,關閉伺服器。
接下來,我們開始安裝celery。

現在,我們通過簡單的三步將celery整合到django專案中。

步驟一:建立celery.py

在“picha“目錄下,建立celery.py,程式碼如下:

請注意程式碼中的註釋。

步驟二:引入celery應用

為了確保在django啟動時載入了celery應用,在settings.py旁邊新建__init__.py,並新增以下程式碼到__init__.py中。

完成以上步驟後,你的專案目錄應該是這樣的:

步驟三:安裝 Redis作為Celery的“中介軟體”

Celery使用中介軟體在django專案與celery監控者之間傳遞訊息。在本教程中,我們使用redis作為訊息中間代理。

首先,從官方下載頁面或通過brew(BREW安裝Redis)安裝Redis,然後開啟你的終端上,在一個新的終端視窗,啟動伺服器:

你可以通過在終端中輸入如下命令測試Redis是否正常工作。

Redis應該回復PONG – 試試吧!
一旦Redis正常啟動了,把下面的程式碼新增到你的settings.py檔案中:

你還需要新增Redis的作為Django專案的依賴:

就是這樣了!你現在應該能夠在Django中使用Celery。有關設定Celery與Django的更多資訊,請檢視官方Celery文件。

在繼續下面步驟之前,讓我們進行一些完整性檢查,以確保一切都是正常的。

測試Celery worker已準備好接收任務:

使用CTRL-C殺死該段程式。現在,測試Celery任務排程程式是否已經準備好:

在上述完成時再次終止該程式。

1、Celery任務

Celery利用celery呼叫的常規Python函式作為任務。

例如,讓我們把這個基本函式變為celery的任務:

首先,新增一個裝飾器。

然後你可以通過以下方式利用celery非同步執行該任務:

很簡單,對不對?

所以,這對於解決類似你要載入一個網頁,而不需要使用者等待一些後臺程式的完成這些型別的任務來說是非常完美的。

讓我們來看一個例子…

讓我們再回到Django專案的版本3,它包括一個接受來自使用者的反饋的應用程式,人們形象地稱之為反饋:

安裝新的必要軟體,啟動應用程式,並導航到http://localhost:8000/feedback/。你應該看到如下結果:

讓我們連線celery任務。

2、新增任務

基本上,使用者提交反饋表後,我們希望讓他繼續以他舒服的方式往下進行,而我們在後臺進行處理反饋,傳送電子郵件等等。

要做到這一點,首先新增一個叫tasks.py的檔案到“feedback”目錄:

然後按照如下內容更新forms.py:

大體上,send_feedback_email_task.delay(email, message)的函式過程,併傳送反饋電子郵件等都是在使用者繼續使用該網站的同時作為後臺程式執行。

注:在views.py中的success_url被設定為將使用者重定向到/ 目錄,這個目錄還不存在。我們會在下一節設定這個終點啟動。

3、週期任務

通常情況下,你經常需要安排一個任務在特定的時間執行 – 例如,一個web scraper 可能需要每天都執行。這樣的任務,被稱為週期性任務,很容易建立利用celery啟動。

celery使用“celery beat”來安排定期任務。celery beat定期執行任務,然後由celery worker執行任務。

例如,下面的任務計劃每15分鐘執行一次:

讓我們通過往Django專案中新增功能來看一個更強大的例子。

回到Django專案版本4,它包括另一個新的應用程式,叫做photos,這個應用程式使用 Flickr API獲取新照片用來顯示在網站:

安裝新的必要軟體,執行遷移,然後啟動伺服器,以確保一切都是好的。重新測試反饋表。這次,它應該重定向好了。
下一步是什麼?

既然我們需要週期性的呼叫Flickr API,以獲取更多的照片新增到我們的網站,我們可以新增一個celery任務。

4、新增任務

往photos應用中新增一個tasks.py。

在這裡,我們通過在一個task中包裝這個函式,來實現每15分鐘執行一次save_latest_flickr_image()函式。該@periodic_task裝飾器抽象出程式碼來執行celery任務,使得tasks.py乾淨,易於閱讀!

5、本地執行

準備開始執行了?

在Django應用程式和Redis執行的前提下,開啟兩個新的終端視窗/標籤。在每一個新的視窗中,導航到你的專案目錄,啟用你的虛擬環境,然後執行下面的命令(每個視窗一個):

當你訪問http://127.0.0.1:8000/ 網址的時候,你現在應該能看到一個圖片。我們的應用程式每15分鐘從Flickr 獲取一張圖片。

通過photos/tasks.py檢視程式碼。點選“Feedback”按鈕傳送一些反饋意見:


以上是通過celery任務執行的。更多的請檢視feedback/tasks.py。

就這樣,你成功的啟動並執行了 Picha專案!

當你本地開發Django專案時,這是一個很好的測試,但是當你需要部署到生產環境- 就像 DigitalOcean時,就不那麼合適了。為此,建議你通過使用Supervisor在後臺作為一個守護程式執行celery worker和排程器。

6、遠端執行

安裝很簡單。從版本庫中獲取版本5(如果你還沒有的話)。然後,SSH到遠端伺服器,並執行:

然後,通過在遠端伺服器上“/etc/supervisor/conf.d/” 目錄下新增配置檔案來告知Supervisor celery的workers。在我們的例子中,我們需要兩個這樣的配置檔案 – 一個用於Celery worker,一個是Celery scheduler。

在本地,在專案的根目錄下建立一個“supervisor”的資料夾,然後新增下面的檔案。

Celery Worker: picha_celery.conf

注:確保更新這些檔案的路徑,以匹配你的遠端伺服器的檔案系統。

基本上,這些supervisor 配置檔案告訴supervisord如何執行並管理我們的’programs’(因為它們是由supervisord呼叫)。

在上面的例子中,我們已經建立了兩個名為“pichacelery”和“pichacelerybeat”的supervisord程式。

現在,只需將這些檔案拷貝到遠端伺服器的/etc/supervisor/conf.d/目錄下。

我們還需要在遠端伺服器上建立上面指令碼中提到的日誌檔案:

最後,執行以下命令,使 Supervisor 知道它所管理的程式的存在 – 例如,pichacelery和pichacelerybeat:

執行以下命令停止,啟動,和/或檢查pichacelery程式的狀態:

你可以通過閱讀官方文件獲取Supervisor的更多資訊。

7、最後提示

1. 千萬不要傳遞Django模型物件到celery任務。為了避免模型物件在傳遞給celery任務之前已經改變了,傳遞celery的主鍵給celery。然後,在執行之前使用主鍵從資料庫中獲取物件。
2. 預設celery排程會在本地建立一些檔案儲存它的排程表。這些檔案是“celerybeat-schedule.db”和“celerybeat.pid”。如果你在使用版本控制系統,比如Git(你應該使用!),請忽略這個檔案,不要將它們新增到你的程式碼庫中,因為它們是為本地執行的程式服務的。

8、下一步

以上就是將celery整合到一個django專案的基本介紹。

想要更多?

1. 深入研究官方celery使用者指南,以瞭解更多資訊。
2. 建立一個Fabfile來設定Supervisor和配置檔案。確保新增命令到reread和 update Supervisor。
3. 從repo中獲取這個專案,並開啟一個Pull 請求來新增一個新的celery任務。

編碼快樂!

相關文章