5 files changed, 125 insertions, 0 deletions
diff --git a/lib/rainbows/fiber/rev/heartbeat.rb b/lib/rainbows/fiber/rev/heartbeat.rb
new file mode 100644
index 0000000..9411b4a
--- /dev/null
+++ b/lib/rainbows/fiber/rev/heartbeat.rb
@@ -0,0 +1,8 @@
+# -*- encoding: binary -*-
+# :enddoc:
+class Rainbows::Fiber::Rev::Heartbeat < Rev::TimerWatcher
+  G = Rainbows::G
+  def on_timer
+    exit if (! G.tick && G.cur <= 0)
+  end
+end
diff --git a/lib/rainbows/fiber/rev/kato.rb b/lib/rainbows/fiber/rev/kato.rb
new file mode 100644
index 0000000..056b6ef
--- /dev/null
+++ b/lib/rainbows/fiber/rev/kato.rb
@@ -0,0 +1,22 @@
+# -*- encoding: binary -*-
+# :enddoc:
+# keep-alive timeout class
+class Rainbows::Fiber::Rev::Kato < Rev::TimerWatcher
+  def initialize
+    @watch = []
+    super(1, true)
+  end
+
+  def <<(fiber)
+    @watch << fiber
+    enable unless enabled?
+  end
+
+  def on_timer
+    @watch.uniq!
+    while f = @watch.shift
+      f.resume if f.alive?
+    end
+    disable
+  end
+end
diff --git a/lib/rainbows/fiber/rev/methods.rb b/lib/rainbows/fiber/rev/methods.rb
new file mode 100644
index 0000000..5f4367e
--- /dev/null
+++ b/lib/rainbows/fiber/rev/methods.rb
@@ -0,0 +1,48 @@
+# -*- encoding: binary -*-
+# :enddoc:
+module Rainbows::Fiber::Rev::Methods
+  class Watcher < Rev::IOWatcher
+    def initialize(fio, flag)
+      @f = fio.f || Fiber.current
+      super(fio, flag)
+      attach(Rev::Loop.default)
+    end
+
+    def on_readable
+      @f.resume
+    end
+
+    alias on_writable on_readable
+  end
+
+  def initialize(*args)
+    @f = Fiber.current
+    super(*args)
+    @r = @w = false
+  end
+
+  def close
+    @w.detach if @w
+    @r.detach if @r
+    @r = @w = false
+    super
+  end
+
+  def wait_writable
+    @w ||= Watcher.new(self, :w)
+    @w.enable unless @w.enabled?
+    Fiber.yield
+    @w.disable
+  end
+
+  def wait_readable
+    @r ||= Watcher.new(self, :r)
+    @r.enable unless @r.enabled?
+    KATO << @f
+    Fiber.yield
+    @r.disable
+  end
+end
+
+Rainbows::Fiber::IO.__send__(:include, Rainbows::Fiber::Rev::Methods)
+Rainbows::Client.__send__(:include, Rainbows::Fiber::Rev::Methods)
diff --git a/lib/rainbows/fiber/rev/server.rb b/lib/rainbows/fiber/rev/server.rb
new file mode 100644
index 0000000..9998cde
--- /dev/null
+++ b/lib/rainbows/fiber/rev/server.rb
@@ -0,0 +1,32 @@
+# -*- encoding: binary -*-
+# :enddoc:
+class Rainbows::Fiber::Rev::Server < Rev::IOWatcher
+  G = Rainbows::G
+  include Rainbows::ProcessClient
+
+  def to_io
+    @io
+  end
+
+  def initialize(io)
+    @io = io
+    super(self, :r)
+  end
+
+  def close
+    detach if attached?
+    @io.close
+  end
+
+  def on_readable
+    return if G.cur >= MAX
+    c = @io.kgio_tryaccept and Fiber.new { process(c) }.resume
+  end
+
+  def process(io)
+    G.cur += 1
+    process_client(io)
+  ensure
+    G.cur -= 1
+  end
+end
diff --git a/lib/rainbows/fiber/rev/sleeper.rb b/lib/rainbows/fiber/rev/sleeper.rb
new file mode 100644
index 0000000..51f4527
--- /dev/null
+++ b/lib/rainbows/fiber/rev/sleeper.rb
@@ -0,0 +1,15 @@
+# -*- encoding: binary -*-
+# :enddoc:
+class Rainbows::Fiber::Rev::Sleeper < Rev::TimerWatcher
+
+  def initialize(seconds)
+    @f = Fiber.current
+    super(seconds, false)
+    attach(Rev::Loop.default)
+    Fiber.yield
+  end
+
+  def on_timer
+    @f.resume
+  end
+end