Merge pull request #42 from drifting-in-space/paulgb/version-updates

Fix timer
jamsocket · May 5, 2024 · 1f33689 · 1f33689
2 parents 09b3b5c + edbaeb1
commit 1f33689
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/aper-stateroom/src/lib.rs b/aper-stateroom/src/lib.rs
@@ -56,9 +56,8 @@ impl<P: StateProgram + Default> AperStateroomService<P> {
         }
 
         if let Some(ev) = &susp {
-            if let Ok(dur) = ev.timestamp.signed_duration_since(Utc::now()).to_std() {
-                ctx.set_timer(dur.as_millis() as u32);
-            }
+            let dur = ev.timestamp.signed_duration_since(Utc::now());
+            ctx.set_timer(dur.num_milliseconds().max(0) as u32);
         }
 
         self.suspended_event = susp;
@@ -125,6 +124,10 @@ impl<P: StateProgram + Default> StateroomService for AperStateroomService<P>
 where
     P::T: Unpin + Send + Sync + 'static,
 {
+    fn init(&mut self, ctx: &impl StateroomContext) {
+        self.update_suspended_event(ctx);
+    }
+
     fn connect(&mut self, client_id: ClientId, ctx: &impl StateroomContext) {
         let response = StateProgramMessage::InitialState {
             timestamp: Utc::now(),