Addressed PR Comments

Signed-off-by: Ryan Bogan <rbogan@amazon.com>
opensearch-project · Jan 17, 2023 · 5088ce4 · 5088ce4
1 parent 61d44d3
commit 5088ce4
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/server/src/main/java/org/opensearch/extensions/action/ExtensionTransportActionsHandler.java b/server/src/main/java/org/opensearch/extensions/action/ExtensionTransportActionsHandler.java
@@ -201,7 +201,13 @@ public String executor() {
             if (e.getCause() instanceof TimeoutException) {
                 logger.info("No response from extension to request.");
             }
-            throw Exception.class.cast(e.getCause());
+            if (e.getCause() instanceof RuntimeException) {
+                throw (RuntimeException) e.getCause();
+            } else if (e.getCause() instanceof Error) {
+                throw (Error) e.getCause();
+            } else {
+                throw new RuntimeException(e.getCause());
+            }
         }
         return extensionActionResponse;
     }