autocommit 2022-02-28 00:14:11 CET

src/utils/run_toplevel.rs

@@ -0,0 +1,38 @@
+use crate::async_engine::SpawnedFuture;
+use crate::{AsyncEngine, AsyncQueue};
+use std::rc::Rc;
+
+pub struct RunToplevelFuture {
+    _future: SpawnedFuture<()>,
+}
+
+pub struct RunToplevel {
+    queue: AsyncQueue<Box<dyn FnOnce()>>,
+}
+
+impl RunToplevel {
+    pub fn install(eng: &Rc<AsyncEngine>) -> (RunToplevelFuture, Rc<RunToplevel>) {
+        let slf = Rc::new(RunToplevel {
+            queue: Default::default(),
+        });
+        let future = eng.spawn({
+            let slf = slf.clone();
+            async move {
+                loop {
+                    let f = slf.queue.pop().await;
+                    f();
+                }
+            }
+        });
+        let future = RunToplevelFuture { _future: future };
+        (future, slf)
+    }
+
+    pub fn schedule<F: FnOnce() + 'static>(&self, f: F) {
+        self.schedule_dyn(Box::new(f));
+    }
+
+    fn schedule_dyn(&self, f: Box<dyn FnOnce()>) {
+        self.queue.push(f);
+    }
+}