Ruby - retry function with different params

user_7123913 0
Viewed 492

I'd like to retry a function with different params depending on the result of the first iteration:

Giving a retry function like follow:

 def retry_on_fail(**args)
   yield
 rescue StandardError => e
    args = args.merge(different_param => true) if e.class == `specific_error`
    retry

Is there a way to do so? I didn't find it yet... Thanks!

retry-logic error-handling ruby-on-rails ruby
2 Answers

You can yield however many times you want in a method and the trick is really passing the arguments to the block:

# given
class SpecificError < StandardError; end

def retry_on_fail(**args)
  yield(args)
rescue SpecificError 
  yield(args.merge(different_param: true))
end

retry_on_fail do |args|
  raise SpecificError if args.empty?
  args
end
# returns { different_param: true }

There is also a slight differnce here flow wise - retry runs the whole method from the top and this will just call the block again. If thats what you want you could do:

def retry_on_fail(**args)
  yield(args)
rescue SpecificError 
  args.merge!(different_param: true) 
  retry
end

But this has the potential to create an endless loop if the block raises the same exception again.

user_5448250

Try this

def retry_on_fail(**args)
  rescue_args = args 
  begin
    yield(rescue_args)
  rescue StandardError => e
    rescue_args = rescue_args.merge(different_param => true) if e.class == `specific_error`
    retry
  end  
end
user_3865400
Related