Celery - 一个任务运行N个任务,等待它们并处理结果

时间:2017-08-29 22:19:13

标签: python celery

我需要以下工作流程:

    首先运行
  • ParentTask
  • 在某些时候,它产生了ChildTask的N个实例,它们并行运行
  • ParentTask等待那些完成,收集结果,以某种方式处理它们并完成

这似乎很容易。不幸的是,从任务中调用Task().delay()(我用来调用任务)似乎完全被忽略了。我完全迷失在这里。

如果您更喜欢代码方法,我也会将其包含在内。

from celery.task import Task
from celery.result import AsyncResult

class ParentTask(Task):
    def run(self, *args, **kwargs):
        # do some stuff
        ids = [ChildTask().delay().id for _ in range(N)] # this seems to do nothing here
        results = [AsyncResult(t) for t in ids]
        while not all([r.ready() for r in results]): # wait for child tasks to finish
            sleep(.100)
        # do some stuff again
        # return results

class ChildTask(Task):
    def run(self, *args, **kwargs):
        # do some child stuff
        # return child results

ParentTask().delay() # this delay works fine

感谢您提供任何线索!

1 个答案:

答案 0 :(得分:0)

好的,我明白了。一种工作方法可能看起来像这样(当然,任务可以做任何需要的东西):

from time import sleep
from celery.task import Task
from celery import chain, group

class PreTask(Task):
    def run(self, *args, **kwargs):
        x = 0
        for i in range(100000):
            x += 1
        return x


class MidTask(Task):
    def run(self, *args, **kwargs):
        sleep(5)
        return 42


class PostTask(Task):
    def run(self, *args, **kwargs):
        return args


# call it like this
res = chain(PreTask().s() | group(MidTask().s() for _ in range(5)) | PostTask().s()).apply_async()

# and get the result for example like this
while(True):
    if res.ready():
        print(res.get())
    sleep(1)

希望它有所帮助。