Modify SuspendedException to subclass Throwable rather than RuntimeEx…

…ception (#124)
restatedev · Oct 26, 2023 · 0ee6f08 · 0ee6f08
1 parent fe4bf82
commit 0ee6f08
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/sdk-core/src/main/java/dev/restate/sdk/core/SuspendedException.java b/sdk-core/src/main/java/dev/restate/sdk/core/SuspendedException.java
@@ -1,9 +1,14 @@
 package dev.restate.sdk.core;
 
-public final class SuspendedException extends RuntimeException {
+public final class SuspendedException extends Throwable {
   @SuppressWarnings("StaticAssignmentOfThrowable")
   public static final SuspendedException INSTANCE = new SuspendedException();
 
+  @SuppressWarnings("unchecked")
+  public static <E extends Throwable> void sneakyThrow() throws E {
+    throw (E) SuspendedException.INSTANCE;
+  }
+
   private SuspendedException() {
     super("SuspendedException");
     setStackTrace(new StackTraceElement[] {});

diff --git a/sdk-java-blocking/src/main/java/dev/restate/sdk/blocking/Util.java b/sdk-java-blocking/src/main/java/dev/restate/sdk/blocking/Util.java
@@ -17,7 +17,8 @@ static <T> T awaitCompletableFuture(CompletableFuture<T> future) {
     try {
       return future.get();
     } catch (InterruptedException | CancellationException e) {
-      throw SuspendedException.INSTANCE;
+      SuspendedException.sneakyThrow();
+      return null; // Previous statement throws an exception
     } catch (ExecutionException e) {
       throw (RuntimeException) e.getCause();
     }