General purpose class for retrying code that may fail
# File lib/bundler/retry.rb, line 11 def initialize(name, exceptions = nil, attempts = nil) @name = name attempts ||= default_attempts @exceptions = Array(exceptions) || [] @total_runs = attempts.next # will run once, then upto attempts.times end
# File lib/bundler/retry.rb, line 23 def attempt(&block) @current_run = 0 @failed = false @error = nil while keep_trying? do run(&block) end @result end
# File lib/bundler/retry.rb, line 18 def default_attempts return Integer(self.class.attempts) if self.class.attempts DEFAULT_ATTEMPTS end
# File lib/bundler/retry.rb, line 43 def fail(e) @failed = true raise e if last_attempt? || @exceptions.any?{ |k| e.is_a?(k) } return true unless name Bundler.ui.warn "Retrying#{" #{name}" if name} due to error (#{current_run.next}/#{total_runs}): #{e.class} #{e.message}" end
# File lib/bundler/retry.rb, line 50 def keep_trying? return true if current_run.zero? return false if last_attempt? return true if @failed end
# File lib/bundler/retry.rb, line 56 def last_attempt? current_run >= total_runs end
# File lib/bundler/retry.rb, line 35 def run(&block) @failed = false @current_run += 1 @result = block.call rescue => e fail(e) end