芹菜链任务

时间:2014-10-09 18:17:57

标签: django python-2.7 celery django-celery

上下文:

tasks.py

def chunkify(lst,n):
    return [ lst[i::n] for i in xrange(n) ]

@task
def swarm_restart(procs):
   chunks = chunkify(procs, 4)
    res = chain(
        group_restart.s(( [ (proc.name, proc.host.name) for proc in chunks[0] ] )),
        group_restart.s(( [ (proc.name, proc.host.name) for proc in chunks[1] ] )),
        group_restart.s(( [ (proc.name, proc.host.name) for proc in chunks[2] ] )),
        group_restart.s(( [ (proc.name, proc.host.name) for proc in chunks[3] ] )),
    )()

@ task
def group_restart(procs):
    # this task runs only once, seems to be called just 1 time
    res = group( proc_restart.s(proc[0], proc[1]) for proc in procs ).apply_async()

@ task
def proc_restart(proc_name, hostname):
    # This method works, tested several times
    proc = Host.objects.get(name=hostname).get_proc(proc_name)
    proc.restart()

views.py

def call():
    procs = get_procs()
    tasks.swarm_restart.delay(procs)

我得到的错误: TypeError: group_restart() takes exactly 1 argument (2 given)

我做错了什么,有灯吗?

顺便说一句。 celery == 3.0.25,django-celery == 3.0.23

1 个答案:

答案 0 :(得分:1)

如果查看swarm_restart任务,则表示您正在处理group_restart个任务。这里,链中的第一个任务将执行正常,但第二个任务将抛出错误。

  

TypeError:group_restart()只取1个参数(给定2个)

因为,第一个任务的结果作为参数传递给它。链中的下一个任务也会发生同样的情况。

例如,

from celery import task, chain

@app.task
def t1():
    return 't1'

@app.task
def t2():
   return 't2'

wrong_chain = chain( t1.s(), t2.s() )

如果执行wrong_chain,即使您没有将任何参数传递给t2

,也会执行类似的错误

因此,您必须根据自己的需要改变工作流程。