From d4e1a7a0707986bfba25a2579f7496cbf60d50dc Mon Sep 17 00:00:00 2001 From: Salad Dais Date: Mon, 3 May 2021 02:07:03 +0000 Subject: [PATCH] Fix queue consumption under 3.9 --- hippolyzer/lib/base/message/message_handler.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/hippolyzer/lib/base/message/message_handler.py b/hippolyzer/lib/base/message/message_handler.py index 27b0d69..19bb2ce 100644 --- a/hippolyzer/lib/base/message/message_handler.py +++ b/hippolyzer/lib/base/message/message_handler.py @@ -80,11 +80,10 @@ class MessageHandler(Generic[_T]): notifiers = self._subscribe_all(message_names, _handler_wrapper, predicate=predicate) async def _get_wrapper(): - try: - return await msg_queue.get() - finally: - # Consumption is completion - msg_queue.task_done() + msg = await msg_queue.get() + # Consumption is completion + msg_queue.task_done() + return msg try: yield _get_wrapper