Skip to content

Commit d16cee2

Browse files
committed
refactor(receiver): simplify prefetch queue cleanup
1 parent a37ac15 commit d16cee2

1 file changed

Lines changed: 2 additions & 5 deletions

File tree

taskiq/receiver/receiver.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -637,11 +637,8 @@ async def _discard_queued_messages(
637637
) -> None:
638638
"""Release capacity for abandoned queue entries."""
639639
discarded_messages = 0
640-
while True:
641-
try:
642-
queued_message = queue.get_nowait()
643-
except asyncio.QueueEmpty:
644-
break
640+
while not queue.empty():
641+
queued_message = queue.get_nowait()
645642
if isinstance(queued_message, _PrefetchedMessage):
646643
discarded_messages += 1
647644
if queued_message.owns_delivery_slot:

0 commit comments

Comments
 (0)