Fix queue consumption under 3.9

2021-05-03 02:07:03 +00:00
parent d401842eef
commit d4e1a7a070
1 changed files with 4 additions and 5 deletions
--- 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