Sha256: 5b9dcf74a3d35b118ab50394ad00eefc582d65bcd0dbfda977c172361994b30c
Contents?: true
Size: 1.96 KB
Versions: 7
Compression:
Stored size: 1.96 KB
Contents
module NightcrawlerSwift
class Gateway
attr_reader :resource, :attempts, :current_retry_time
RETRY_BLACKLIST = [
RestClient::Unauthorized,
RestClient::ResourceNotFound,
RestClient::UnprocessableEntity
]
def initialize url
@url = url
@attempts = 0
@current_retry_time = 1
@retries = NightcrawlerSwift.options.retries
@max_retry_time = NightcrawlerSwift.options.max_retry_time
@resource = RestClient::Resource.new(
@url,
{timeout: options.timeout}.merge(ssl_options)
)
end
def request &block
begin
@attempts += 1
block.call(resource)
rescue => e
raise e unless recoverable?(e)
wait(e) and retry
end
rescue RestClient::Unauthorized => e
raise Exceptions::UnauthorizedError.new(e)
rescue RestClient::ResourceNotFound => e
raise Exceptions::NotFoundError.new(e)
rescue RestClient::UnprocessableEntity => e
raise Exceptions::ValidationError.new(e)
rescue => e
raise Exceptions::ConnectionError.new(e)
end
private
def log message
NightcrawlerSwift.logger.debug message
end
def options
NightcrawlerSwift.options
end
def ssl_options
[
:verify_ssl,
:ssl_client_cert,
:ssl_client_key,
:ssl_ca_file,
:ssl_version
].inject({}) {|hash, key|
hash.tap {
# OpenStruct in Ruby 1.9.x doesn't have the method []
hash[key] = options.send(key)
}
}.compact
end
def recoverable? e
@retries and
!RETRY_BLACKLIST.include?(e.class) and
@attempts <= @retries
end
def wait e
number = "#{@attempts}/#{@retries}"
log "Attempt #{number} to call '#{@url}', waiting #{@current_retry_time}s and retrying. Error: #{e.message}"
sleep @current_retry_time
@current_retry_time = [@current_retry_time * 2, @max_retry_time].min
end
end
end
Version data entries
7 entries across 7 versions & 1 rubygems