From ad821f70a2488a91f2be1ac53cb2e64f50743989 Mon Sep 17 00:00:00 2001 From: Eric Wong Date: Tue, 28 Sep 2010 17:40:01 -0700 Subject: start using kgio library It removes the burden of byte slicing and setting file descriptor flags. In some cases, we can remove unnecessary peeraddr calls, too. --- lib/rainbows/rev/client.rb | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) (limited to 'lib/rainbows/rev/client.rb') diff --git a/lib/rainbows/rev/client.rb b/lib/rainbows/rev/client.rb index 8bfeb31..0c02525 100644 --- a/lib/rainbows/rev/client.rb +++ b/lib/rainbows/rev/client.rb @@ -5,7 +5,6 @@ module Rainbows module Rev class Client < ::Rev::IO - include Rainbows::ByteSlice include Rainbows::EvCore G = Rainbows::G F = Rainbows::StreamFile @@ -28,13 +27,14 @@ module Rainbows def write(buf) if @_write_buffer.empty? begin - w = @_io.write_nonblock(buf) - return enable_write_watcher if w == Rack::Utils.bytesize(buf) - # we never care for the return value, but yes, we may return - # a "fake" short write from super(buf) if anybody cares. - buf = byte_slice(buf, w..-1) - rescue Errno::EAGAIN - break # fall through to super(buf) + case rv = @_io.kgio_trywrite(buf) + when nil + return enable_write_watcher + when Kgio::WaitWritable + break # fall through to super(buf) + when String + buf = rv # retry, skb could grow or been drained + end rescue => e return handle_error(e) end while true @@ -104,7 +104,7 @@ module Rainbows def app_call KATO.delete(self) @env[RACK_INPUT] = @input - @env[REMOTE_ADDR] = @remote_addr + @env[REMOTE_ADDR] = @_io.kgio_addr response = APP.call(@env.update(RACK_DEFAULTS)) rev_write_response(response, alive = @hp.keepalive? && G.alive) -- cgit v1.2.3-24-ge0c7