JezK
Edit File: net_http_hacked.rb
# We are hacking net/http to change semantics of streaming handling # from "block" semantics to regular "return" semantics. # We need it to construct a streamable rack triplet: # # [status, headers, streamable_body] # # See http://github.com/zerowidth/rack-streaming-proxy # for alternative that uses additional process. # # BTW I don't like monkey patching either # but this is not real monkey patching. # I just added some methods and named them very uniquely # to avoid eventual conflicts. You're safe. Trust me. # # Also, in Ruby 1.9.2 you could use Fibers to avoid hacking net/http. require 'net/https' class Net::HTTP # Original #request with block semantics. # # def request(req, body = nil, &block) # unless started? # start { # req['connection'] ||= 'close' # return request(req, body, &block) # } # end # if proxy_user() # unless use_ssl? # req.proxy_basic_auth proxy_user(), proxy_pass() # end # end # # req.set_body_internal body # begin_transport req # req.exec @socket, @curr_http_version, edit_path(req.path) # begin # res = HTTPResponse.read_new(@socket) # end while res.kind_of?(HTTPContinue) # res.reading_body(@socket, req.response_body_permitted?) { # yield res if block_given? # } # end_transport req, res # # res # end def begin_request_hacked(req) begin_transport req req.exec @socket, @curr_http_version, edit_path(req.path) begin res = Net::HTTPResponse.read_new(@socket) end while res.kind_of?(Net::HTTPContinue) res.begin_reading_body_hacked(@socket, req.response_body_permitted?) @req_hacked, @res_hacked = req, res @res_hacked end def end_request_hacked @res_hacked.end_reading_body_hacked end_transport @req_hacked, @res_hacked @res_hacked end end class Net::HTTPResponse # Original #reading_body with block semantics # # def reading_body(sock, reqmethodallowbody) #:nodoc: internal use only # @socket = sock # @body_exist = reqmethodallowbody && self.class.body_permitted? # begin # yield # self.body # ensure to read body # ensure # @socket = nil # end # end def begin_reading_body_hacked(sock, reqmethodallowbody) @socket = sock @body_exist = reqmethodallowbody && self.class.body_permitted? end def end_reading_body_hacked self.body @socket = nil end end