mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-10-04 00:00:37 -04:00
452657c805
These are deadline oriented for the request and response, i.e. a write timeout tracks the full time it takes to write the request, whereas the read timeout does the same for receiving the response. For back-compat, they're infinite by default. v1 may change that, and will have to provide a safe fallback for endless "stream" requests and responses.
88 lines
1.9 KiB
Ruby
88 lines
1.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module HTTPX
|
|
class Timers
|
|
def initialize
|
|
@intervals = []
|
|
end
|
|
|
|
def after(interval_in_secs, &blk)
|
|
return unless interval_in_secs
|
|
|
|
# I'm assuming here that most requests will have the same
|
|
# request timeout, as in most cases they share common set of
|
|
# options. A user setting different request timeouts for 100s of
|
|
# requests will already have a hard time dealing with that.
|
|
unless (interval = @intervals.find { |t| t == interval_in_secs })
|
|
interval = Interval.new(interval_in_secs)
|
|
@intervals << interval
|
|
@intervals.sort!
|
|
end
|
|
|
|
interval << blk
|
|
end
|
|
|
|
def wait_interval
|
|
return if @intervals.empty?
|
|
|
|
@next_interval_at = Utils.now
|
|
|
|
@intervals.first.interval
|
|
end
|
|
|
|
def fire(error = nil)
|
|
raise error if error && error.timeout != @intervals.first
|
|
return if @intervals.empty? || !@next_interval_at
|
|
|
|
elapsed_time = Utils.elapsed_time(@next_interval_at)
|
|
|
|
@intervals.delete_if { |interval| interval.elapse(elapsed_time) <= 0 }
|
|
|
|
@next_interval_at = nil if @intervals.empty?
|
|
end
|
|
|
|
def cancel
|
|
@next_interval_at = nil
|
|
@intervals.clear
|
|
end
|
|
|
|
class Interval
|
|
include Comparable
|
|
|
|
attr_reader :interval
|
|
|
|
def initialize(interval)
|
|
@interval = interval
|
|
@callbacks = []
|
|
end
|
|
|
|
def <=>(other)
|
|
@interval <=> other.interval
|
|
end
|
|
|
|
def ==(other)
|
|
return @interval == other if other.is_a?(Numeric)
|
|
|
|
@interval == other.to_f # rubocop:disable Lint/FloatComparison
|
|
end
|
|
|
|
def to_f
|
|
Float(@interval)
|
|
end
|
|
|
|
def <<(callback)
|
|
@callbacks << callback
|
|
end
|
|
|
|
def elapse(elapsed)
|
|
@interval -= elapsed
|
|
|
|
@callbacks.each(&:call) if @interval <= 0
|
|
|
|
@interval
|
|
end
|
|
end
|
|
private_constant :Interval
|
|
end
|
|
end
|