Improve exception logic

ballerina-platform · Nov 19, 2024 · 0880179 · 0880179
1 parent aa40a35
commit 0880179
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/native/src/main/java/io/ballerina/stdlib/graphql/runtime/utils/ModuleUtils.java b/native/src/main/java/io/ballerina/stdlib/graphql/runtime/utils/ModuleUtils.java
@@ -33,7 +33,8 @@ public class ModuleUtils {
 
     private static Module module;
 
-    private ModuleUtils() {}
+    private ModuleUtils() {
+    }
 
     public static void setModule(Environment environment) {
         module = environment.getCurrentModule();
@@ -46,6 +47,11 @@ public static Module getModule() {
     public static Object getResult(CompletableFuture<Object> balFuture) {
         try {
             return balFuture.get();
+        } catch (BError error) {
+            throw error;
+        } catch (InterruptedException e) {
+            Thread.currentThread().interrupt();
+            throw ErrorCreator.createError(e);
         } catch (Throwable throwable) {
             throw ErrorCreator.createError(throwable);
         }