itsi-scheduler 0.1.0 → 0.1.2

data/lib/itsi/scheduler.rb

@@ -6,6 +6,142 @@ require_relative "scheduler/itsi_scheduler"
 module Itsi
   class Scheduler
     class Error < StandardError; end
-    # Your code goes here...
+
+    def self.resume_token
+      @resume_token ||= 0
+      @resume_token += 1
+    end
+
+    def initialize
+      @join_waiters = {}.compare_by_identity
+      @token_map = {}.compare_by_identity
+      @resume_tokens = {}.compare_by_identity
+      @unblocked = [[], []]
+      @unblock_idx = 0
+      @unblocked_mux = Mutex.new
+      @resume_fiber = method(:resume_fiber).to_proc
+      @resume_fiber_with_readiness = method(:resume_fiber_with_readiness).to_proc
+      @resume_blocked = method(:resume_blocked).to_proc
+    end
+
+    def block(_, timeout, fiber = Fiber.current, token = Scheduler.resume_token)
+      @join_waiters[fiber] = true
+
+      start_timer(timeout, token) if timeout
+      @resume_tokens[token] = fiber
+      @token_map[fiber] = token
+      Fiber.yield
+    ensure
+      @resume_tokens.delete(token)
+      @token_map.delete(fiber)
+      @join_waiters.delete(fiber)
+    end
+
+    # Register an IO waiter.
+    # This will get resumed by our scheduler inside the call to
+    # fetch_events.
+    def io_wait(io, events, duration)
+      fiber = Fiber.current
+      token = Scheduler.resume_token
+      readiness = register_io_wait(io.fileno, events, duration, token)
+      readiness || block(nil, duration, fiber, token)
+    end
+
+    def unblock(_blocker, fiber)
+      @unblocked_mux.synchronize do
+        @unblocked[@unblock_idx] << fiber
+      end
+      wake
+    end
+
+    def kernel_sleep(duration)
+      block nil, duration
+    end
+
+    def tick
+      events = fetch_due_events
+      timers = fetch_due_timers
+      unblocked = switch_unblock_batch
+      events&.each(&@resume_fiber_with_readiness)
+      unblocked.each(&@resume_blocked)
+      unblocked.clear
+      timers&.each(&@resume_fiber)
+    end
+
+    def resume_fiber(token)
+      if (fiber = @resume_tokens.delete(token))
+        fiber.resume
+      end
+    rescue StandardError => e
+      warn "Failed to resume fiber #{fiber}: #{e.message}"
+    end
+
+    def resume_fiber_with_readiness((token, readiness))
+      if (fiber = @resume_tokens.delete(token))
+        fiber.resume(readiness)
+      end
+    rescue StandardError => e
+      warn "Failed to resume fiber #{fiber}: #{e.message}"
+    end
+
+    def resume_blocked(fiber)
+      if (token = @token_map[fiber])
+        resume_fiber(token)
+      elsif fiber.alive?
+        fiber.resume
+      end
+    end
+
+    def switch_unblock_batch
+      @unblocked_mux.synchronize do
+        current = @unblocked[@unblock_idx]
+        @unblock_idx = (@unblock_idx + 1) % 2
+        current
+      end
+    end
+
+    # Yields upwards to the scheduler, with an intention to
+    # resume the fiber that yielded ASAP.
+    def yield
+      kernel_sleep(0) if work?
+    end
+
+    # Keep running until we've got no timers we're awaiting, no pending IO, no temporary yields,
+    # no pending unblocks.
+    def work?
+      !@unblocked[@unblock_idx].empty? || !@join_waiters.empty? || has_pending_io?
+    end
+
+    # Run until no more work needs doing.
+    def run
+      tick while work?
+      debug "Exit Scheduler"
+    end
+
+    # Hook invoked at the end of the thread.
+    # Will start our scheduler's Reactor.
+    def scheduler_close
+      run
+    ensure
+      @closed ||= true
+      freeze
+    end
+
+    # Need to defer to Process::Status rather than our extension
+    # as we don't have a means of creating our own Process::Status.
+    def process_wait(pid, flags)
+      Thread.new do
+        Process::Status.wait(pid, flags)
+      end.value
+    end
+
+    def closed?
+      @closed
+    end
+
+    # Spin up a new fiber and immediately resume it.
+    def fiber(&blk)
+      Fiber.new(blocking: false, &blk).tap(&:resume)
+    end
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: itsi-scheduler
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Wouter Coppieters
 bindir: exe
 cert_chain: []
-date: 2025-03-01 00:00:00.000000000 Z
+date: 2025-03-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: libclang
@@ -56,13 +56,47 @@ files:
 - Rakefile
 - ext/itsi_error/Cargo.lock
 - ext/itsi_error/Cargo.toml
+- ext/itsi_error/src/from.rs
 - ext/itsi_error/src/lib.rs
+- ext/itsi_instrument_entry/Cargo.toml
+- ext/itsi_instrument_entry/src/lib.rs
 - ext/itsi_rb_helpers/Cargo.lock
 - ext/itsi_rb_helpers/Cargo.toml
+- ext/itsi_rb_helpers/src/heap_value.rs
 - ext/itsi_rb_helpers/src/lib.rs
 - ext/itsi_scheduler/Cargo.toml
 - ext/itsi_scheduler/extconf.rb
+- ext/itsi_scheduler/src/itsi_scheduler.rs
+- ext/itsi_scheduler/src/itsi_scheduler/io_helpers.rs
+- ext/itsi_scheduler/src/itsi_scheduler/io_waiter.rs
+- ext/itsi_scheduler/src/itsi_scheduler/timer.rs
 - ext/itsi_scheduler/src/lib.rs
+- ext/itsi_server/Cargo.toml
+- ext/itsi_server/extconf.rb
+- ext/itsi_server/src/body_proxy/big_bytes.rs
+- ext/itsi_server/src/body_proxy/itsi_body_proxy.rs
+- ext/itsi_server/src/body_proxy/mod.rs
+- ext/itsi_server/src/lib.rs
+- ext/itsi_server/src/request/itsi_request.rs
+- ext/itsi_server/src/request/mod.rs
+- ext/itsi_server/src/response/itsi_response.rs
+- ext/itsi_server/src/response/mod.rs
+- ext/itsi_server/src/server/bind.rs
+- ext/itsi_server/src/server/bind_protocol.rs
+- ext/itsi_server/src/server/io_stream.rs
+- ext/itsi_server/src/server/itsi_ca/itsi_ca.crt
+- ext/itsi_server/src/server/itsi_ca/itsi_ca.key
+- ext/itsi_server/src/server/itsi_server.rs
+- ext/itsi_server/src/server/lifecycle_event.rs
+- ext/itsi_server/src/server/listener.rs
+- ext/itsi_server/src/server/mod.rs
+- ext/itsi_server/src/server/process_worker.rs
+- ext/itsi_server/src/server/serve_strategy/cluster_mode.rs
+- ext/itsi_server/src/server/serve_strategy/mod.rs
+- ext/itsi_server/src/server/serve_strategy/single_mode.rs
+- ext/itsi_server/src/server/signal.rs
+- ext/itsi_server/src/server/thread_worker.rs
+- ext/itsi_server/src/server/tls.rs
 - ext/itsi_tracing/Cargo.lock
 - ext/itsi_tracing/Cargo.toml
 - ext/itsi_tracing/src/lib.rs
@@ -83,12 +117,12 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.1.0
+      version: 3.0.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.3.11
+      version: 3.1.11
 requirements: []
 rubygems_version: 3.6.2
 specification_version: 4