retry_upto.rb

retry_upto.rb

# Ruby's `retry` with steroids:
#
# - retry up to 5 times without waiting between them and retrying after any exception
#
#     retry_upto(5) do ... end
#
# - retry up to 5 times, waiting 2 seconds between retries and retrying after any exception
#
#     retry_upto(5, :wait => 2) do ... end
#
# - retry up to 5 times without waiting between retries, retrying only after a ZeroDivisionError
#
#     retry_upto(5, :rescue_only => ZeroDivisionError) do ... end
#
# - retry up to 5 times, waiting 2 seconds between retries, retrying only after a ZeroDivisionError
#
#     retry_upto(5, :wait => 2, :rescue_only => ZeroDivisionError) do ... end

def retry_upto(max_retries = 1, options = {}, &blk)
  yield
rescue (options[:rescue_only] || Exception)
  sleep(options[:wait] || 0)
  retry_upto(max_retries - 1, options, &blk) if max_retries > 1
end