module ActiveUtils::NetworkConnectionRetries

def self.included(base)
  base.send(:attr_accessor, :retry_safe)
end

def self.log(logger, level, message, tag=nil)
  tag ||= self.class.to_s
  message = "[#{tag}] #{message}"
  logger.send(level, message) if logger
end

def derived_error_message(errors, klass)
  key = (errors.keys & klass.ancestors).first
  key ? errors[key] : nil
end

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

def retry_exceptions(options={})
  connection_errors = DEFAULT_CONNECTION_ERRORS.merge(options[:connection_exceptions] || {})
  retry_network_exceptions(options) do
    begin
      yield
    rescue Errno::ECONNREFUSED => e
      raise ActiveUtils::RetriableConnectionError, "The remote server refused the connection"
    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

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