polyphony 0.47.3 → 0.49.0

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

data/test/test_io.rb

@@ -92,6 +92,32 @@ class IOTest < MiniTest::Test
     assert_raises(EOFError) { i.readpartial(1) }    
   end
 
+  def test_gets
+    i, o = IO.pipe
+
+    buf = []
+    f = spin do
+      while (l = i.gets)
+        buf << l
+      end
+    end
+
+    snooze
+    assert_equal [], buf
+
+    o << 'fab'
+    snooze
+    assert_equal [], buf
+    
+    o << "ulous\n"
+    10.times { snooze }
+    assert_equal ["fabulous\n"], buf
+
+    o.close
+    f.await
+    assert_equal ["fabulous\n"], buf
+  end
+
   def test_getc
     i, o = IO.pipe
 

data/test/test_socket.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require_relative 'helper'
+require 'fileutils'
 
 class SocketTest < MiniTest::Test
   def setup
@@ -24,7 +25,32 @@ class SocketTest < MiniTest::Test
     snooze
     client = TCPSocket.new('127.0.0.1', port)
     client.write("1234\n")
-    assert_equal "1234\n", client.readpartial(8192)
+    assert_equal "1234\n", client.recv(8192)
+    client.close
+  ensure
+    server_fiber&.stop
+    server_fiber&.await
+    server&.close
+  end
+
+  def test_unix_socket
+    path = '/tmp/test_unix_socket'
+    FileUtils.rm(path) rescue nil
+    server = UNIXServer.new(path)
+    server_fiber = spin do
+      server.accept_loop do |socket|
+        spin do
+          while (data = socket.gets(8192))
+            socket << data
+          end
+        end
+      end
+    end
+
+    snooze
+    client = UNIXSocket.new(path)
+    client.write("1234\n")
+    assert_equal "1234\n", client.recv(8192)
     client.close
   ensure
     server_fiber&.stop
@@ -34,18 +60,18 @@ class SocketTest < MiniTest::Test
 end
 
 class HTTPClientTest < MiniTest::Test
-  require 'httparty'
   require 'json'
 
   def test_http
-    res = HTTParty.get('http://worldtimeapi.org/api/timezone/Europe/Paris')
+    res = HTTParty.get('http://ipinfo.io/')
+
     response = JSON.load(res.body)
-    assert_equal "CET", response['abbreviation']
+    assert_equal 'https://ipinfo.io/missingauth', response['readme']
   end
 
   def test_https
-    res = HTTParty.get('https://worldtimeapi.org/api/timezone/Europe/Paris')
+    res = HTTParty.get('https://ipinfo.io/')
     response = JSON.load(res.body)
-    assert_equal "CET", response['abbreviation']
+    assert_equal 'https://ipinfo.io/missingauth', response['readme']
   end
 end

data/test/test_supervise.rb

@@ -20,9 +20,10 @@ class SuperviseTest < MiniTest::Test
 
     f2 << 'bar'
     f2.await
+    assert_equal :runnable, p.state
     snooze
 
-    assert_equal :waiting, p.state
+    assert_equal :dead, p.state
   end
 
   def test_supervise_with_restart

data/test/test_timer.rb

@@ -0,0 +1,122 @@
+# frozen_string_literal: true
+
+require_relative 'helper'
+
+class TimerMoveOnAfterTest < MiniTest::Test
+  def setup
+    @timer = Polyphony::Timer.new(resolution: 0.01)
+  end
+
+  def teardown
+    @timer.stop
+  end
+
+  def test_move_on_after
+    t0 = Time.now
+    v = @timer.move_on_after(0.1) do
+      sleep 1
+      :foo
+    end
+    t1 = Time.now
+
+    assert_in_range 0.1..0.15, t1 - t0
+    assert_nil v
+  end
+
+  def test_move_on_after_with_value
+    t0 = Time.now
+    v = @timer.move_on_after(0.01, with_value: :bar) do
+      sleep 1
+      :foo
+    end
+    t1 = Time.now
+
+    assert_in_range 0.01..0.02, t1 - t0
+    assert_equal :bar, v
+  end
+
+  def test_move_on_after_with_reset
+    t0 = Time.now
+    v = @timer.move_on_after(0.01, with_value: :moved_on) do
+      sleep 0.007
+      @timer.reset
+      sleep 0.007
+      nil
+    end
+    t1 = Time.now
+
+    assert_nil v
+    assert_in_range 0.014..0.02, t1 - t0
+  end
+end
+
+class CancelAfterTest < MiniTest::Test
+  def setup
+    @timer = Polyphony::Timer.new(resolution: 0.01)
+  end
+
+  def teardown
+    @timer.stop
+  end
+
+  def test_cancel_after
+    t0 = Time.now
+
+    assert_raises Polyphony::Cancel do
+      @timer.cancel_after(0.01) do
+        sleep 1
+        :foo
+      end
+    end
+    t1 = Time.now
+    assert_in_range 0.01..0.02, t1 - t0
+  end
+
+  def test_cancel_after_with_reset
+    t0 = Time.now
+    @timer.cancel_after(0.01) do
+      sleep 0.007
+      @timer.reset
+      sleep 0.007
+    end
+    t1 = Time.now
+    assert_in_range 0.014..0.024, t1 - t0
+  end
+
+  class CustomException < Exception
+  end
+
+  def test_cancel_after_with_custom_exception
+    assert_raises CustomException do
+      @timer.cancel_after(0.01, with_exception: CustomException) do
+        sleep 1
+        :foo
+      end
+    end
+
+    begin
+      err = nil
+      @timer.cancel_after(0.01, with_exception: [CustomException, 'custom message']) do
+        sleep 1
+        :foo
+      end
+    rescue Exception => err
+    ensure
+      assert_kind_of CustomException, err
+      assert_equal 'custom message', err.message
+    end
+
+
+    begin
+      e = nil
+      @timer.cancel_after(0.01, with_exception: 'foo') do
+        sleep 1
+        :foo
+      end
+    rescue => e
+    ensure
+      assert_kind_of RuntimeError, e
+      assert_equal 'foo', e.message
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polyphony
 version: !ruby/object:Gem::Version
