From ffa328276e9395c6bbdda72afe66636a525e0f66 Mon Sep 17 00:00:00 2001
From: Ahmed Farghal <me@asoli.dev>
Date: Fri, 26 Jan 2024 12:19:33 +0000
Subject: [PATCH] [WIP] Awakeable IDs following the new ID scheme

---
 sdk-core/src/main/java/dev/restate/sdk/core/Entries.java       | 1 +
 sdk-core/src/main/java/dev/restate/sdk/core/SyscallsImpl.java  | 3 ++-
 .../test/java/dev/restate/sdk/core/AwakeableIdTestSuite.java   | 3 ++-
 3 files changed, 5 insertions(+), 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
index 6ca35bf4..b64b16be 100644
--- 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 {
+  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
index ab543d50..fcb9a027 100644
--- 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
index a813ce6c..b1dd8726 100644
--- 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())