Skip to content

Commit 9dee4ad

Browse files
ID:FPCO-30582;DONE:100;HOURS:4; Add dynamic arguments in middleware
1 parent 8b16f4d commit 9dee4ad

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

fdk_extension/utilities/aiohttp_retry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import asyncio
22

3-
async def retry_middleware(handler, request_type, url, headers, *args):
3+
async def retry_middleware(handler, *args, **kwargs):
44
initial_delay = 30
55
delay = initial_delay
66
max_retries = float('inf') # Retry forever
@@ -11,7 +11,7 @@ async def retry_middleware(handler, request_type, url, headers, *args):
1111
attempt += 1
1212
try:
1313
# Call the handler to execute the actual request
14-
response = await handler(request_type, url, headers, *args)
14+
response = await handler(*args, **kwargs)
1515
status_code = response.get('status_code')
1616
if status_code in allowed_codes:
1717
if attempt <= 3:

fdk_extension/webhook.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ async def get_event_config(self, handler_config: dict) -> dict:
345345
body=data,
346346
exclude_headers=list(headers.keys())
347347
)
348-
response = await retry_middleware(AiohttpHelper().aiohttp_request, request_type="POST", url=url, headers=headers, data=data)
348+
response = await retry_middleware(AiohttpHelper().aiohttp_request, request_type="POST", url=url, data=data, headers=headers)
349349
response_data: dict = response["json"]
350350
event_config["event_configs"] = response_data.get("event_configs")
351351
logger.debug(f"Webhook events config received: {ujson.dumps(response_data)}")

0 commit comments

Comments
 (0)