module ActiveUtils::NetworkConnectionRetries

Constants

DEFAULT_CONNECTION_ERRORS
DEFAULT_RETRIES
DEFAULT_RETRY_ERRORS

Public Class Methods

included(base) click to toggle source
# File lib/active_utils/network_connection_retries.rb, line 17
def self.included(base)
  base.send(:attr_accessor, :retry_safe)
end

Private Class Methods

log(logger, level, message, tag=nil) click to toggle source
# File lib/active_utils/network_connection_retries.rb, line 73
def self.log(logger, level, message, tag=nil)
  tag ||= self.class.to_s
  message = "[#{tag}] #{message}"
  logger.send(level, message) if logger
end

Public Instance Methods

retry_exceptions(options={}) { || ... } click to toggle source
# File lib/active_utils/network_connection_retries.rb, line 21
def retry_exceptions(options={})
  connection_errors = DEFAULT_CONNECTION_ERRORS.merge(options[:connection_exceptions] || {})
  retry_errors = DEFAULT_RETRY_ERRORS.merge(options[:retriable_exceptions] || {})

  retry_network_exceptions(options) do
    begin
      yield
    rescue *retry_errors.keys => e
      raise ActiveUtils::RetriableConnectionError, derived_error_message(retry_errors, e.class)
    rescue OpenSSL::X509::CertificateError => e
      NetworkConnectionRetries.log(options[:logger], :error, e.message, options[:tag])
      raise ActiveUtils::ClientCertificateError, "The remote server did not accept the provided SSL certificate"
    rescue Zlib::BufError => e
      raise ActiveUtils::InvalidResponseError, "The remote server replied with an invalid response"
    rescue *connection_errors.keys => e
      raise ActiveUtils::ConnectionError, derived_error_message(connection_errors, e.class)
    end
  end
end

Private Instance Methods

derived_error_message(errors, klass) click to toggle source
# File lib/active_utils/network_connection_retries.rb, line 84
def derived_error_message(errors, klass)
  key = (errors.keys & klass.ancestors).first
  key ? errors[key] : nil
end
log_with_retry_details(logger, attempts, time, message, tag) click to toggle source
# File lib/active_utils/network_connection_retries.rb, line 80
def log_with_retry_details(logger, attempts, time, message, tag)
  NetworkConnectionRetries.log(logger, :info, "connection_attempt=%d connection_request_time=%.4fs connection_msg=\"%s\"" % [attempts, time, message], tag)
end
retry_network_exceptions(options = {}) { || ... } click to toggle source
# File lib/active_utils/network_connection_retries.rb, line 43
def retry_network_exceptions(options = {})
  initial_retries = options[:max_retries] || DEFAULT_RETRIES
  retries = initial_retries
  request_start = nil

  begin
    request_start = Time.now.to_f
    result = yield
    log_with_retry_details(options[:logger], initial_retries-retries + 1, Time.now.to_f - request_start, "success", options[:tag])
    result
  rescue ActiveUtils::RetriableConnectionError => e
    retries -= 1

    log_with_retry_details(options[:logger], initial_retries-retries, Time.now.to_f - request_start, e.message, options[:tag])
    unless retries.zero?
      Kernel.sleep(options[:delay]) if options[:delay]
      retry
    end
    raise ActiveUtils::ConnectionError, e.message
  rescue ActiveUtils::ConnectionError, ActiveUtils::InvalidResponseError => e
    retries -= 1
    log_with_retry_details(options[:logger], initial_retries-retries, Time.now.to_f - request_start, e.message, options[:tag])
    if (options[:retry_safe] || retry_safe) && !retries.zero?
      Kernel.sleep(options[:delay]) if options[:delay]
      retry
    end
    raise
  end
end