itsi-scheduler 0.1.5 → 0.2.2

data/ext/itsi_scheduler/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "itsi-scheduler"
-version = "0.1.0"
+version = "0.2.2"
 edition = "2021"
 authors = ["Wouter Coppieters <wc@pico.net.nz>"]
 license = "MIT"

data/ext/itsi_scheduler/src/itsi_scheduler.rs

@@ -65,7 +65,7 @@ impl ItsiScheduler {
     pub fn wake(&self) -> MagnusResult<()> {
         self.waker.lock().wake().map_err(|_| {
             magnus::Error::new(
-                magnus::exception::exception(),
+                magnus::exception::standard_error(),
                 "Failed to wake the scheduler",
             )
         })?;
@@ -115,6 +115,11 @@ impl ItsiScheduler {
             self.timers.lock().push(timer_entry);
         }
     }
+
+    pub fn clear_timer(&self, token: usize) {
+        self.timers.lock().retain(|timer| timer.token.0 != token);
+    }
+
     pub fn has_pending_io(&self) -> bool {
         !self.timers.lock().is_empty() || !self.io_waiters.lock().is_empty()
     }
@@ -148,8 +153,9 @@ impl ItsiScheduler {
                 let mut events = self.events.lock();
                 {
                     let mut poll = self.poll.lock();
-                    poll.poll(&mut events, timeout)
-                        .map_err(|e| ItsiError::ArgumentError(format!("poll error: {}", e)))?;
+                    if let Err(_err) = poll.poll(&mut events, timeout) {
+                        return Ok(due_fibers);
+                    }
                 };
 
                 for event in events.iter() {

data/ext/itsi_scheduler/src/lib.rs

@@ -19,6 +19,7 @@ fn init(ruby: &Ruby) -> Result<(), Error> {
     scheduler.define_method("debug", method!(ItsiScheduler::debug, 1))?;
     scheduler.define_method("warn", method!(ItsiScheduler::warn, 1))?;
     scheduler.define_method("start_timer", method!(ItsiScheduler::start_timer, 2))?;
+    scheduler.define_method("clear_timer", method!(ItsiScheduler::clear_timer, 1))?;
     scheduler.define_method(
         "address_resolve",
         method!(ItsiScheduler::address_resolve, 1),