addeddelay

Author: Andrei Neagu

packages/service-library/src/servicelib/deferred_tasks/_base_deferred_handler.py

@@ -46,11 +46,14 @@ async def get_retries(cls, context: DeferredContext) -> NonNegativeInt:
         return 0
 
     @classmethod
-    async def get_retry_delay(cls, context: DeferredContext) -> timedelta:
+    async def get_retry_delay(
+        cls, context: DeferredContext, remaining_attempts: NonNegativeInt
+    ) -> timedelta:
         """
         returns: the delay between eatch retry attempt (default: 0s)
         """
         assert context  # nosec
+        assert remaining_attempts  # nosec
         return timedelta(seconds=0)
 
     @classmethod

packages/service-library/src/servicelib/deferred_tasks/_deferred_manager.py

@@ -460,7 +460,15 @@ async def _fs_handle_error_result(  # pylint:disable=method-hidden
             task_schedule.result, TaskResultCancelledError
         ):
             _logger.debug("Schedule retry attempt for task_uid '%s'", task_uid)
-            # does not retry if task was cancelled
+
+            subclass = self.__get_subclass(task_schedule.class_unique_reference)
+            deferred_context = self.__get_deferred_context(task_schedule.start_context)
+            sleep_interval = await subclass.get_retry_delay(
+                context=deferred_context,
+                remaining_attempts=task_schedule.execution_attempts,
+            )
+            await asyncio.sleep(sleep_interval.total_seconds())
+
             task_schedule.state = TaskState.SUBMIT_TASK
             await self._task_tracker.save(task_uid, task_schedule)
             await self.__publish_to_queue(task_uid, _FastStreamRabbitQueue.SUBMIT_TASK)