Make Handle trait public, add EventLoop.schedule_handle method

Author: gi0baro

src/event_loop.rs

@@ -43,7 +43,7 @@ struct TCPListenerHandleData {
     server: TCPServerRef,
 }
 
-pub(crate) struct EventLoopRunState {
+pub struct EventLoopRunState {
     buf: Box<[u8]>,
     events: event::Events,
     pub read_buf: Box<[u8]>,
@@ -631,6 +631,41 @@ impl EventLoop {
 
         Ok(())
     }
+
+    #[allow(clippy::missing_errors_doc)]
+    pub fn schedule_handle(&self, handle: impl Handle + Send + 'static, delay: Option<Duration>) -> Result<()> {
+        match delay {
+            Some(delay) => {
+                let when = (Instant::now().duration_since(self.epoch) + delay).as_micros();
+                let timer = Timer {
+                    handle: Box::new(handle),
+                    when,
+                };
+                {
+                    let mut guard = self
+                        .handles_sched
+                        .lock()
+                        .map_err(|_| anyhow::anyhow!("lock acquisition failed"))?;
+                    guard.push(timer);
+                }
+            }
+            None => {
+                {
+                    let mut guard = self
+                        .handles_ready
+                        .lock()
+                        .map_err(|_| anyhow::anyhow!("lock acquisition failed"))?;
+                    guard.push_back(Box::new(handle));
+                }
+                self.counter_ready.fetch_add(1, atomic::Ordering::Release);
+            }
+        }
+        if self.idle.load(atomic::Ordering::Acquire) {
+            self.wake();
+        }
+
+        Ok(())
+    }
 }
 
 #[pymethods]

src/handles.rs

@@ -10,7 +10,7 @@ use crate::{
 #[cfg(not(PyPy))]
 use crate::py::run_in_ctx1;
 
-pub(crate) trait Handle {
+pub trait Handle {
     fn run(&self, py: Python, event_loop: &EventLoop, state: &mut EventLoopRunState);
     fn cancelled(&self) -> bool {
         false

src/lib.rs

@@ -2,7 +2,7 @@ use pyo3::prelude::*;
 use std::sync::OnceLock;
 
 pub mod event_loop;
-mod handles;
+pub mod handles;
 mod io;
 mod log;
 mod py;