EventProcessor draft 3

sdk/eventhub/azure-eventhubs/azure/eph/__init__.py → sdk/eventhub/azure-eventhubs/azure/eventhub/eventprocessor/__init__.py

@@ -1,3 +1,8 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# -----------------------------------------------------------------------------------
+
 from .event_processor import EventProcessor
 from .partition_processor import PartitionProcessor
 from .partition_manager import PartitionManager

sdk/eventhub/azure-eventhubs/azure/eventhub/eventprocessor/checkpoint_manager.py

@@ -0,0 +1,31 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# -----------------------------------------------------------------------------------
+
+
+from .partition_manager import PartitionManager
+
+
+class CheckpointManager(object):
+    """Every PartitionProcessor has a CheckpointManager to save the partition's checkpoint.
+
+    """
+    def __init__(self, partition_id, eventhub_name, consumer_group_name, instance_id, partition_manager: PartitionManager):
+        self._partition_id = partition_id
+        self._eventhub_name = eventhub_name
+        self._consumer_group_name = consumer_group_name
+        self._instance_id = instance_id
+        self._partition_manager = partition_manager
+
+    async def update_checkpoint(self,
+                                offset, sequence_number):
+        """Users call this method in PartitionProcessor.process_events() to save checkpoints
+
+        :param offset: offset of the processed EventData
+        :param sequence_number: sequence_number of the processed EventData
+        :return: None
+        """
+        await self._partition_manager.\
+            update_checkpoint(self._eventhub_name, self._consumer_group_name, self._partition_id, self._instance_id,
+            offset, sequence_number)

sdk/eventhub/azure-eventhubs/azure/eventhub/eventprocessor/close_reason.py

@@ -0,0 +1,13 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# -----------------------------------------------------------------------------------
+
+from enum import Enum
+
+
+class CloseReason(Enum):
+    SHUTDOWN = 0  # user call EventProcessor.stop()
+    LEASE_LOST = 1  # lose the ownership of a partition.
+    EVENTHUB_EXCEPTION = 2  # Exception happens during receiving events
+    USER_EXCEPTION = 3  # user's code in EventProcessor.process_events() raises an exception