[WIP] Awakeable IDs following the new ID scheme

restatedev · Jan 26, 2024 · 331365f · 331365f
1 parent 318a5be
commit 331365f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/sdk-core/src/main/java/dev/restate/sdk/core/Entries.java b/sdk-core/src/main/java/dev/restate/sdk/core/Entries.java
@@ -19,6 +19,7 @@
 import java.util.function.Function;
 
 final class Entries {
+  public static final String AWAKEABLE_IDENTIFIER_PREFIX = "prom_1";
 
   private Entries() {}
 

diff --git a/sdk-core/src/main/java/dev/restate/sdk/core/SyscallsImpl.java b/sdk-core/src/main/java/dev/restate/sdk/core/SyscallsImpl.java
@@ -265,7 +265,8 @@ public void awakeable(SyscallCallback<Map.Entry<String, DeferredResult<ByteStrin
                                         .rewind()));
 
                     return new AbstractMap.SimpleImmutableEntry<>(
-                        Base64.getUrlEncoder().encodeToString(awakeableId.toByteArray()),
+                        Entries.AWAKEABLE_IDENTIFIER_PREFIX
+                            + Base64.getUrlEncoder().encodeToString(awakeableId.toByteArray()),
                         deferredResult);
                   },
                   callback));

diff --git a/sdk-core/src/test/java/dev/restate/sdk/core/AwakeableIdTestSuite.java b/sdk-core/src/test/java/dev/restate/sdk/core/AwakeableIdTestSuite.java
@@ -42,7 +42,8 @@ public Stream<TestDefinition> definitions() {
     expectedAwakeableId.putInt(1);
     expectedAwakeableId.rewind();
     String base64ExpectedAwakeableId =
-        Base64.getUrlEncoder().encodeToString(expectedAwakeableId.array());
+        Entries.AWAKEABLE_IDENTIFIER_PREFIX
+            + Base64.getUrlEncoder().encodeToString(expectedAwakeableId.array());
 
     return Stream.of(
         testInvocation(this::returnAwakeableId, GreeterGrpc.getGreetMethod())