polyphony 0.38 → 0.43

data/lib/polyphony/core/channel.rb

@@ -26,7 +26,7 @@ module Polyphony
     end
 
     def receive
-      Gyro.ref
+      Thread.current.agent.ref
       if @payload_queue.empty?
         @waiting_queue << Fiber.current
         suspend
@@ -34,7 +34,7 @@ module Polyphony
         receive_from_queue
       end
     ensure
-      Gyro.unref
+      Thread.current.agent.unref
     end
 
     def receive_from_queue
@@ -43,4 +43,4 @@ module Polyphony
       payload
     end
   end
-end
+end

data/lib/polyphony/core/exceptions.rb

@@ -33,4 +33,4 @@ module Polyphony
 
   # Restart is used to restart a fiber
   class Restart < BaseException; end
-end
+end

data/lib/polyphony/core/global_api.rb

@@ -45,13 +45,16 @@ module Polyphony
     end
 
     def every(interval)
-      timer = Gyro::Timer.new(interval, interval)
+      next_time = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC) + interval
       loop do
-        timer.await
+        now = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC)
+        Thread.current.agent.sleep(next_time - now)
         yield
+        loop do
+          next_time += interval
+          break if next_time > now
+        end
       end
-    ensure
-      timer.stop
     end
 
     def move_on_after(interval, with_value: nil, &block)
@@ -93,28 +96,27 @@ module Polyphony
     def sleep(duration = nil)
       return sleep_forever unless duration
 
-      timer = Gyro::Timer.new(duration, 0)
-      timer.await
+      Thread.current.agent.sleep duration
     end
 
     def sleep_forever
-      Thread.current.fiber_ref
+      Thread.current.agent.ref
       suspend
     ensure
-      Thread.current.fiber_unref
+      Thread.current.agent.unref
     end
 
     def throttled_loop(rate, count: nil, &block)
       throttler = Polyphony::Throttler.new(rate)
       if count
-        count.times { throttler.(&block) }
+        count.times { |_i| throttler.(&block) }
       else
         loop { throttler.(&block) }
       end
     ensure
-      throttler.stop
+      throttler&.stop
     end
   end
 end
 
-Object.include Polyphony::GlobalAPI
+Object.include Polyphony::GlobalAPI

data/lib/polyphony/core/resource_pool.rb

@@ -23,13 +23,13 @@ module Polyphony
     end
 
     def acquire
-      Gyro.ref
+      Thread.current.agent.ref
       resource = wait_for_resource
       return unless resource
 
       yield resource
     ensure
-      Gyro.unref
+      Thread.current.agent.unref
       release(resource) if resource
     end
 
@@ -104,4 +104,4 @@ module Polyphony
       (@limit - @size).times { @stock << allocate }
     end
   end
-end
+end

data/lib/polyphony/core/sync.rb

@@ -4,7 +4,7 @@ module Polyphony
   # Implements mutex lock for synchronizing access to a shared resource
   class Mutex
     def initialize
-      @waiting_fibers = Gyro::Queue.new
+      @waiting_fibers = Polyphony::Queue.new
     end
 
     def synchronize
@@ -18,4 +18,4 @@ module Polyphony
       snooze
     end
   end
-end
+end

data/lib/polyphony/core/thread_pool.rb

@@ -8,29 +8,29 @@ module Polyphony
     attr_reader :size
 
     def self.process(&block)
-      @default_pool ||= self.new
+      @default_pool ||= new
       @default_pool.process(&block)
     end
 
     def self.reset
       return unless @default_pool
-      
+
       @default_pool.stop
       @default_pool = nil
     end
 
     def initialize(size = Etc.nprocessors)
       @size = size
-      @task_queue = Gyro::Queue.new
+      @task_queue = Polyphony::Queue.new
       @threads = (1..@size).map { Thread.new { thread_loop } }
     end
 
     def process(&block)
       setup unless @task_queue
 
-      async = Fiber.current.auto_async
-      @task_queue << [block, async]
-      async.await
+      watcher = Fiber.current.auto_watcher
+      @task_queue << [block, watcher]
+      watcher.await
     end
 
     def cast(&block)

data/lib/polyphony/core/throttler.rb

