fix: Disabled excessive throttling for BatchSubscriber.

airtai · Dec 21, 2024 · 01a4b2c · 01a4b2c
1 parent 51477a5
commit 01a4b2c
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/faststream/confluent/subscriber/usecase.py b/faststream/confluent/subscriber/usecase.py
@@ -378,18 +378,14 @@ def __init__(
 
     async def get_msg(self) -> Optional[Tuple["Message", ...]]:
         assert self.consumer, "You should setup subscriber at first."  # nosec B101
-
-        messages = await self.consumer.getmany(
-            timeout=self.polling_interval,
-            max_records=self.max_records,
+        return (
+            await self.consumer.getmany(
+                timeout=self.polling_interval,
+                max_records=self.max_records,
+            )
+            or None
         )
 
-        if not messages:  # TODO: why we are sleeping here?
-            await anyio.sleep(self.polling_interval)
-            return None
-
-        return messages
-
     def get_log_context(
         self,
         message: Optional["StreamMessage[Tuple[Message, ...]]"],