Add exception handling during doProgress call in HandlerContextImpl to prevent unhandled errors. This seems to be the cause of so many unhandled suspension exceptions, and possibly the reason for threads left hanging.

Author: slinkydeveloper

sdk-core/src/main/java/dev/restate/sdk/core/HandlerContextImpl.java

@@ -399,7 +399,14 @@ private void pollAsyncResultInner(AsyncResultInternal<?> asyncResult) {
       }
 
       // Not ready yet, let's try to do some progress
-      StateMachine.DoProgressResponse response = this.stateMachine.doProgress(uncompletedLeaves);
+      StateMachine.DoProgressResponse response;
+      try {
+        response = this.stateMachine.doProgress(uncompletedLeaves);
+      } catch (Throwable e) {
+        this.failWithoutContextSwitch(e);
+        asyncResult.publicFuture().completeExceptionally(AbortedExecutionException.INSTANCE);
+        return;
+      }
 
       if (response instanceof StateMachine.DoProgressResponse.AnyCompleted) {
         // Let it loop now

sdk-core/src/main/java/dev/restate/sdk/core/statemachine/StateMachineImpl.java

@@ -152,7 +152,6 @@ public void onNext(Slice slice) {
 
   @Override
   public void onError(Throwable throwable) {
-    LOG.trace("Got failure", throwable);
     this.stateContext.getCurrentState().hitError(throwable, null, null, this.stateContext);
     cancelInputSubscription();
   }