Improve exception logic

ballerina-platform · Nov 19, 2024 · 64e448f · 64e448f
1 parent a98fbbe
commit 64e448f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/native/src/main/java/io/ballerina/stdlib/persist/sql/ModuleUtils.java b/native/src/main/java/io/ballerina/stdlib/persist/sql/ModuleUtils.java
@@ -21,6 +21,7 @@
 import io.ballerina.runtime.api.Environment;
 import io.ballerina.runtime.api.Module;
 import io.ballerina.runtime.api.creators.ErrorCreator;
+import io.ballerina.runtime.api.values.BError;
 
 import java.util.concurrent.CompletableFuture;
 
@@ -47,6 +48,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);
         }