diff options
author | Eric Wong <normalperson@yhbt.net> | 2010-10-21 16:25:39 -0700 |
---|---|---|
committer | Eric Wong <normalperson@yhbt.net> | 2010-10-22 18:37:45 +0000 |
commit | 15631717fce044fbad2f386a7b1c7daf4bdd83d2 (patch) | |
tree | f32c80aafb8b5fe13fefe9a1e3765dd757ccde7d /lib/rainbows/fiber/io/methods.rb | |
parent | d4a2b5dd2b85f4b2d3bb120ee1e1b0dde31bc25c (diff) | |
download | rainbows-15631717fce044fbad2f386a7b1c7daf4bdd83d2.tar.gz |
Despite the large number of changes, most of it is code movement here.
Diffstat (limited to 'lib/rainbows/fiber/io/methods.rb')
-rw-r--r-- | lib/rainbows/fiber/io/methods.rb | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/lib/rainbows/fiber/io/methods.rb b/lib/rainbows/fiber/io/methods.rb new file mode 100644 index 0000000..663fdb4 --- /dev/null +++ b/lib/rainbows/fiber/io/methods.rb @@ -0,0 +1,44 @@ +# -*- encoding: binary -*- + +module Rainbows::Fiber::IO::Methods + RD = Rainbows::Fiber::RD + WR = Rainbows::Fiber::WR + attr_accessor :f + + # for wrapping output response bodies + def each(&block) + if buf = kgio_read(16384) + yield buf + yield buf while kgio_read(16384, buf) + end + self + end + + def close + fd = fileno + RD[fd] = WR[fd] = nil + super + end + + def wait_readable + fd = fileno + @f = Fiber.current + RD[fd] = self + Fiber.yield + RD[fd] = nil + end + + def wait_writable + fd = fileno + @f = Fiber.current + WR[fd] = self + Fiber.yield + WR[fd] = nil + end + + def self.included(klass) + if klass.method_defined?(:kgio_write) + klass.__send__(:alias_method, :write, :kgio_write) + end + end +end |