about summary refs log tree commit homepage
path: root/lib/rainbows/fiber/coolio/methods.rb
diff options
context:
space:
mode:
authorEric Wong <normalperson@yhbt.net>2010-12-28 17:59:27 -0800
committerEric Wong <normalperson@yhbt.net>2010-12-28 18:00:04 -0800
commit40445641f11f01c6a24bf96c8b80eed5fd33a512 (patch)
tree57a7652cc03f46407d51babfd04d72d1c401ac99 /lib/rainbows/fiber/coolio/methods.rb
parent3495d59763e6159975debf32728dc53fc41c5ea1 (diff)
downloadrainbows-40445641f11f01c6a24bf96c8b80eed5fd33a512.tar.gz
complete Rev => Coolio renaming
We use Cool.io internally everywhere now, but preserve
Rev-based models for anybody using them.
Diffstat (limited to 'lib/rainbows/fiber/coolio/methods.rb')
-rw-r--r--lib/rainbows/fiber/coolio/methods.rb47
1 files changed, 47 insertions, 0 deletions
diff --git a/lib/rainbows/fiber/coolio/methods.rb b/lib/rainbows/fiber/coolio/methods.rb
new file mode 100644
index 0000000..64b0ee6
--- /dev/null
+++ b/lib/rainbows/fiber/coolio/methods.rb
@@ -0,0 +1,47 @@
+# -*- encoding: binary -*-
+# :enddoc:
+module Rainbows::Fiber::Coolio::Methods
+  class Watcher < Coolio::IOWatcher
+    def initialize(fio, flag)
+      @f = Fiber.current
+      super(fio, flag)
+      attach(Coolio::Loop.default)
+    end
+
+    def on_readable
+      @f.resume
+    end
+
+    alias on_writable on_readable
+  end
+
+  def close
+    @w.detach if defined?(@w) && @w.attached?
+    @r.detach if defined?(@r) && @r.attached?
+    super
+  end
+
+  def kgio_wait_writable
+    @w = Watcher.new(self, :w) unless defined?(@w)
+    @w.enable unless @w.enabled?
+    Fiber.yield
+    @w.disable
+  end
+
+  def kgio_wait_readable
+    @r = Watcher.new(self, :r) unless defined?(@r)
+    @r.enable unless @r.enabled?
+    Fiber.yield
+    @r.disable
+  end
+end
+
+[
+  Rainbows::Fiber::IO,
+  Rainbows::Client,
+  # the next two trigger autoload, ugh, oh well...
+  Rainbows::Fiber::IO::Socket,
+  Rainbows::Fiber::IO::Pipe
+].each do |klass|
+  klass.__send__(:include, Rainbows::Fiber::Coolio::Methods)
+end