Skip formatting errors in backend error scenarios

wso2 · Nov 14, 2024 · f8afe23 · f8afe23
1 parent 3b124d3
commit f8afe23
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/...ateway/src/main/java/org/wso2/openbanking/cds/gateway/mediators/GatewayErrorMediator.java b/...ateway/src/main/java/org/wso2/openbanking/cds/gateway/mediators/GatewayErrorMediator.java
@@ -82,6 +82,9 @@ public boolean mediate(MessageContext messageContext) {
 
         // Error handling logic.
         JSONObject errorData;
+        String statusCodeString = extractStatusCode(messageContext);
+        int statusCode = Integer.parseInt(statusCodeString);
+
         if ((messageContext.getProperty(GatewayConstants.ERROR_CODE)) != null) {
 
             int errorCode = (int) messageContext.getProperty(GatewayConstants.ERROR_CODE);
@@ -109,12 +112,14 @@ public boolean mediate(MessageContext messageContext) {
             // Assume the errors thrown from the relevant internal webapps of these APIs are
             // already formatted according to the CDS format
             return true;
+        } else if (messageContext.getProperty(GatewayConstants.ENDPOINT_ADDRESS) != null &&
+                statusCode != HttpStatus.SC_REQUEST_TIMEOUT) {
+            // Assume the errors coming from the backend are properly formatted.
+            return true;
         } else {
-            String statusCodeString = extractStatusCode(messageContext);
             if (StringUtils.isBlank(statusCodeString)) {
                 return true;
             }
-            int statusCode = Integer.parseInt(statusCodeString);
             ErrorConstants.AUErrorEnum errorEnum;
             if ("406".equals(statusCodeString)) {
                 errorEnum = ErrorConstants.AUErrorEnum.INVALID_ACCEPT_HEADER;

diff --git a/...ng.cds.gateway/src/main/java/org/wso2/openbanking/cds/gateway/utils/GatewayConstants.java b/...ng.cds.gateway/src/main/java/org/wso2/openbanking/cds/gateway/utils/GatewayConstants.java
@@ -137,6 +137,8 @@ private GatewayConstants() {
     public static final String REST_METHOD = "REST_METHOD";
     public static final String OAUTH_JWT_ASSERTION = "client_assertion";
     public static final String CLIENT_ID = "client_id";
+    public static final String ENDPOINT_ADDRESS = "ENDPOINT_ADDRESS";
+
 
     // Constants related to executors
     public static final String HTTP_GET = "GET";