polyphony 0.47.4 → 0.49.1

data/lib/polyphony/core/timer.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+module Polyphony
+  # Implements a common timer for running multiple timeouts
+  class Timer
+    def initialize(resolution:)
+      @fiber = spin_loop(interval: resolution) { update }
+      @timeouts = {}
+    end
+
+    def stop
+      @fiber.stop
+    end
+  
+    def cancel_after(duration, with_exception: Polyphony::Cancel)
+      fiber = Fiber.current
+      @timeouts[fiber] = {
+        duration: duration,
+        target_stamp: Time.now + duration,
+        exception: with_exception
+      }
+      yield
+    ensure
+      @timeouts.delete(fiber)
+    end
+
+    def move_on_after(duration, with_value: nil)
+      fiber = Fiber.current
+      @timeouts[fiber] = {
+        duration: duration,
+        target_stamp: Time.now + duration,
+        value: with_value
+      }
+      yield
+    rescue Polyphony::MoveOn => e
+      e.value
+    ensure
+      @timeouts.delete(fiber)
+    end
+  
+    def reset
+      record = @timeouts[Fiber.current]
+      return unless record
+  
+      record[:target_stamp] = Time.now + record[:duration]
+    end
+  
+    private
+
+    def timeout_exception(record)
+      case (exception = record[:exception])
+      when Class then exception.new
+      when Array then exception[0].new(exception[1])
+      when nil then Polyphony::MoveOn.new(record[:value])
+      else RuntimeError.new(exception)
+      end
+    end
+
+    def update
+      now = Time.now
+      # elapsed = nil
+      @timeouts.each do |fiber, record|
+        next if record[:target_stamp] > now
+
+        exception = timeout_exception(record)
+        # (elapsed ||= []) << fiber
+        fiber.schedule exception
+      end
+      # elapsed&.each { |f| @timeouts.delete(f) }
+    end
+  end
+end

data/lib/polyphony/extensions/fiber.rb

@@ -34,9 +34,18 @@ module Polyphony
       schedule Polyphony::Cancel.new
     end
 
-    def terminate
+    def graceful_shutdown=(graceful)
+      @graceful_shutdown = graceful
+    end
+
+    def graceful_shutdown?
+      @graceful_shutdown
+    end
+
+    def terminate(graceful = false)
       return if @running == false
 
+      @graceful_shutdown = graceful
       schedule Polyphony::Terminate.new
     end
 
@@ -66,20 +75,32 @@ module Polyphony
       @on_child_done = proc do |fiber, result|
         self << fiber unless result.is_a?(Exception)
       end
-      loop { supervise_perform(opts) }
+      while true
+        supervise_perform(opts)
+      end
+    rescue Polyphony::MoveOn
+      # generated in #supervise_perform to stop supervisor
     ensure
       @on_child_done = nil
     end
 
     def supervise_perform(opts)
       fiber = receive
-      restart_fiber(fiber, opts) if fiber
+      if fiber && opts[:restart]
+        restart_fiber(fiber, opts)
+      elsif Fiber.current.children.empty?
+        Fiber.current.stop
+      end
     rescue Polyphony::Restart
       restart_all_children
     rescue Exception => e
       Kernel.raise e if e.source_fiber.nil? || e.source_fiber == self
 
-      restart_fiber(e.source_fiber, opts)
+      if opts[:restart]
+        restart_fiber(e.source_fiber, opts)
+      elsif Fiber.current.children.empty?
+        Fiber.current.stop
+      end
     end
 
     def restart_fiber(fiber, opts)
@@ -200,11 +221,14 @@ module Polyphony
       @on_child_done&.(child_fiber, result)
     end
 
-    def terminate_all_children
+    def terminate_all_children(graceful = false)
       return unless @children
 
       e = Polyphony::Terminate.new
-      @children.each_key { |c| c.raise e }
+      @children.each_key do |c|
+        c.graceful_shutdown = true if graceful
+        c.raise e
+      end
     end
 
     def await_all_children
@@ -220,8 +244,8 @@ module Polyphony
       results.values
     end
 
-    def shutdown_all_children
-      terminate_all_children
+    def shutdown_all_children(graceful = false)
+      terminate_all_children(graceful)
       await_all_children
     end
   end

data/lib/polyphony/extensions/io.rb

@@ -119,18 +119,11 @@ class ::IO
   end
 
   alias_method :orig_readpartial, :read
-  def readpartial(len, str = nil)
-    @read_buffer ||= +''
-    result = Thread.current.backend.read(self, @read_buffer, len, false)
+  def readpartial(len, str = +'')
+    result = Thread.current.backend.read(self, str, len, false)
     raise EOFError unless result
 
