JezK

Edit File: http_streaming_response.rb

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

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
      Utils::HeaderHash.new.tap do |h|
        response.to_hash.each { |k, v| h[k] = v }
      end
    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 use_ssl
        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

Anmeldung Newsletter | Fairnetzt-Hochrhein