fix: add retry for exception

segateway · Sep 1, 2023 · 347fdff · 347fdff
1 parent 68b3461
commit 347fdff
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/microsoft_azure_eventhub_source/LogSourcePlugin.py b/microsoft_azure_eventhub_source/LogSourcePlugin.py
@@ -2,9 +2,11 @@
 import logging
 import os
 from datetime import datetime
+import backoff
 
 import orjson
 from azure.eventhub import EventData, PartitionContext, TransportType
+from azure.eventhub.exceptions import EventHubError
 from azure.eventhub.aio import EventHubConsumerClient
 from azure.eventhub.extensions.checkpointstoreblobaio import BlobCheckpointStore
 from dotenv import load_dotenv
@@ -111,6 +113,8 @@ def request_exit(self):
         logger.info("Exit called by syslog-ng")
         self._cancelled = True
 
+    @backoff.on_exception(backoff.expo,
+                      EventHubError)
     async def run_async(self):
         """Actual start of process"""
 

diff --git a/poetry.toml b/poetry.toml
@@ -0,0 +1,2 @@
+[virtualenvs]
+in-project = true