Retry backend request on 502 return

NeonGeckoCom · May 31, 2024 · d24db79 · d24db79
1 parent 9dfa7fa
commit d24db79
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/neon_utils/hana_utils.py b/neon_utils/hana_utils.py
@@ -162,9 +162,14 @@ def request_backend(endpoint: str, request_data: dict,
         except ServerException as e:
             LOG.error(e)
             _get_token(server_url)
-    resp = requests.post(f"{server_url}/{endpoint.lstrip('/')}",
-                         json=request_data, headers=_headers)
+    request_kwargs = {"url": f"{server_url}/{endpoint.lstrip('/')}",
+                      "json": request_data, "headers": _headers}
+    resp = requests.post(**request_kwargs)
+    if resp.status_code == 502:
+        # This is raised occasionally on valid requests. Need to resolve in HANA
+        resp = requests.post(**request_kwargs)
     if resp.ok:
         return resp.json()
+
     else:
         raise ServerException(f"Error response {resp.status_code}: {resp.text}")