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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
| | # -*- encoding: binary -*-
# :enddoc:
module Rainbows::ProcessClient
include Rainbows::Response
include Rainbows::Const
NULL_IO = Unicorn::HttpRequest::NULL_IO
RACK_INPUT = Unicorn::HttpRequest::RACK_INPUT
IC = Unicorn::HttpRequest.input_class
Rainbows.config!(self, :client_header_buffer_size)
def process_loop
@hp = hp = Rainbows::HttpParser.new
kgio_read!(CLIENT_HEADER_BUFFER_SIZE, buf = hp.buf) or return
begin # loop
until env = hp.parse
timed_read(buf2 ||= "") or return
buf << buf2
end
set_input(env, hp)
env[REMOTE_ADDR] = kgio_addr
status, headers, body = APP.call(env.merge!(RACK_DEFAULTS))
if 100 == status.to_i
write(EXPECT_100_RESPONSE)
env.delete(HTTP_EXPECT)
status, headers, body = APP.call(env)
end
write_response(status, headers, body, alive = @hp.next?)
end while alive
# if we get any error, try to write something back to the client
# assuming we haven't closed the socket, but don't get hung up
# if the socket is already closed or broken. We'll always ensure
# the socket is closed at the end of this function
rescue => e
handle_error(e)
ensure
close unless closed?
end
def handle_error(e)
Rainbows::Error.write(self, e)
end
def set_input(env, hp)
env[RACK_INPUT] = 0 == hp.content_length ? NULL_IO : IC.new(self, hp)
end
def process_pipeline(env, hp)
begin
set_input(env, hp)
env[REMOTE_ADDR] = kgio_addr
status, headers, body = APP.call(env.merge!(RACK_DEFAULTS))
if 100 == status.to_i
write(EXPECT_100_RESPONSE)
env.delete(HTTP_EXPECT)
status, headers, body = APP.call(env)
end
write_response(status, headers, body, alive = hp.next?)
end while alive && pipeline_ready(hp)
alive or close
rescue => e
handle_error(e)
end
# override this in subclass/module
def pipeline_ready(hp)
end
end
|