-  version: 0.47.3
+  version: 0.49.0
 platform: ruby
 authors:
 - Sharon Rosner
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-11-12 00:00:00.000000000 Z
+date: 2021-01-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-compiler
@@ -269,7 +269,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: 0.3.0
 description: 
-email: ciconia@gmail.com
+email: sharon@noteflakes.com
 executables: []
 extensions:
 - ext/polyphony/extconf.rb
@@ -383,13 +383,17 @@ files:
 - examples/io/raw.rb
 - examples/io/reline.rb
 - examples/io/system.rb
+- examples/io/tcp_proxy.rb
 - examples/io/tcpserver.rb
 - examples/io/tcpsocket.rb
 - examples/io/tunnel.rb
+- examples/io/unix_socket.rb
 - examples/io/zip.rb
 - examples/performance/fiber_resume.rb
 - examples/performance/fiber_transfer.rb
 - examples/performance/fs_read.rb
+- examples/performance/line_splitting.rb
+- examples/performance/loop.rb
 - examples/performance/mem-usage.rb
 - examples/performance/messaging.rb
 - examples/performance/multi_snooze.rb
@@ -432,7 +436,6 @@ files:
 - ext/liburing/setup.c
 - ext/liburing/syscall.c
 - ext/liburing/syscall.h
-- ext/polyphony/backend.h
 - ext/polyphony/backend_common.h
 - ext/polyphony/backend_io_uring.c
 - ext/polyphony/backend_io_uring_context.c
@@ -473,6 +476,7 @@ files:
 - lib/polyphony/core/sync.rb
 - lib/polyphony/core/thread_pool.rb
 - lib/polyphony/core/throttler.rb
+- lib/polyphony/core/timer.rb
 - lib/polyphony/extensions/core.rb
 - lib/polyphony/extensions/debug.rb
 - lib/polyphony/extensions/fiber.rb
@@ -507,6 +511,7 @@ files:
 - test/test_thread.rb
 - test/test_thread_pool.rb
 - test/test_throttler.rb
+- test/test_timer.rb
 - test/test_trace.rb
 homepage: https://digital-fabric.github.io/polyphony
 licenses:

data/ext/polyphony/backend.h

@@ -1,26 +0,0 @@
-#ifndef BACKEND_H
-#define BACKEND_H
-
-#include "ruby.h"
-
-typedef VALUE (* backend_pending_count_t)(VALUE self);
-typedef VALUE (*backend_poll_t)(VALUE self, VALUE nowait, VALUE current_fiber, VALUE runqueue);
-typedef VALUE (* backend_ref_t)(VALUE self);
-typedef int (* backend_ref_count_t)(VALUE self);
-typedef void (* backend_reset_ref_count_t)(VALUE self);
-typedef VALUE (* backend_unref_t)(VALUE self);
-typedef VALUE (* backend_wait_event_t)(VALUE self, VALUE raise_on_exception);
-typedef VALUE (* backend_wakeup_t)(VALUE self);
-
-typedef struct backend_interface {
-  backend_pending_count_t   pending_count;
-  backend_poll_t            poll;
-  backend_ref_t             ref;
-  backend_ref_count_t       ref_count;
-  backend_reset_ref_count_t reset_ref_count;
-  backend_unref_t           unref;
-  backend_wait_event_t      wait_event;
-  backend_wakeup_t          wakeup;
-} backend_interface_t;
-
-#endif /* BACKEND_H */