From 4fc6f53feed6f4371449b80e452571b5504b8b43 Mon Sep 17 00:00:00 2001 From: Bulat Kurbanov Date: Sun, 21 May 2023 01:21:34 +0200 Subject: [PATCH] Fix --- src/core/taskiq_middlewares.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/core/taskiq_middlewares.py b/src/core/taskiq_middlewares.py index f54f129..0321be5 100644 --- a/src/core/taskiq_middlewares.py +++ b/src/core/taskiq_middlewares.py @@ -12,14 +12,11 @@ logger = logging.getLogger("taskiq_middleware") class FastAPIREtryMiddleware(SimpleRetryMiddleware): @staticmethod def _is_need_to_remove(to_remove: list[Any], value: Any) -> bool: - logger.info(f"{type(value)}, {to_remove}") return type(value) in to_remove async def on_error( self, message: TaskiqMessage, result: TaskiqResult[Any], exception: Exception ) -> None: - logger.info(f"{self.broker.custom_dependency_context}") - types_to_remove = list(self.broker.custom_dependency_context.keys()) message.args = [ @@ -33,4 +30,6 @@ class FastAPIREtryMiddleware(SimpleRetryMiddleware): if not self._is_need_to_remove(types_to_remove, value) } + raise Exception(f"{self.broker.custom_dependency_context=} {message=}") + return await super().on_error(message, result, exception)