Django的Celery集成Flask任務(wù)隊列方案

Django 是一個強大的 Python Web 框架，而 Celery 是一個分布式任務(wù)隊列，用于處理后臺任務(wù)和異步執(zhí)行。將 Celery 集成到 Flask 應(yīng)用中可以幫助你更好地管理后臺任務(wù)。以下是一個簡單的方案，展示如何在 Flask 應(yīng)用中集成 Celery。

1. 安裝依賴

首先，確保你已經(jīng)安裝了 Flask 和 Celery。你可以使用 pip 來安裝這些依賴：

pip install Flask celery

2. 配置 Celery

在你的 Flask 應(yīng)用中配置 Celery。假設(shè)你的應(yīng)用文件名為 app.py，你可以按照以下步驟進行配置：

from flask import Flask
from celery import Celery

# 創(chuàng)建 Flask 應(yīng)用
app = Flask(__name__)

# 配置 Celery
app.config['CELERY_BROKER_URL'] = 'redis://localhost:6379/0'  # 使用 Redis 作為消息代理
app.config['CELERY_RESULT_BACKEND'] = 'redis://localhost:6379/0'  # 使用 Redis 存儲任務(wù)結(jié)果
app.config['CELERY_ACCEPT_CONTENT'] = ['json']
app.config['CELERY_TASK_SERIALIZER'] = 'json'
app.config['CELERY_RESULT_SERIALIZER'] = 'json'
app.config['CELERY_TIMEZONE'] = 'UTC'

# 創(chuàng)建 Celery 實例
celery = Celery(
    app.import_name,
    backend=app.config['CELERY_RESULT_BACKEND'],
    broker=app.config['CELERY_BROKER_URL']
)

# 將 Celery 實例與 Flask 應(yīng)用關(guān)聯(lián)
celery.conf.update(app.config)

# 確保在創(chuàng)建 Celery 實例之前配置好所有選項
@celery.task()
def add(x, y):
    return x + y

3. 創(chuàng)建任務(wù)

在你的 Flask 應(yīng)用中創(chuàng)建一個任務(wù)。例如，你可以創(chuàng)建一個簡單的任務(wù)來添加兩個數(shù)：

@app.route('/add/<int:x>/<int:y>')
def add_numbers(x, y):
    task = add.delay(x, y)
    return f"Task ID: {task.id}"

4. 運行 Celery Worker

啟動 Celery worker 來處理任務(wù)隊列中的任務(wù)。你可以在終端中運行以下命令：

celery -A app.celery worker --loglevel=info

5. 測試任務(wù)

現(xiàn)在你可以訪問 /add/<int:x>/<int:y> 路由來測試任務(wù)是否正常工作。例如，訪問 http://127.0.0.1:5000/add/3/4 應(yīng)該返回類似以下的響應(yīng)：

Task ID: 9f8d4e0b-6a9b-434d-9b5d-41b4a8b3f84a

6. 查看任務(wù)狀態(tài)

你可以使用 Celery 的管理界面或命令行工具來查看任務(wù)的狀態(tài)：

celery -A app.celery inspect active

總結(jié)

通過以上步驟，你已經(jīng)成功地將 Celery 集成到 Flask 應(yīng)用中，并創(chuàng)建了一個簡單的任務(wù)來添加兩個數(shù)。你可以根據(jù)需要擴展這個方案，添加更多的任務(wù)和復(fù)雜的后臺處理邏輯。