Fix little issue with doProgress loop

Author: slinkydeveloper

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

@@ -382,7 +382,7 @@ private void pollAsyncResultInner(AsyncResultInternal<?> asyncResult) {
 
       // Now let's take the unprocessed leaves
       List<Integer> uncompletedLeaves = asyncResult.uncompletedLeaves().toList();
-      if (uncompletedLeaves.size() == 1) {
+      if (uncompletedLeaves.isEmpty()) {
         // Nothing else to do!
         return;
       }

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

@@ -16,7 +16,7 @@ public class ProtocolException extends RuntimeException {
   static final int NOT_FOUND_CODE = 404;
   public static final int UNSUPPORTED_MEDIA_TYPE_CODE = 415;
   public static final int INTERNAL_CODE = 500;
-  public static final int UNSUPPORTED_FEATURE = 573;
+  @Deprecated public static final int UNSUPPORTED_FEATURE = 573;
 
   private final int code;
 
@@ -43,6 +43,7 @@ public static ProtocolException methodNotFound(String serviceName, String handle
         "Cannot find handler '" + serviceName + "/" + handlerName + "'", NOT_FOUND_CODE);
   }
 
+  @Deprecated
   public static ProtocolException idempotencyKeyIsEmpty() {
     return new ProtocolException(
         "The provided idempotency key is empty.",

sdk-core/src/main/java/dev/restate/sdk/core/sharedcore/SharedCoreInstance.java

@@ -17,6 +17,7 @@
 import com.dylibso.chicory.wasm.WasmModule;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.dataformat.cbor.databind.CBORMapper;
+import dev.restate.sdk.core.ProtocolException;
 import dev.restate.sdk.core.sharedcore.generated.SharedCoreWasmMachine;
 import java.io.IOException;
 import java.util.function.Function;
@@ -79,7 +80,7 @@ public <T> T readCborAndFree(long packed, Class<T> outputClazz) {
     try {
       return CBOR.readValue(retBytes, outputClazz);
     } catch (IOException e) {
-      throw new RuntimeException("Failed to decode CBOR", e);
+      throw new ProtocolException("Failed to decode CBOR", ProtocolException.INTERNAL_CODE, e);
     }
   }
 
@@ -96,7 +97,7 @@ public BufferPointer writeCbor(Object input) {
     try {
       cbor = CBOR.writeValueAsBytes(input);
     } catch (JsonProcessingException e) {
-      throw new RuntimeException("Failed to encode CBOR", e);
+      throw new ProtocolException("Failed to encode CBOR", ProtocolException.INTERNAL_CODE, e);
     }
     int ptr = write(cbor);
     return new BufferPointer(ptr, cbor.length);

sdk-core/src/main/java/dev/restate/sdk/core/sharedcore/SharedCoreVM.java

@@ -12,6 +12,7 @@
 import com.fasterxml.jackson.annotation.JsonTypeInfo;
 import com.fasterxml.jackson.annotation.JsonTypeInfo.As;
 import com.fasterxml.jackson.annotation.JsonTypeInfo.Id;
+import dev.restate.sdk.core.ProtocolException;
 import dev.restate.sdk.endpoint.HeadersAccessor;
 import java.util.ArrayList;
 import java.util.Collections;
@@ -64,8 +65,7 @@ public static SharedCoreVM create(HeadersAccessor headersAccessor) {
                     .collect(Collectors.toList())),
             VmNewReturn.class);
     if (newVmReturn instanceof VmNewReturn.Failure f) {
-      throw new RuntimeException(
-          "Failed to create state machine: " + f.code() + ": " + f.message());
+      throw new ProtocolException("Failed to create state machine: " + f.message(), f.code);
     }
     int vmPtr = ((VmNewReturn.Ok) newVmReturn).pointer();
     return new SharedCoreVM(instance, vmPtr);
@@ -825,8 +825,8 @@ private void callWithEmptyReturn(BiFunction<SharedCoreWasm_ModuleExports, Intege
     if (ret instanceof EmptyReturn.Failure f) throw vmError(f.code, f.message);
   }
 
-  private static RuntimeException vmError(int code, String message) {
-    return new RuntimeException("VM error " + code + ": " + message);
+  private static ProtocolException vmError(int code, String message) {
+    return new ProtocolException(message, code);
   }
 
   private static List<String[]> toHeaderList(@Nullable List<Map.Entry<String, String>> headers) {

sdk-core/src/main/rust/Cargo.lock

@@ -7,7 +7,7 @@ edition = "2021"
 crate-type = ["cdylib"]
 
 [dependencies]
-restate-sdk-shared-core = { git = "https://github.com/restatedev/sdk-shared-core", branch = "awaiting_on", features = ["tracing_pretty"] }
+restate-sdk-shared-core = { git = "https://github.com/restatedev/sdk-shared-core", branch = "main", features = ["tracing_pretty"] }
 bytes = "1"
 serde = { version = "1", features = ["derive"] }
 serde_bytes = "0.11"