Attempting to resolve consumer issue with rabbitmq reconnections

objectrocket · Nov 28, 2017 · 15a2626 · 15a2626
1 parent c987c5b
commit 15a2626
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/taskflow/engines/worker_based/proxy.py b/taskflow/engines/worker_based/proxy.py
@@ -216,6 +216,10 @@ def _drain_errback(exc, interval):
                 self._running.set()
                 try:
                     while self._running.is_set():
+                        # This seems to be required when failures occur in
+                        #  rabbitmq, as the consumer doesn't show up in rabbit
+                        #  as a valid consumer without it. (mmontgomery)
+                        consumer.consume()
                         safe_drain(conn, self._drain_events_timeout)
                         if self._on_wait is not None:
                             self._on_wait()