@@ -6,17 +6,24 @@ module Polyphony
     def initialize(rate)
       @rate = rate_from_argument(rate)
       @min_dt = 1.0 / @rate
+      @next_time = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC)
     end
 
-    def call(&block)
-      @timer ||= Gyro::Timer.new(0, @min_dt)
-      @timer.await
-      block.call(self)
+    def call
+      now = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC)
+      delta = @next_time - now
+      Thread.current.agent.sleep(delta) if delta > 0
+      yield self
+
+      loop do
+        @next_time += @min_dt
+        break if @next_time > now
+      end
     end
     alias_method :process, :call
 
     def stop
-      @timer&.stop
+      @stop = true
     end
 
     private
@@ -31,4 +38,4 @@ module Polyphony
       raise "Invalid rate argument #{arg.inspect}"
     end
   end
-end
+end

data/lib/polyphony/event.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Polyphony
+  # Event watcher for thread-safe synchronisation
+  class Event
+    def initialize
+      @i, @o = IO.pipe
+    end
+
+    def await
+      Thread.current.agent.read(@i, +'', 8192, false)
+      raise @value if @value.is_a?(Exception)
+
+      @value
+    end
+
+    def await_no_raise
+      Thread.current.agent.read(@i, +'', 8192, false)
+      @value
+    end
+
+    def signal(value = nil)
+      @value = value
+      Thread.current.agent.write(@o, '1')
+    end
+  end
+end

data/lib/polyphony/extensions/core.rb

@@ -8,8 +8,6 @@ require_relative '../core/exceptions'
 
 # Exeption overrides
 class ::Exception
-  EXIT_EXCEPTION_CLASSES = [::Interrupt, ::SystemExit].freeze
-
   class << self
     attr_accessor :__disable_sanitized_backtrace__
   end
@@ -24,7 +22,7 @@ class ::Exception
 
   alias_method :orig_backtrace, :backtrace
   def backtrace
-    unless @first_backtrace_call || EXIT_EXCEPTION_CLASSES.include?(self.class)
+    unless @first_backtrace_call
       @first_backtrace_call = true
       return orig_backtrace
     end
@@ -52,10 +50,13 @@ end
 
 # Overrides for Process
 module ::Process
-  def self.detach(pid)
-    fiber = spin { Gyro::Child.new(pid).await }
-    fiber.define_singleton_method(:pid) { pid }
-    fiber
+  class << self
+    alias_method :orig_detach, :detach
+    def detach(pid)
+      fiber = spin { Thread.current.agent.waitpid(pid) }
+      fiber.define_singleton_method(:pid) { pid }
+      fiber
+    end
   end
 end
 
