fiber_scheduler 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: fd5761664b48ce8937367c38ae5424c2aad1ffcbac56fac2c002de020f08b6fc
+  data.tar.gz: 426a54cbbd96a1b5eb54794b2a6b295be2af98413cbfb7978780ca57ff8347f7
+SHA512:
+  metadata.gz: 9cdf31791771ecd2fec6fee05a7849954e71a2d6596ea429ec99300a558bd8c129a8e402e7bcae6c1c093765ac9738b4944e70de26a423ae62c440f4c9d10e18
+  data.tar.gz: 4cf8d6bd02b7a51ef03e902da0bb25bbd5fae16ebf82c1b5aa7cf68d84b723b458be1b9106258fff527e17704ffba3850a0b377b735bf21c0638b62c7930d20d

data/lib/fiber/scheduler.rb

@@ -0,0 +1 @@
+require_relative "../fiber_scheduler"

data/lib/fiber_scheduler/trigger.rb

@@ -0,0 +1,40 @@
+class FiberScheduler
+  class Trigger
+    include Comparable
+
+    attr_reader :time
+
+    def initialize(duration, &block)
+      @time = Process.clock_gettime(Process::CLOCK_MONOTONIC) + duration
+      @block = block
+
+      @disabled = nil
+    end
+
+    def <=>(other)
+      raise unless other.is_a?(self.class)
+
+      @time <=> other.time
+    end
+
+    def call
+      @block.call
+    end
+
+    def interval
+      @time - Process.clock_gettime(Process::CLOCK_MONOTONIC)
+    end
+
+    def disable
+      @disabled = true
+    end
+
+    def disabled?
+      @disabled
+    end
+
+    def inspect
+      "#<#{self.class} time=#{@time}>"
+    end
+  end
+end

data/lib/fiber_scheduler/triggers.rb

@@ -0,0 +1,72 @@
+require_relative "trigger"
+
+class FiberScheduler
+  class Triggers
+    def initialize
+      # Array is sorted by Trigger#time
+      @triggers = []
+    end
+
+    def call
+      now = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+
+      while @triggers.any? && @triggers.first.time <= now
+        trigger = @triggers.shift
+        unless trigger.disabled?
+          trigger.call
+        end
+      end
+    end
+
+    def add(duration, &block)
+      trigger = Trigger.new(duration, &block)
+
+      if @triggers.empty?
+        @triggers << trigger
+        return trigger
+      end
+
+      # binary search
+      min = 0
+      max = @triggers.size - 1
+      while min <= max
+        index = (min + max) / 2
+        t = @triggers[index]
+
+        if t > trigger
+          if index.zero? || @triggers[index - 1] <= trigger
+            # found it
+            break
+          else
+            # @triggers[index - 1] > trigger
+            max = index - 1
+          end
+        else
+          # t <= trigger
+          index += 1
+          min = index
+        end
+      end
+
+      @triggers.insert(index, trigger)
+      trigger
+    end
+
+    def interval
+      # Prune disabled triggers
+      while @triggers.first&.disabled?
+        @triggers.shift
+      end
+
+      return if @triggers.empty?
+
+      interval = @triggers.first.interval
+
+      interval >= 0 ? interval : 0
+    end
+
+    def inspect
+      @triggers.inspect
+    end
+  end
+end

data/lib/fiber_scheduler/version.rb

@@ -0,0 +1,3 @@
+class FiberScheduler
+  VERSION = "0.0.1".freeze
+end

data/lib/fiber_scheduler.rb

