|
@@ -313,15 +313,19 @@ class TaskWrapper(object):
|
|
|
# Make sure task has not already been executed.
|
|
|
self._set_executed_bit()
|
|
|
|
|
|
- self.send_event("task-accepted", uuid=self.task_id)
|
|
|
-
|
|
|
args = self._get_tracer_args(loglevel, logfile)
|
|
|
self.time_start = time.time()
|
|
|
result = pool.apply_async(execute_and_trace, args=args,
|
|
|
+ accept_callback=self.on_accepted,
|
|
|
callbacks=[self.on_success], errbacks=[self.on_failure])
|
|
|
- self.on_ack()
|
|
|
return result
|
|
|
|
|
|
+ def on_accepted(self):
|
|
|
+ self.on_ack()
|
|
|
+ self.send_event("task-accepted", uuid=self.task_id)
|
|
|
+ self.logger.debug("Task accepted: %s[%s]" % (
|
|
|
+ self.task_name, self.task_id))
|
|
|
+
|
|
|
def on_success(self, ret_value):
|
|
|
"""The handler used if the task was successfully processed (
|
|
|
without raising an exception)."""
|