-    if str
-      str << @read_buffer
-    else
-      str = @read_buffer
-    end
-    @read_buffer = +''
-    str
+    result
   end
 
   alias_method :orig_write, :write
@@ -154,15 +147,16 @@ class ::IO
 
     @read_buffer ||= +''
 
-    loop do
+    while true
       idx = @read_buffer.index(sep)
       return @read_buffer.slice!(0, idx + sep_size) if idx
 
-      data = readpartial(8192)
+      data = readpartial(8192, +'')
+      return nil unless data
       @read_buffer << data
-    rescue EOFError
-      return nil
     end
+  rescue EOFError
+    return nil
   end
 
   # def print(*args)

data/lib/polyphony/extensions/openssl.rb

@@ -25,7 +25,7 @@ class ::OpenSSL::SSL::SSLSocket
 
   alias_method :orig_accept, :accept
   def accept
-    loop do
+    while true
       result = accept_nonblock(exception: false)
       case result
       when :wait_readable then Thread.current.backend.wait_io(io, false)
@@ -37,14 +37,14 @@ class ::OpenSSL::SSL::SSLSocket
   end
 
   def accept_loop
-    loop do
+    while true
       yield accept
     end
   end
 
   alias_method :orig_sysread, :sysread
   def sysread(maxlen, buf = +'')
-    loop do
+    while true
       case (result = read_nonblock(maxlen, buf, exception: false))
       when :wait_readable then Thread.current.backend.wait_io(io, false)
       when :wait_writable then Thread.current.backend.wait_io(io, true)
@@ -55,7 +55,7 @@ class ::OpenSSL::SSL::SSLSocket
 
   alias_method :orig_syswrite, :syswrite
   def syswrite(buf)
-    loop do
+    while true
       case (result = write_nonblock(buf, exception: false))
       when :wait_readable then Thread.current.backend.wait_io(io, false)
       when :wait_writable then Thread.current.backend.wait_io(io, true)

data/lib/polyphony/extensions/socket.rb

@@ -8,7 +8,11 @@ require_relative '../core/thread_pool'
 # Socket overrides (eventually rewritten in C)
 class ::Socket
   def accept
-    Thread.current.backend.accept(self)
+    Thread.current.backend.accept(self, TCPSocket)
+  end
+
+  def accept_loop(&block)
+    Thread.current.backend.accept_loop(self, TCPSocket, &block)
   end
 
   NO_EXCEPTION = { exception: false }.freeze
@@ -29,7 +33,7 @@ class ::Socket
 
   def recvfrom(maxlen, flags = 0)
     @read_buffer ||= +''
-    loop do
+    while true
       result = recvfrom_nonblock(maxlen, flags, @read_buffer, **NO_EXCEPTION)
       case result
       when nil then raise IOError
@@ -134,7 +138,7 @@ class ::TCPSocket
   end
 
   def recv(maxlen, flags = 0, outbuf = nil)
-    Thread.current.backend.recv(self, buf || +'', maxlen)
+    Thread.current.backend.recv(self, outbuf || +'', maxlen)
   end
 
   def recv_loop(&block)
@@ -188,11 +192,12 @@ class ::TCPServer
 
   alias_method :orig_accept, :accept
   def accept
-    @io.accept
+    Thread.current.backend.accept(@io, TCPSocket)
+    # @io.accept
   end
 
   def accept_loop(&block)
-    Thread.current.backend.accept_loop(@io, &block)
+    Thread.current.backend.accept_loop(@io, TCPSocket, &block)
   end
 
   alias_method :orig_close, :close
@@ -204,19 +209,17 @@ end
 class ::UNIXServer
   alias_method :orig_accept, :accept
  def accept
-    Thread.current.backend.accept(self)
+    Thread.current.backend.accept(self, UNIXSocket)
   end
 
   def accept_loop(&block)
-    Thread.current.backend.accept_loop(self, &block)
+    Thread.current.backend.accept_loop(self, UNIXSocket, &block)
   end
-
-
 end
 
 class ::UNIXSocket
   def recv(maxlen, flags = 0, outbuf = nil)
-    Thread.current.backend.recv(self, buf || +'', maxlen)
+    Thread.current.backend.recv(self, outbuf || +'', maxlen)
   end
 
   def recv_loop(&block)

data/lib/polyphony/net.rb

@@ -8,10 +8,7 @@ module Polyphony
   module Net
     class << self
       def tcp_connect(host, port, opts = {})
-        socket = ::Socket.new(:INET, :STREAM).tap do |s|
-          addr = ::Socket.sockaddr_in(port, host)
-          s.connect(addr)
-        end
+        socket = TCPSocket.new(host, port)
         if opts[:secure_context] || opts[:secure]
           secure_socket(socket, opts[:secure_context], opts.merge(host: host))
         else
