Module: ExceptionHelper::Retry::ClassMethods

Defined in:
lib/exception_helper/retry.rb

Instance Method Summary collapse

Instance Method Details

#retry_on_failure(*exception_list, &block) ⇒ Object

execute the given block, retrying only when one of the given exceptions is raised



29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 
# File 'lib/exception_helper/retry.rb', line 29

def retry_on_failure(*exception_list, &block)
  opts = exception_list.last.is_a?(Hash) ? exception_list.pop : {}
  opts = {:retry_count => 3}.merge(opts)
  retry_count = opts[:retry_count]
  begin
    yield block
  rescue *exception_list => e
    if retry_count > 0
      retry_count -= 1
      logger.info "Exception, trying again #{retry_count} more times"
      sleep opts[:retry_sleep].to_f if opts[:retry_sleep]
      retry
    else
      logger.error "Too many exceptions...re-raising"
      raise
    end
  end
end

#wrap_with_retry(*methods) ⇒ Object

Wraps a method with ‘retry_on_failure` to handle exceptions. It does this via alias_method_chain, so the original method name will be preserved and the method will simply be decorated with the retry logic.



50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/exception_helper/retry.rb', line 50

def wrap_with_retry(*methods)
  opts = methods.last.is_a?(Hash) ? methods.pop : {}
  exception_list = opts.delete(:exceptions)
  methods.each do |method|
    # Extract the punctuation character from the method name if one exists.
    aliased_method, punctuation = method.to_s.sub(/([?!=])$/, ''), $1

    # Create the decorated method that `alias_method_chain` will look for.
    define_method("#{aliased_method}_with_retry#{punctuation}") do |*args|
      retry_on_failure(*exception_list, opts) do
        send("#{aliased_method}_without_retry#{punctuation}", *args)
      end
    end

    # Set up the method chain.
    alias_method_chain method, :retry
  end
end