mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-10-16 00:01:48 -04:00
7169f6aaaf
this allows calling #status or #headers on a stream response, without buffering the whole response, as it's happening now; this will only work for methods which do not rely on the whole payload to be available, but that should be ok for the stream plugin usage Fixes https://github.com/janko/down/issues/98
32 lines
631 B
Ruby
32 lines
631 B
Ruby
# frozen_string_literal: true
|
|
|
|
module RequestInspector
|
|
module InstanceMethods
|
|
attr_reader :calls, :total_responses
|
|
|
|
def initialize(*args)
|
|
super
|
|
# we're comparing against max-retries + 1, because the calls increment will happen
|
|
# also in the last call, where the request is not going to be retried.
|
|
@calls = -1
|
|
@total_responses = []
|
|
end
|
|
|
|
def reset
|
|
@calls = -1
|
|
@total_responses.clear
|
|
end
|
|
|
|
private
|
|
|
|
def fetch_response(*)
|
|
response = super
|
|
if response
|
|
@calls += 1
|
|
@total_responses << response
|
|
end
|
|
response
|
|
end
|
|
end
|
|
end
|