@@ -23,7 +20,7 @@ module Polyphony
         host ||= '0.0.0.0'
         raise 'Port number not specified' unless port
 
-        socket = socket_from_options(host, port, opts)
+        socket = listening_socket_from_options(host, port, opts)
         if opts[:secure_context] || opts[:secure]
           secure_server(socket, opts[:secure_context], opts)
         else
@@ -31,7 +28,7 @@ module Polyphony
         end
       end
 
-      def socket_from_options(host, port, opts)
+      def listening_socket_from_options(host, port, opts)
         ::Socket.new(:INET, :STREAM).tap do |s|
           s.reuse_addr if opts[:reuse_addr]
           s.dont_linger if opts[:dont_linger]

data/lib/polyphony/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Polyphony
-  VERSION = '0.47.4'
+  VERSION = '0.49.1'
 end

data/polyphony.gemspec

@@ -6,7 +6,7 @@ Gem::Specification.new do |s|
   s.licenses    = ['MIT']
   s.summary     = 'Fine grained concurrency for Ruby'
   s.author      = 'Sharon Rosner'
-  s.email       = 'ciconia@gmail.com'
+  s.email       = 'sharon@noteflakes.com'
   s.files       = `git ls-files`.split
   s.homepage    = 'https://digital-fabric.github.io/polyphony'
   s.metadata    = {

data/test/helper.rb

@@ -4,6 +4,7 @@ require 'bundler/setup'
 
 require_relative './coverage' if ENV['COVERAGE']
 
+require 'httparty'
 require 'polyphony'
 
 require 'fileutils'

data/test/test_backend.rb

@@ -105,7 +105,7 @@ class BackendTest < MiniTest::Test
 
     clients = []
     server_fiber = spin do
-      @backend.accept_loop(server) { |c| clients << c }
+      @backend.accept_loop(server, TCPSocket) { |c| clients << c }
     end
 
     c1 = TCPSocket.new('127.0.0.1', 1234)

data/test/test_fiber.rb

@@ -639,7 +639,6 @@ class FiberTest < MiniTest::Test
         i.close
         f.await
       rescue Exception => e
-        trace e
         o << e.class.name
         o.close
       end
@@ -1038,3 +1037,67 @@ class RestartTest < MiniTest::Test
     assert_equal [f, 'foo', 'bar', :done, f2, 'baz', 42, :done], buffer
   end
 end
+
+class GracefulTerminationTest < MiniTest::Test
+  def test_graceful_termination
+    buffer = []
+    f = spin do
+      buffer << 1
+      snooze
+      buffer << 2
+      sleep 3
+      buffer << 3
+    ensure
+      buffer << 4 if Fiber.current.graceful_shutdown?
+    end
+
+    3.times { snooze }
+    f.terminate(false)
+    f.await
+    assert_equal [1, 2], buffer
+
+    buffer = []
+    f = spin do
+      buffer << 1
+      snooze
+      buffer << 2
+      sleep 3
+      buffer << 3
+    ensure
+      buffer << 4 if Fiber.current.graceful_shutdown?
+    end
+
+    3.times { snooze }
+    f.terminate(true)
+    f.await
+    assert_equal [1, 2, 4], buffer
+  end
+
+  def test_graceful_child_shutdown
+    buffer = []
+    f0 = spin do
+      f1 = spin do
+        sleep
+      ensure
+        buffer << 1 if Fiber.current.graceful_shutdown?
+      end
+
+      f2 = spin do
+        sleep
+      ensure
+        buffer << 2 if Fiber.current.graceful_shutdown?
+      end
+
+      sleep
+    ensure
+      Fiber.current.terminate_all_children(true) if Fiber.current.graceful_shutdown?
+      Fiber.current.await_all_children
+    end
+
+    3.times { snooze }
+    f0.terminate(true)
+    f0.await
+
+    assert_equal [1, 2], buffer
+  end
+end

data/test/test_global_api.rb

@@ -307,6 +307,36 @@ class SpinLoopTest < MiniTest::Test
     f.stop
     assert_in_range 1..3, counter
   end
+
+  def test_spin_loop_break
+    i = 0
+    f = spin_loop do
+      i += 1
+      snooze
+      break if i >= 5
+    end
+    f.await
+    assert_equal 5, i
+
+    i = 0
+    f = spin_loop do
+      i += 1
+      snooze
+      raise StopIteration if i >= 5
+    end
+    f.await
+    assert_equal 5, i
+  end
+
+  def test_throttled_spin_loop_break
+    i = 0
+    f = spin_loop(rate: 100) do
+      i += 1
+      break if i >= 5
+    end
+    f.await
+    assert_equal 5, i
+  end
 end
 
 class SpinScopeTest < MiniTest::Test