More informative logs (#24)

restatedev · Dec 13, 2024 · 0a45d53 · 0a45d53
1 parent 3c342c4
commit 0a45d53
Show file tree

Hide file tree

Showing 6 changed files with 137 additions and 40 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -313,8 +313,12 @@ pub trait VM: Sized {
 
     fn sys_state_clear_all(&mut self) -> VMResult<()>;
 
-    fn sys_sleep(&mut self, wake_up_time_since_unix_epoch: Duration)
-        -> VMResult<AsyncResultHandle>;
+    /// Note: `now_since_unix_epoch` is only used for debugging purposes
+    fn sys_sleep(
+        &mut self,
+        wake_up_time_since_unix_epoch: Duration,
+        now_since_unix_epoch: Option<Duration>,
+    ) -> VMResult<AsyncResultHandle>;
 
     fn sys_call(&mut self, target: Target, input: Bytes) -> VMResult<AsyncResultHandle>;
 

diff --git a/src/tests/sleep.rs b/src/tests/sleep.rs
@@ -22,7 +22,7 @@ fn sleep_suspends() {
         .run(|vm| {
             vm.sys_input().unwrap();
 
-            let h1 = vm.sys_sleep(Duration::from_secs(1)).unwrap();
+            let h1 = vm.sys_sleep(Duration::from_secs(1), None).unwrap();
             vm.notify_await_point(h1);
             let h1_result = vm.take_async_result(h1);
             if let Err(SuspendedOrVMError::Suspended(_)) = &h1_result {
@@ -67,7 +67,7 @@ fn sleep_completed() {
         .run(|vm| {
             vm.sys_input().unwrap();
 
-            let h1 = vm.sys_sleep(Duration::from_secs(1)).unwrap();
+            let h1 = vm.sys_sleep(Duration::from_secs(1), None).unwrap();
             vm.notify_await_point(h1);
             let h1_result = vm.take_async_result(h1);
             if let Err(SuspendedOrVMError::Suspended(_)) = &h1_result {
@@ -115,7 +115,7 @@ fn sleep_still_sleeping() {
         .run(|vm| {
             vm.sys_input().unwrap();
 
-            let h1 = vm.sys_sleep(Duration::from_secs(1)).unwrap();
+            let h1 = vm.sys_sleep(Duration::from_secs(1), None).unwrap();
             vm.notify_await_point(h1);
             let h1_result = vm.take_async_result(h1);
             if let Err(SuspendedOrVMError::Suspended(_)) = &h1_result {

diff --git a/src/vm/context.rs b/src/vm/context.rs
@@ -213,6 +213,13 @@ impl RunState {
     pub(crate) fn is_running(&self) -> bool {
         matches!(self, RunState::Running(_))
     }
+
+    pub(crate) fn name(&self) -> Option<&str> {
+        match self {
+            RunState::Running(n) => Some(n),
+            RunState::NotRunning => None,
+        }
+    }
 }
 
 pub(crate) enum EagerGetState {