Merge pull request #13 from gofynd/FPCO-30582-retry-middleware--main

ID:FPCO-30582;DONE:100;HOURS:4; Add retry method for server unavailib…

Author: brijeshgajjarfynd

fdk_extension/extension.py

@@ -8,6 +8,7 @@
 from fdk_client.application.ApplicationClient import ApplicationClient
 from fdk_client.common.utils import get_headers_with_signature
 from fdk_client.common.aiohttp_helper import AiohttpHelper
+from .utilities.aiohttp_retry import retry_middleware
 
 from . import __version__
 from .constants import ONLINE_ACCESS_MODE, OFFLINE_ACCESS_MODE, FYND_CLUSTER
@@ -208,7 +209,7 @@ async def get_extension_details(self) -> dict:
                 headers=headers,
                 exclude_headers=list(headers.keys())
             )
-            response = await AiohttpHelper().aiohttp_request(request_type="GET", url=url, headers=headers)
+            response = await retry_middleware(AiohttpHelper().aiohttp_request, request_type="GET", url=url, headers=headers)
             if response["status_code"] == 200:
                 return response["json"]
             else:

fdk_extension/utilities/aiohttp_retry.py

@@ -0,0 +1,28 @@
+import asyncio
+
+async def retry_middleware(handler, *args, **kwargs):
+    initial_delay = 30
+    delay = initial_delay
+    max_retries = float('inf')  # Retry forever
+    allowed_codes = (502, 503, 504)
+
+    attempt = 0
+    while True:
+        attempt += 1
+        try:
+            # Call the handler to execute the actual request
+            response = await handler(*args, **kwargs)
+            status_code = response.get('status_code')
+            if status_code in allowed_codes:
+                if attempt <= 3:
+                    delay = initial_delay
+                else:
+                    delay = (attempt - 3) * 60  # Increase delay after initial retries
+                print(f"RetryInterceptor: Retrying request after {delay} seconds. Attempt: {attempt} Code: {status_code}")
+                await asyncio.sleep(delay)
+            else:
+                return response
+        except Exception as e:
+            raise Exception(e)
+
+    raise Exception(f"Failed to request after {max_retries} retries")

fdk_extension/webhook.py

@@ -13,6 +13,7 @@
 from .exceptions import FdkWebhookProcessError
 from .exceptions import FdkWebhookRegistrationError
 from .utilities.logger import get_logger
+from .utilities.aiohttp_retry import retry_middleware
 
 from fdk_client.common.aiohttp_helper import AiohttpHelper
 from fdk_client.common.utils import get_headers_with_signature
@@ -344,7 +345,7 @@ async def get_event_config(self, handler_config: dict) -> dict:
                 body=data,
                 exclude_headers=list(headers.keys())
             )
-            response = await AiohttpHelper().aiohttp_request(request_type="POST", url=url, data=data, headers=headers)
+            response = await retry_middleware(AiohttpHelper().aiohttp_request, request_type="POST", url=url, data=data, headers=headers)
             response_data: dict = response["json"]
             event_config["event_configs"] = response_data.get("event_configs")
             logger.debug(f"Webhook events config received: {ujson.dumps(response_data)}")