Contents

require 'rest-core/event'

module RestCore
  class Retry
    def self.members; [:max_retries, :retry_exceptions]; end
    include Middleware

    DefaultRetryExceptions = [IOError, SystemCallError]

    def call env, &k
      if env[DRY]
        app.call(env, &k)
      else
        app.call(env){ |res| process(res, k) }
      end
    end

    def process res, k
      times = max_retries(res)
      return k.call(res) if times <= 0
      errors = retry_exceptions(res) || DefaultRetryExceptions

      if idx = res[FAIL].index{ |f| errors.find{ |e| f.kind_of?(e) } }
        err = res[FAIL].delete_at(idx)
        error_callback(res, err)
        env = res.merge('max_retries' => times - 1)
        event = Event::Retrying.new(nil, "(#{times}) for: #{err.inspect}")
        give_promise(call(log(env, event), &k))
      else
        k.call(res)
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
rest-core-4.0.1	lib/rest-core/middleware/retry.rb
rest-core-4.0.0	lib/rest-core/middleware/retry.rb