从celery后端获取所有任务的列表

时间:2017-03-16 13:56:23

标签: python celery

我有一个芹菜应用程序,它使用后端来存储已完成任务的结果。

当任务排队/运行时,我可以获得有关它们的信息,但是在完成后,如何从结果后端获取所有任务ID的列表?

我想在python应用程序中执行此操作,并且不依赖于特定的后端(例如,我可能希望将来在文件系统和MySQL之间切换作为结果存储)。

1 个答案:

答案 0 :(得分:2)

我认为最好的解决方案是存储有关在db中完成的任务的信息。首先,它很容易处理。 以下是SQLite的示例。我们的任务表:

# You can add specific columns for args, kwargs etc.
# it is just an example 
CREATE TABLE celery_tasks (
    "id" INTEGER PRIMARY KEY,
    "task_id" TEXT NOT NULL,
    "task_name" TEXT NOT NULL,
    "state" TEXT NOT NULL,
    "created" TEXT NOT NULL
)

我们的芹菜申请tasks.py

import celery
from celery.signals import task_postrun
from celery.task import Task
import sqlite3
from datetime import datetime


@task_postrun.connect()
def task_postrun(signal=None, sender=None, task_id=None, task=None,
                 args=None, kwargs=None, retval=None, state=None):
    # For example we don't want to store info about specific tasks 
    ignored_tasks = ('tasks.ignore_task', )

    if task.name not in ignored_tasks:
        # write info about a finished task into SQLite
        conn = sqlite3.connect('db')
        conn.execute(
            'INSERT INTO celery_tasks (task_id, task_name, state, created) VALUES (?,?,?,?)',
            (task_id, task.name, state, datetime.now())
        )

        conn.commit()
        conn.close()


app = celery.Celery(
   'tasks',
   broker='redis://localhost:6379/0',
   backend='redis://localhost:6379/0',
)


@app.task
def success_task():
    pass


@app.task
def failure_task():
    raise Exception('something wrong')


@app.task
def ignore_task():
    """
    Example of the task that we want to ignore for SQLite.
    """
    pass

run_tasks.py

from tasks import success_task, failure_task, ignore_task

success_task.delay()
failure_task.delay()
ignore_task.delay()

因此,在此之后,您可以使用常规SQL查询在代码的任何位置获取有关已完成任务的任何信息。(SELECT * FROM celery_tasks WHERE created ... AND ...

您也可以不时清理表格。 我认为使用db是一个很好的解决方案。

还有一种方法。

您可以配置CELERY_RESULT_BACKEND设置。在这种情况下,celery将创建celery_tasksetmetacelery_taskmeta个表。任务数据将自动实现:

app = Celery(
    'app_name',
    broker='CELERY_BROKER_URL...',
    backend='db+mysql://credentials...',
)

希望这有帮助。