changes

Cargo.toml

@@ -1,5 +1,5 @@
 [package]
-name = "aftermath"
+name = "aftermath-rs"
 version = "0.1.0"
 edition = "2021"
 

src/lib.rs

@@ -56,6 +56,7 @@ impl Drop for Aftermath {
     }
 }
 
+#[derive(Debug, PartialEq, Eq)]
 pub enum Status {
     NotStarted = 0,
     CollectingData,
@@ -73,6 +74,17 @@ impl Status {
             std::mem::transmute(status)
         }
     }
+    pub fn wait_for_status(timeout: Option<std::time::Duration>) -> Self {
+        let mut status = Self::get();
+        let delta = std::time::Duration::from_millis(50);
+        let mut time = std::time::Duration::new(0, 0);
+        while status != Status::CollectingDataFailed && status != Status::Finished && timeout.map_or(true, |t| time < t) {
+            std::thread::sleep(delta);
+            time += delta;
+            status = Self::get();
+        }
+        status
+    }
 }
 
 pub trait AftermathDelegate: Send + Sync + 'static {