@@ -0,0 +1,150 @@
+require "io/event"
+require "resolv"
+require_relative "fiber_scheduler/triggers"
+
+module Kernel
+  def FiberScheduler(&block)
+    if Fiber.scheduler.nil?
+      scheduler = FiberScheduler.new
+      Fiber.set_scheduler(scheduler)
+
+      begin
+        yield
+        scheduler.close
+      ensure
+        Fiber.set_scheduler(nil)
+      end
+    else
+      # Fiber.scheduler already set, just schedule a task.
+      Fiber.schedule(&block)
+    end
+  end
+end
+
+class FiberScheduler
+  TimeoutError = Class.new(RuntimeError)
+  IOWaitTimeout = Class.new(TimeoutError)
+
+  def initialize
+    @selector = IO::Event::Selector.new(Fiber.current)
+    @triggers = Triggers.new
+
+    @count = 0
+    @nested = []
+  end
+
+  def run
+    while @count > 0
+      if @nested.empty?
+        @selector.select(@triggers.interval)
+        @triggers.call
+      else
+        while @nested.any?
+          fiber = @nested.pop
+          fiber.transfer
+        end
+      end
+    end
+  end
+
+  # Fiber::SchedulerInterface methods below
+
+  def close
+    return unless @selector
+
+    begin
+      run
+    ensure
+      @selector.close
+      @selector = nil
+    end
+  end
+
+  def block(blocker, timeout)
+    return @selector.transfer unless timeout
+
+    fiber = Fiber.current
+    trigger = @triggers.add(timeout) do
+      fiber.transfer if fiber.alive?
+    end
+
+    begin
+      @selector.transfer
+    ensure
+      trigger.disable
+    end
+  end
+
+  def unblock(blocker, fiber)
+    @selector.push(fiber)
+  end
+
+  def kernel_sleep(duration = nil)
+    return @selector.transfer unless duration
+
+    block(:sleep, duration)
+  end
+
+  def address_resolve(hostname)
+    Resolv.getaddresses(hostname)
+  end
+
+  def io_wait(io, events, timeout = nil)
+    fiber = Fiber.current
+    return @selector.io_wait(fiber, io, events) unless timeout
+
+    trigger = @triggers.raise_in(timeout, IOWaitTimeout)
+    # trigger = @triggers.add(timeout) do
+    #   fiber.raise(IOWaitTimeout) if fiber.alive?
+    # end
+
+    begin
+      @selector.io_wait(fiber, io, events)
+    rescue IOWaitTimeout
+      false
+    ensure
+      trigger.disable
+    end
+  end
+
+  def io_read(io, buffer, length)
+    @selector.io_read(Fiber.current, io, buffer, length)
+  end
+
+  def io_write(io, buffer, length)
+    @selector.io_write(Fiber.current, io, buffer, length)
+  end
+
+  def process_wait(pid, flags)
+    @selector.process_wait(Fiber.current, pid, flags)
+  end
+
+  def timeout_after(duration, exception = TimeoutError, message = "timeout")
+    fiber = Fiber.current
+    trigger = @triggers.add(duration) do
+      fiber.raise(exception, message) if fiber.alive?
+    end
+
+    begin
+      yield duration
+    ensure
+      trigger.disable
+    end
+  end
+
+  def fiber(&block)
+    unless Fiber.blocking?
+      # nested Fiber.schedule
+      @nested << Fiber.current
+    end
+
+    fiber = Fiber.new(blocking: false) do
+      @count += 1
+      block.call
+    ensure
+      @count -= 1
+    end
+
+    fiber.tap(&:transfer)
+  end
+end

metadata

@@ -0,0 +1,117 @@
+--- !ruby/object:Gem::Specification
+name: fiber_scheduler
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Bruno Sutic
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2022-02-11 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: io-event
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: async
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.11'
+- !ruby/object:Gem::Dependency
+  name: rubocop-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+- !ruby/object:Gem::Dependency
+  name: standard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+description:
+email: code@brunosutic.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/fiber/scheduler.rb
+- lib/fiber_scheduler.rb
+- lib/fiber_scheduler/trigger.rb
+- lib/fiber_scheduler/triggers.rb
+- lib/fiber_scheduler/version.rb
+homepage: https://github.com/bruno-/fiber_scheduler
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.1.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.3.3
+signing_key:
+specification_version: 4
+summary: Fiber scheduler
+test_files: []