@@ -67,10 +68,9 @@ module ::Kernel
   def `(cmd)
     Open3.popen3(cmd) do |i, o, e, _t|
       i.close
-      while (l = e.readpartial(8192))
-        $stderr << l
-      end
-      o.read
+      err = e.read
+      $stderr << err if err
+      o.read || ''
     end
   end
 
@@ -91,6 +91,7 @@ module ::Kernel
   def gets(*_args)
     if !ARGV.empty? || @gets_fiber
       @gets_fiber ||= Fiber.new(&ARGV_GETS_LOOP)
+      @gets_fiber.thread = Thread.current
       result = @gets_fiber.alive? && @gets_fiber.safe_transfer(Fiber.current)
       return result if result
 
@@ -104,14 +105,21 @@ module ::Kernel
   def system(*args)
     Open3.popen2(*args) do |i, o, _t|
       i.close
-      while (l = o.readpartial(8192))
-        $stdout << l
-      end
+      pipe_to_eof(o, $stdout)
     end
     true
   rescue SystemCallError
     nil
   end
+
+  def pipe_to_eof(src, dest)
+    loop do
+      data = src.readpartial(8192)
+      dest << data
+    rescue EOFError
+      break
+    end
+  end
 end
 
 # Override Timeout to use cancel scope

data/lib/polyphony/extensions/fiber.rb

@@ -238,7 +238,7 @@ module Polyphony
       @parent = parent
       @caller = caller
       @block = block
-      @mailbox = Gyro::Queue.new
+      @mailbox = Polyphony::Queue.new
       __fiber_trace__(:fiber_create, self)
       schedule
     end
@@ -268,7 +268,7 @@ module Polyphony
     # allows the fiber to be scheduled and to receive messages.
     def setup_raw
       @thread = Thread.current
-      @mailbox = Gyro::Queue.new
+      @mailbox = Polyphony::Queue.new
     end
 
     def setup_main_fiber
@@ -277,11 +277,11 @@ module Polyphony
       @thread = Thread.current
       @running = true
       @children&.clear
-      @mailbox = Gyro::Queue.new
+      @mailbox = Polyphony::Queue.new
     end
 
     def restart_self(first_value)
-      @mailbox = Gyro::Queue.new
+      @mailbox = Polyphony::Queue.new
       @when_done_procs = nil
       @waiting_fibers = nil
       run(first_value)

data/lib/polyphony/extensions/io.rb

@@ -2,7 +2,7 @@
 
 require 'open3'
 
-# IO overrides
+# IO class method patches
 class ::IO
   class << self
     alias_method :orig_binread, :binread
@@ -72,7 +72,10 @@ class ::IO
       Open3.popen2(cmd) { |_i, o, _t| yield o }
     end
   end
+end
 
+# IO instance method patches
+class ::IO
   # def each(sep = $/, limit = nil, chomp: nil)
   #   sep, limit = $/, sep if sep.is_a?(Integer)
   # end
@@ -93,6 +96,39 @@ class ::IO
   # def getc
   # end
 
+  alias_method :orig_read, :read
+  def read(len = 1 << 30)
+    @read_buffer ||= +''
+    result = Thread.current.agent.read(self, @read_buffer, len, true)
+    return nil unless result
+
+    already_read = @read_buffer
+    @read_buffer = +''
+    already_read
+  end
+
+  alias_method :orig_readpartial, :read
+  def readpartial(len)
+    @read_buffer ||= +''
+    result = Thread.current.agent.read(self, @read_buffer, len, false)
+    raise EOFError unless result
+
+    already_read = @read_buffer
+    @read_buffer = +''
+    already_read
+  end
+
+  alias_method :orig_write, :write
+  def write(str)
+    Thread.current.agent.write(self, str)
+  end
+
+  alias_method :orig_write_chevron, :<<
+  def <<(str)
+    Thread.current.agent.write(self, str)
+    self
+  end
+
   alias_method :orig_gets, :gets
   def gets(sep = $/, _limit = nil, _chomp: nil)
     if sep.is_a?(Integer)
@@ -101,23 +137,17 @@ class ::IO
     end
     sep_size = sep.bytesize
 
-    @gets_buffer ||= +''
+    @read_buffer ||= +''
 
     loop do
-      idx = @gets_buffer.index(sep)
-      return @gets_buffer.slice!(0, idx + sep_size) if idx
-
-      if (data = readpartial(8192))
-        @gets_buffer << data
-      else
-        return nil if @gets_buffer.empty?
+      idx = @read_buffer.index(sep)
+      return @read_buffer.slice!(0, idx + sep_size) if idx
 
-        line = @gets_buffer.freeze
-        @gets_buffer = +''
-        return line
-      end
+      data = readpartial(8192)
+      @read_buffer << data
+    rescue EOFError
+      return nil
     end
-    # orig_gets(sep, limit, chomp: chomp)
   end
 
   # def print(*args)
@@ -171,16 +201,20 @@ class ::IO
     buf ? readpartial(maxlen, buf) : readpartial(maxlen)
   end
 
-  alias_method :orig_read, :read
-  def read(length = nil, outbuf = nil)
-    if length
-      return outbuf ? readpartial(length) : readpartial(length, outbuf)
-    end
-
-    until eof?
-      outbuf ||= +''
-      outbuf << readpartial(8192)
-    end
-    outbuf
+  def read_loop(&block)
+    Thread.current.agent.read_loop(self, &block)
   end
+
+  # alias_method :orig_read, :read
+  # def read(length = nil, outbuf = nil)
+  #   if length
+  #     return outbuf ? readpartial(length) : readpartial(length, outbuf)
+  #   end
+
+  #   until eof?
+  #     outbuf ||= +''
+  #     outbuf << readpartial(8192)
+  #   end
+  #   outbuf
+  # end
 end