multiprocessing.Pool：启动任务时收到通知

Question

我像这样使用multiprocessing.Pool来执行许多任务。

def execute(task):
    # run task, return result
def on_completion(task_result):
    # process task result

async_results = [pool.apply_async(execute,
                                  args=[task],
                                  callback=on_completion)
                 for task in self.tasks]

# wait for results

我的完成处理程序由池以一种很好的，串行化的方式调用，因此我不必担心其实现中的线程安全性。

但是，我也想在任务开始时收到通知。是否有一种优雅的方法可以完成以下任务？

def on_start(arg): # Whatever arg(s) were passed to the execute function
    # Called when task starts to run

pool.apply_async(run_task,
                 args=[task],
                 start_callback=on_start,
                 completion_callback=on_completion)