[Integration][Gitlab] Add timeout to merge requests

integrations/gitlab/gitlab_integration/events/hooks/base.py

@@ -11,6 +11,16 @@
     GitlabPortAppConfig,
     GitlabMemberSelector,
 )
+import asyncio
+from tenacity import retry, stop_after_attempt, wait_fixed, wait_exponential, retry_if_exception_type
+
+
+class HookConfig:
+    TIMEOUT = 120  # seconds
+    BACKOFF_EXPONENTIAL = 2
+    BACKOFF_MAX_ATTEMPTS = 5
+    BACKOFF_MAX_TIME = 60  # seconds
+    BACKOFF_MIN_TIME = 1  # seconds
 
 
 class HookHandler(ABC):
@@ -64,6 +74,15 @@ async def _register_object_with_members(self, kind: str, gitlab_object: RESTObje
 
 
 class ProjectHandler(HookHandler):
+    @retry(
+        wait=wait_exponential(
+            multiplier=HookConfig.BACKOFF_EXPONENTIAL,
+            min=HookConfig.BACKOFF_MIN_TIME,
+            max=HookConfig.BACKOFF_MAX_TIME,
+        ),
+        retry = retry_if_exception_type(asyncio.TimeoutError),
+        stop=stop_after_attempt(HookConfig.BACKOFF_MAX_ATTEMPTS),
+    )
     async def on_hook(self, event: str, body: dict[str, Any]) -> None:
         logger.info(f"Handling {event}")
 
@@ -77,8 +96,15 @@ async def on_hook(self, event: str, body: dict[str, Any]) -> None:
                 f"Handling hook {event} for project {project.path_with_namespace}"
             )
             try:
-                await self._on_hook(body, project)
+                asyncio.wait_for(
+                    self._on_hook(body, project), timeout=HookConfig.TIMEOUT
+                )
                 logger.info(f"Finished handling {event}")
+            except asyncio.TimeoutError:
+                logger.error(
+                    f"Timeout while handling hook {event} for project {project.path_with_namespace}"
+                )
+                raise
             except Exception as e:
                 logger.error(
                     f"Error handling hook {event} for project {project.path_with_namespace}. Error: {e}"
@@ -94,14 +120,27 @@ async def _on_hook(self, body: dict[str, Any], gitlab_project: Project) -> None:
 
 
 class GroupHandler(HookHandler):
+    @retry(
+        retry=retry_if_exception_type(asyncio.TimeoutError),
+        wait=wait_exponential(
+            multiplier=HookConfig.BACKOFF_EXPONENTIAL,
+            min=HookConfig.BACKOFF_MIN_TIME,
+            max=HookConfig.BACKOFF_MAX_TIME,
+        ),
+        stop=stop_after_attempt(HookConfig.BACKOFF_MAX_ATTEMPTS),
+    )
     async def on_hook(self, event: str, body: dict[str, Any]) -> None:
         logger.info(f"Handling {event}")
 
-        group_id = body.get("group_id", body.get("group", {}).get("id"))
-        group = await self.gitlab_service.get_group(group_id)
-        await self._on_hook(body, group)
-        group_path = body.get("full_path", body.get("group_path"))
-        logger.info(f"Finished handling {event} for group {group_path}")
+        try:
+            group_id = body.get("group_id", body.get("group", {}).get("id"))
+            group = await self.gitlab_service.get_group(group_id)
+            asyncio.wait_for(self._on_hook(body, group), timeout=HookConfig.TIMEOUT)
+            group_path = body.get("full_path", body.get("group_path"))
+            logger.info(f"Finished handling {event} for group {group_path}")
+        except asyncio.TimeoutError:
+            logger.error(f"Timeout while handling hook {event} for group {group_path}")
+            raise
 
     @abstractmethod
     async def _on_hook(