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
72
73
74
75
76
77
78
79
80
81
82
|
require "net_http_hacked"
require "stringio"
module Rack
# Wraps the hacked net/http in a Rack way.
class HttpStreamingResponse
STATUSES_WITH_NO_ENTITY_BODY = {
204 => true,
205 => true,
304 => true
}.freeze
attr_accessor :use_ssl, :verify_mode, :read_timeout, :ssl_version, :cert, :key
def initialize(request, host, port = nil)
@request, @host, @port = request, host, port
end
def body
self
end
def code
response.code.to_i.tap do |response_code|
STATUSES_WITH_NO_ENTITY_BODY[response_code] && close_connection
end
end
# #status is deprecated
alias_method :status, :code
def headers
Rack::Proxy.build_header_hash(response.to_hash)
end
# Can be called only once!
def each(&block)
return if connection_closed
response.read_body(&block)
ensure
close_connection
end
def to_s
@to_s ||= StringIO.new.tap { |io| each { |line| io << line } }.string
end
protected
# Net::HTTPResponse
def response
@response ||= session.begin_request_hacked(request)
end
# Net::HTTP
def session
@session ||= Net::HTTP.new(host, port).tap do |http|
http.use_ssl = use_ssl
http.verify_mode = verify_mode
http.read_timeout = read_timeout
http.ssl_version = ssl_version if ssl_version
http.cert = cert if cert
http.key = key if key
http.start
end
end
private
attr_reader :request, :host, :port
attr_accessor :connection_closed
def close_connection
return if connection_closed
session.end_request_hacked
session.finish
self.connection_closed = true
end
end
end
|
