diff options
Diffstat (limited to 'lib/rainbows/fiber/coolio/methods.rb')
-rw-r--r-- | lib/rainbows/fiber/coolio/methods.rb | 47 |
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 |