1 files changed, 48 insertions, 0 deletions
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)