itsi-scheduler 0.2.7 → 0.2.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d1bc3d31e647642d096846f41852991ea7f31c3b2c1e70c4ed07ec7c0676e48e
-  data.tar.gz: 1d335eb5d0fbe3cf02031cedb096db4b8bbf8cdd49db899b6e2db4efe30d2830
+  metadata.gz: a909654e8cfbe862c3bae6a211a7c49f2852ba76870d1aa1e7de91122fdbc6cc
+  data.tar.gz: 6b2d6d281008d2ee680553d9dbde3b32911d713cc438bfefc5119e642cb16dfc
 SHA512:
-  metadata.gz: 9e9a99c1ce6e278159dd31a525810ddf2661563f823dd54bc4c6e5686fcb9b70e7137d3a9622aa1f794553f2e26c43b5c1d37bda2b549aa0306b6c24c410083f
-  data.tar.gz: 30d3ac6eefb849cdf6d53db6dfe420af9586be2b1d7efa2ce5605e29bd5bf686683033e48ebe835d41cabd29abfdce953131f8ee640e7a8ecb902ac235123248
+  metadata.gz: 5b73a90da6b1f94f14704b7a9bf195e6eb2a5d6e1ec7638c7440610146bf738cd4999abf84e38acff4db4d32b4cfb0bb97d1e86fa3447f6ca1565c37301de59c
+  data.tar.gz: 4572f871c428b4387cd9150f3f825e3495f634a694bb5791ec8a58afcd7bd0cd47f996cf7d0efe54d2ceb869f04c586a140fc1f85865fd065f6d1965b02cb3d5

data/Cargo.lock

@@ -213,7 +213,7 @@ checksum = "4a5f13b858c8d314ee3e8f639011f7ccefe71f97f96e50151fb991f267928e2c"
 
 [[package]]
 name = "itsi-scheduler"
-version = "0.2.7"
+version = "0.2.8"
 dependencies = [
  "bytes",
  "derive_more",

data/ext/itsi_scheduler/Cargo.toml

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

data/ext/itsi_scheduler/src/itsi_scheduler.rs

@@ -7,7 +7,7 @@ use itsi_error::ItsiError;
 use itsi_rb_helpers::{call_without_gvl, create_ruby_thread};
 use magnus::{
     error::Result as MagnusResult,
-    value::{InnerValue, Opaque, ReprValue},
+    value::{InnerValue, Lazy, LazyId, Opaque, ReprValue},
     Module, RClass, Ruby, Value,
 };
 use mio::{Events, Poll, Token, Waker};
@@ -19,7 +19,7 @@ use std::{
     time::Duration,
 };
 use timer::Timer;
-use tracing::{debug, info, warn};
+use tracing::{debug, error, info, warn};
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
 pub(crate) struct Readiness(i16);
@@ -31,6 +31,9 @@ impl std::fmt::Debug for ItsiScheduler {
 }
 
 const WAKE_TOKEN: Token = Token(0);
+static ID_CURRENT: LazyId = LazyId::new("current");
+static CLASS_FIBER: Lazy<RClass> =
+    Lazy::new(|ruby| ruby.module_kernel().const_get("Fiber").unwrap());
 
 #[magnus::wrap(class = "Itsi::Scheduler", free_immediately, size)]
 pub(crate) struct ItsiScheduler {
@@ -225,14 +228,17 @@ impl ItsiScheduler {
         let result: Arc<RwLock<Option<T>>> = Arc::new(RwLock::new(None));
         let result_clone = Arc::clone(&result);
 
-        let current_fiber = Opaque::from(ruby.fiber_current());
-        let scheduler = Opaque::from(
-            ruby.module_kernel()
-                .const_get::<_, RClass>("Fiber")
-                .unwrap()
-                .funcall::<_, _, Value>("scheduler", ())
-                .unwrap(),
-        );
+        let class_fiber = ruby.get_inner(&CLASS_FIBER);
+        let current_fiber = ruby
+            .get_inner(&CLASS_FIBER)
+            .funcall::<_, _, Value>(*ID_CURRENT, ());
+
+        if current_fiber.is_err() {
+            error!("Failed to get current fiber");
+            return Err(ItsiError::ArgumentError("Failed to get current fiber".to_string()).into());
+        }
+        let current_fiber = Opaque::from(current_fiber.unwrap());
+        let scheduler = Opaque::from(class_fiber.funcall::<_, _, Value>("scheduler", ()).unwrap());
 
         create_ruby_thread(move || {
             call_without_gvl(|| {

data/ext/itsi_server/Cargo.toml

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

data/lib/itsi/scheduler/version.rb

@@ -2,6 +2,6 @@
 
 module Itsi
   class Scheduler
-    VERSION = "0.2.7"
+    VERSION = "0.2.8"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: itsi-scheduler
 version: !ruby/object:Gem::Version
-  version: 0.2.7
+  version: 0.2.8
 platform: ruby
 authors:
 - Wouter Coppieters
@@ -197,7 +197,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.0.0
+      version: '3.1'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="