Contents

# frozen_string_literal: true

require 'dry/effects/provider'
require 'dry/effects/halt'

module Dry
  module Effects
    module Providers
      class Retry < Provider[:retry]
        include Dry::Equalizer(:scope, :limit, :attempts)

        param :scope

        attr_reader :attempts

        attr_reader :limit

        # Yield the block with the handler installed
        #
        # @api private
        def call(_, limit)
          @limit = limit
          @attempts = 0

          loop do
            begin
              return attempt { yield }
            rescue halt
            end
          end
        end

        def retry
          Instructions.Raise(halt.new)
        end

        def attempt
          if attempts_exhausted?
            nil
          else
            @attempts += 1
            yield
          end
        end

        def attempts_exhausted?
          attempts.equal?(limit)
        end

        def halt
          Halt[scope]
        end

        def provide?(effect)
          super && scope.equal?(effect.scope)
        end

        # @return [String]
        # @api public
        def represent
          "retry[#{scope} #{attempts}/#{limit}]"
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
dry-effects-0.1.1	lib/dry/effects/providers/retry.rb
dry-effects-0.1.0	lib/dry/effects/providers/retry.rb