Sha256: 04acfa2b7a9f6a3adbc92fb3329d66d3f6386279762184dc6bc0257b60370500

Contents?: true

Size: 1.61 KB

Versions: 36

Compression:

Stored size: 1.61 KB

Contents

# 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

    private

    def run(&block)
      @failed = false
      @current_run += 1
      @result = block.call
    rescue StandardError => e
      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

Version data entries

36 entries across 36 versions & 2 rubygems

Version Path
rubygems-update-3.2.16 bundler/lib/bundler/retry.rb
bundler-2.2.16 lib/bundler/retry.rb
rubygems-update-3.2.15 bundler/lib/bundler/retry.rb
bundler-2.2.15 lib/bundler/retry.rb
rubygems-update-3.2.14 bundler/lib/bundler/retry.rb
bundler-2.2.14 lib/bundler/retry.rb
rubygems-update-3.2.13 bundler/lib/bundler/retry.rb
bundler-2.2.13 lib/bundler/retry.rb
rubygems-update-3.2.12 bundler/lib/bundler/retry.rb
bundler-2.2.12 lib/bundler/retry.rb
rubygems-update-3.2.11 bundler/lib/bundler/retry.rb
bundler-2.2.11 lib/bundler/retry.rb
rubygems-update-3.2.10 bundler/lib/bundler/retry.rb
bundler-2.2.10 lib/bundler/retry.rb
rubygems-update-3.2.9 bundler/lib/bundler/retry.rb
bundler-2.2.9 lib/bundler/retry.rb
rubygems-update-3.2.8 bundler/lib/bundler/retry.rb
bundler-2.2.8 lib/bundler/retry.rb
rubygems-update-3.2.7 bundler/lib/bundler/retry.rb
bundler-2.2.7 lib/bundler/retry.rb