Merge pull request #1183 from SEKOIA-IO/fix/azure_stop_method

Fix: Azure stop method

Azure/CHANGELOG.md

@@ -7,6 +7,14 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## Unreleased
 
+## 2024-11-20 - 2.6.3
+
+### Fixed
+
+- Remove shutdown to simplify the workflow
+- Fix the way to handle stop
+
+
 ## 2024-11-15 - 2.6.2
 
 ### Fixed

Azure/connectors/azure_eventhub.py

@@ -13,7 +13,7 @@
 from azure.eventhub.aio import EventHubConsumerClient, PartitionContext
 from azure.eventhub.extensions.checkpointstoreblobaio import BlobCheckpointStore
 from sekoia_automation.aio.connector import AsyncConnector
-from sekoia_automation.connector import DefaultConnectorConfiguration
+from sekoia_automation.connector import DefaultConnectorConfiguration, Connector
 
 from .metrics import EVENTS_LAG, FORWARD_EVENTS_DURATION, INCOMING_MESSAGES, MESSAGES_AGE, OUTCOMING_EVENTS
 
@@ -165,6 +165,12 @@ async def receive_events(self) -> None:
             max_wait_time=self._consumption_max_wait_time,
         )
 
+    def stop(self, *args: Any, **kwargs: Optional[Any]) -> None:  # pragma: no cover
+        """
+        Stop the connector
+        """
+        super(Connector, self).stop(*args, **kwargs)
+
     async def async_run(self) -> None:  # pragma: no cover
         while self.running:
             try:
@@ -174,10 +180,12 @@ async def async_run(self) -> None:  # pragma: no cover
                 self.log_exception(ex, message="Failed to consume messages")
                 self._has_more_events = False
 
+            await self.client.close()
+
             if not self._has_more_events:
                 await asyncio.sleep(self._frequency)
 
-        await self.client.close()
+        await self._session.close()
 
     def run(self) -> None:  # pragma: no cover
         self.log("Azure EventHub Trigger has started")

Azure/manifest.json

@@ -8,7 +8,7 @@
   "name": "Microsoft Azure",
   "uuid": "525eecc0-9eee-484d-92bd-039117cf4dac",
   "slug": "azure",
-  "version": "2.6.2",
+  "version": "2.6.3",
   "categories": [
     "Cloud Providers"
   ]