commit 49c864c484c96ce222a289a1c1d96c64cb0f7fbc
author Adam Snaider <adam.snaider@bluerivertech.com> Mon Oct 09 18:32:48 2023 -0700
committer James Kuszmaul <jabukuszmaul+collab@gmail.com> Mon Feb 19 13:49:36 2024 -0800
tree df53dd2293b279b02c5baa61d97200a6dce4b081
parent 4769bb4dde68673fcddb77bf471b51f61bcf4dd7

Add run_for in Rust SimulatedEventLoop

Change-Id: Icf23a9c834f11ff78dbcde0b0ff3c31519767c92
Signed-off-by: James Kuszmaul <james.kuszmaul@bluerivertech.com>

aos/events/simulated_event_loop.rs

diff --git a/aos/events/simulated_event_loop.rs b/aos/events/simulated_event_loop.rs
index 90c4c86..0677efb 100644
--- a/aos/events/simulated_event_loop.rs
+++ b/aos/events/simulated_event_loop.rs
@@ -1,4 +1,4 @@
-use std::{marker::PhantomData, pin::Pin, ptr};
+use std::{marker::PhantomData, pin::Pin, ptr, time::Duration};
 
 use autocxx::WithinBox;
 use cxx::UniquePtr;
@@ -107,6 +107,15 @@
         self.as_mut().Run();
     }
 
+    pub fn run_for(&mut self, duration: Duration) {
+        self.as_mut().RunFor(
+            duration
+                .as_nanos()
+                .try_into()
+                .expect("Out of range: Internal clock uses 64 bits"),
+        );
+    }
+
     // TODO(Brian): Expose OnStartup. Just take a callback for creating things, and rely on
     // dropping the created objects instead of OnShutdown.
     // pub fn spawn_on_startup(&mut self, spawner: impl FnMut());

aos/events/simulated_event_loop_for_rust.h

diff --git a/aos/events/simulated_event_loop_for_rust.h b/aos/events/simulated_event_loop_for_rust.h
index a8d630c..663d163 100644
--- a/aos/events/simulated_event_loop_for_rust.h
+++ b/aos/events/simulated_event_loop_for_rust.h
@@ -17,6 +17,9 @@
   }
 
   void Run() { factory_.Run(); }
+  void RunFor(int64_t nanos) {
+    factory_.RunFor(std::chrono::nanoseconds(nanos));
+  }
 
   std::unique_ptr<ExitHandle> MakeExitHandle() {
     return factory_.MakeExitHandle();