2018-11-02 19:07:56 -04:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
module Bundler
|
|
|
|
# General purpose class for retrying code that may fail
|
|
|
|
class Retry
|
|
|
|
attr_accessor :name, :total_runs, :current_run
|
|
|
|
|
|
|
|
class << self
|
|
|
|
def default_attempts
|
|
|
|
default_retries + 1
|
|
|
|
end
|
|
|
|
alias_method :attempts, :default_attempts
|
|
|
|
|
|
|
|
def default_retries
|
|
|
|
Bundler.settings[:retry]
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def initialize(name, exceptions = nil, retries = self.class.default_retries)
|
|
|
|
@name = name
|
|
|
|
@retries = retries
|
|
|
|
@exceptions = Array(exceptions) || []
|
|
|
|
@total_runs = @retries + 1 # will run once, then upto attempts.times
|
|
|
|
end
|
|
|
|
|
|
|
|
def attempt(&block)
|
|
|
|
@current_run = 0
|
|
|
|
@failed = false
|
|
|
|
@error = nil
|
|
|
|
run(&block) while keep_trying?
|
|
|
|
@result
|
|
|
|
end
|
|
|
|
alias_method :attempts, :attempt
|
|
|
|
|
2020-10-15 00:20:25 -04:00
|
|
|
private
|
2018-11-02 19:07:56 -04:00
|
|
|
|
|
|
|
def run(&block)
|
2019-04-14 02:01:35 -04:00
|
|
|
@failed = false
|
2018-11-02 19:07:56 -04:00
|
|
|
@current_run += 1
|
|
|
|
@result = block.call
|
2019-04-14 02:01:35 -04:00
|
|
|
rescue StandardError => e
|
2018-11-02 19:07:56 -04:00
|
|
|
fail_attempt(e)
|
|
|
|
end
|
|
|
|
|
|
|
|
def fail_attempt(e)
|
|
|
|
@failed = true
|
|
|
|
if last_attempt? || @exceptions.any? {|k| e.is_a?(k) }
|
|
|
|
Bundler.ui.info "" unless Bundler.ui.debug?
|
|
|
|
raise e
|
|
|
|
end
|
|
|
|
return true unless name
|
|
|
|
Bundler.ui.info "" unless Bundler.ui.debug? # Add new line incase dots preceded this
|
|
|
|
Bundler.ui.warn "Retrying #{name} due to error (#{current_run.next}/#{total_runs}): #{e.class} #{e.message}", Bundler.ui.debug?
|
|
|
|
end
|
|
|
|
|
|
|
|
def keep_trying?
|
|
|
|
return true if current_run.zero?
|
|
|
|
return false if last_attempt?
|
|
|
|
return true if @failed
|
|
|
|
end
|
|
|
|
|
|
|
|
def last_attempt?
|
|
|
|
current_run >= total_runs
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|