added timeout to middleware requests

care/utils/assetintegration/base.py

@@ -16,6 +16,7 @@ def __init__(self, meta):
         self.host = self.meta["local_ip_address"]
         self.middleware_hostname = self.meta["middleware_hostname"]
         self.insecure_connection = self.meta.get("insecure_connection", False)
+        self.timeout = getattr(settings, "MIDDLEWARE_API_TIMEOUT", 20)
 
     def handle_action(self, action):
         pass
@@ -37,12 +38,17 @@ def api_post(self, url, data=None):
             url,
             json=data,
             headers=self.get_headers(),
+            timeout=self.timeout,
         )
         try:
             response = req.json()
             if req.status_code >= 400:
                 raise APIException(response, req.status_code)
             return response
+
+        except requests.Timeout:
+            raise APIException({"error": "Request Timeout"}, 504)
+
         except json.decoder.JSONDecodeError:
             raise APIException({"error": "Invalid Response"}, req.status_code)
 
@@ -51,11 +57,16 @@ def api_get(self, url, data=None):
             url,
             params=data,
             headers=self.get_headers(),
+            timeout=self.timeout,
         )
         try:
             if req.status_code >= 400:
                 raise APIException(req.text, req.status_code)
             response = req.json()
             return response
+
+        except requests.Timeout:
+            raise APIException({"error": "Request Timeout"}, 504)
+
         except json.decoder.JSONDecodeError:
             raise APIException({"error": "Invalid Response"}, req.status_code)

config/settings/base.py

@@ -6,10 +6,10 @@
 import json
 from datetime import datetime, timedelta
 from pathlib import Path
-from django.utils.translation import gettext_lazy as _
 
 import environ
 from authlib.jose import JsonWebKey
+from django.utils.translation import gettext_lazy as _
 from healthy_django.healthcheck.celery_queue_length import (
     DjangoCeleryQueueLengthHealthCheck,
 )
@@ -663,3 +663,6 @@
 TASK_SUMMARIZE_DISTRICT_PATIENT = env.bool(
     "TASK_SUMMARIZE_DISTRICT_PATIENT", default=True
 )
+
+# Timeout for middleware request (in seconds)
+MIDDLEWARE_API_TIMEOUT = 20