Properly wait until the inbound queue is empty #63

eandersson · Oct 17, 2018 · a6418d7 · a6418d7
1 parent bc9af98
commit a6418d7
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/amqpstorm/channel.py b/amqpstorm/channel.py
@@ -112,10 +112,13 @@ def build_inbound_messages(self, break_on_empty=False, to_tuple=False,
         while not self.is_closed:
             message = self._build_message(auto_decode=auto_decode)
             if not message:
+                sleep(IDLE_WAIT)
+                self.check_for_errors()
                 if break_on_empty:
+                    sleep(IDLE_WAIT * 10)
+                    if self._inbound:
+                        continue
                     break
-                self.check_for_errors()
-                sleep(IDLE_WAIT)
                 continue
             if to_tuple:
                 yield message.to_tuple()