Add name parameter to sleep (#136)

Author: slinkydeveloper

python/restate/context.py

@@ -404,7 +404,7 @@ def run_typed(self,
         """
 
     @abc.abstractmethod
-    def sleep(self, delta: timedelta) -> RestateDurableSleepFuture:
+    def sleep(self, delta: timedelta, name: Optional[str] = None) -> RestateDurableSleepFuture:
         """
         Suspends the current invocation for the given duration
         """

python/restate/server_context.py

@@ -626,10 +626,10 @@ def run_typed(
         self.run_coros_to_execute[handle] = lambda : self.create_run_coroutine(handle, func, options.serde, options.max_attempts, options.max_retry_duration)
         return self.create_future(handle, options.serde)
 
-    def sleep(self, delta: timedelta) -> RestateDurableSleepFuture:
+    def sleep(self, delta: timedelta, name: Optional[str] = None) -> RestateDurableSleepFuture:
         # convert timedelta to milliseconds
         millis = int(delta.total_seconds() * 1000)
-        handle = self.vm.sys_sleep(millis)
+        handle = self.vm.sys_sleep(millis, name)
         update_restate_context_is_replaying(self.vm)
         return self.create_sleep_future(handle) # type: ignore
 

python/restate/vm.py

@@ -293,9 +293,9 @@ def sys_clear_all_state(self):
         """Clear the state associated with the given name."""
         self.vm.sys_clear_all_state()
 
-    def sys_sleep(self, millis: int):
+    def sys_sleep(self, millis: int, name: typing.Optional[str] = None):
         """Ask to sleep for a given duration"""
-        return self.vm.sys_sleep(millis)
+        return self.vm.sys_sleep(millis, name)
 
     def sys_call(self,
                  service: str,

src/lib.rs

@@ -439,14 +439,15 @@ impl PyVM {
     fn sys_sleep(
         mut self_: PyRefMut<'_, Self>,
         millis: u64,
+        name: Option<String>
     ) -> Result<PyNotificationHandle, PyVMError> {
         let now = SystemTime::now()
             .duration_since(SystemTime::UNIX_EPOCH)
             .expect("Duration since unix epoch cannot fail");
         self_
             .vm
             .sys_sleep(
-                String::default(),
+                name.unwrap_or_default(),
                 now + Duration::from_millis(millis),
                 Some(now),
             )