Fiber*: add Rainbows::Fiber.sleep method

It works exactly like Actor.sleep and similar to Kernel.sleep
(no way to sleep indefinitely), but is compatible with the
IO.select-based Fiber scheduler we run.  This method only works
within the context of a Rainbows! application dispatch.

1 files changed, 24 insertions, 0 deletions

diff --git a/lib/rainbows/fiber.rb b/lib/rainbows/fiber.rb
index e7d64ca..94502a3 100644
--- a/lib/rainbows/fiber.rb
+++ b/lib/rainbows/fiber.rb
@@ -6,10 +6,33 @@ module Rainbows
   module Fiber
     RD = {}
     WR = {}
+    ZZ = {}
+
+    def self.sleep(seconds)
+      ZZ[::Fiber.current] = Time.now + seconds
+      ::Fiber.yield
+    end
 
     module Base
       include Rainbows::Base
 
+      # wakes up any sleepers that need to be woken and
+      # returns an interval to IO.select on
+      def timer
+        max = nil
+        now = Time.now
+        ZZ.delete_if { |fib, time|
+          if now >= time
+            fib.resume
+            now = Time.now
+          else
+            max = time
+            false
+          end
+        }
+        max.nil? || max > (now + 1) ? 1 : max - now
+      end
+
       def process_client(client)
         G.cur += 1
         io = client.to_io
@@ -46,6 +69,7 @@ module Rainbows
         G.cur -= 1
         RD.delete(client)
         WR.delete(client)
+        ZZ.delete(client.f)
       end
 
     end