1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
| | # -*- encoding: binary -*-
# :enddoc:
# non-portable body handling for Fiber-based concurrency goes here
# this module is required and included in worker processes only
# this is meant to be included _after_ Rainbows::Response::Body
module Rainbows::Fiber::Body # :nodoc:
# TODO non-blocking splice(2) under Linux
ALIASES = {
:write_body_stream => :write_body_each
}
# the sendfile 1.0.0+ gem includes IO#sendfile_nonblock
if ::IO.method_defined?(:sendfile_nonblock)
def write_body_file(client, body, range)
sock, n = client.to_io, nil
offset, count = range ? range : [ 0, body.stat.size ]
begin
offset += (n = sock.sendfile_nonblock(body, offset, count))
rescue Errno::EAGAIN
client.wait_writable
retry
rescue EOFError
break
end while (count -= n) > 0
end
else
ALIASES[:write_body] = :write_body_each
end
def self.included(klass)
ALIASES.each do |new_method, orig_method|
klass.__send__(:alias_method, new_method, orig_method)
end
end
end
|