Increase asyinc waiting time for RabbitMQ subscriber (#168)

* Increase asying waiting time for RabbitMQ subscriber * Await `callback()` as well Co-authored-by: Stef Piatek <[email protected]> --------- Co-authored-by: Stef Piatek <[email protected]>
SAFEHR-data · Dec 4, 2023 · 1405b86 · 1405b86
1 parent 0195754
commit 1405b86
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/pixl_core/src/core/patient_queue/subscriber.py b/pixl_core/src/core/patient_queue/subscriber.py
@@ -60,10 +60,8 @@ async def run(self, callback: Callable[[bytes], Awaitable[None]]) -> None:
         async with self._queue.iterator() as queue_iter:
             async for message in queue_iter:
                 if not self.token_bucket.has_token:
-                    await asyncio.gather(
-                        message.reject(requeue=True),
-                        asyncio.sleep(1e-3),
-                    )
+                    await asyncio.sleep(0.01)
+                    await message.reject(requeue=True)
                     continue
 
                 # Messages need to be acknowledged before a callback otherwise
@@ -72,10 +70,8 @@ async def run(self, callback: Callable[[bytes], Awaitable[None]]) -> None:
                 await message.ack()
 
                 try:
-                    await asyncio.gather(
-                        callback(message.body),
-                        asyncio.sleep(1e-3),  # Avoid very fast callbacks
-                    )
+                    await asyncio.sleep(0.01)  # Avoid very fast callbacks
+                    await callback(message.body)
                 except Exception:
                     LOGGER.exception(
                         "Failed to process %s" "Not re-queuing message",