added ability to toggle pump shutdown when all messages on a pump are…

… processed.
yunhaoling · Nov 19, 2017 · c18145b · c18145b
1 parent 963be97
commit c18145b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/eventhubsprocessor/eh_partition_pump.py b/eventhubsprocessor/eh_partition_pump.py
@@ -111,9 +111,9 @@ async def run(self):
                     logging.info("No events received, queue size %d, delivered %d",
                                 self.eh_partition_pump.partition_receive_handler.messages.qsize(),
                                 self.eh_partition_pump.partition_receive_handler.delivered)
-                await self.process_error_async(err)
-        # Handle close
-
+                if self.eh_partition_pump.host.eh_options.release_pump_on_timeout:
+                    await self.process_error_async(err)
+    
     async def process_events_async(self, events):
         """
         # This method is called on the thread that the EH client uses to run the pump.

diff --git a/eventhubsprocessor/eph.py b/eventhubsprocessor/eph.py
@@ -49,4 +49,5 @@ def __init__(self):
         self.max_batch_size = 10
         self.prefetch_count = 300
         self.receive_timeout = 60
+        self.release_pump_on_timeout = True
         self.initial_offset_provider = "-1"