Sha256: f368a6f143dbb1ffb8f8cb9ea14161e26572e0fbd4b51399ef66108a6c928ef3
Contents?: true
Size: 1.38 KB
Versions: 10
Compression:
Stored size: 1.38 KB
Contents
# -*- encoding: binary -*-
# :enddoc:
class Rainbows::StreamResponseEpoll::Client
OUT = SleepyPenguin::Epoll::OUT
N = Raindrops.new(1)
EP = SleepyPenguin::Epoll.new
timeout = Rainbows.server.timeout
thr = Thread.new do
begin
EP.wait(nil, timeout) { |_,client| client.epoll_run }
rescue Errno::EINTR
rescue => e
Rainbows::Error.listen_loop(e)
end while Rainbows.alive || N[0] > 0
end
Rainbows.at_quit { thr.join(timeout) }
attr_reader :to_io
def initialize(io, unwritten)
@finish = false
@to_io = io
@wr_queue = [ unwritten.dup ]
EP.set(self, OUT)
end
def write(str)
@wr_queue << str.dup
end
def close
@finish = true
end
def hijack(hijack)
@finish = hijack
end
def epoll_run
return if @to_io.closed?
buf = @wr_queue.shift or return on_write_complete
case rv = @to_io.kgio_trywrite(buf)
when nil
buf = @wr_queue.shift or return on_write_complete
when String # retry, socket buffer may grow
buf = rv
when :wait_writable
return @wr_queue.unshift(buf)
end while true
rescue => err
@to_io.close
N.decr(0, 1)
end
def on_write_complete
if true == @finish
@to_io.shutdown
@to_io.close
N.decr(0, 1)
elsif @finish.respond_to?(:call) # hijacked
EP.delete(self)
N.decr(0, 1)
@finish.call(@to_io)
end
end
end
Version data entries
10 entries across 10 versions & 1 rubygems