Module: Resque::Plugins::Retry

Includes:
Logging
Included in:
ExponentialBackoff
Defined in:
lib/resque/plugins/retry.rb,
lib/resque/plugins/retry/logging.rb

Overview

If you want your job to retry on failure, simply extend your module/class with this module:

class DeliverWebHook
  extend Resque::Plugins::Retry # allows 1 retry by default.
  @queue = :web_hooks

  def self.perform(url, hook_id, hmac_key)
    heavy_lifting
  end
end

Easily do something custom:

class DeliverWebHook
  extend Resque::Plugins::Retry
  @queue = :web_hooks

  @retry_limit = 8  # default: 1
  @retry_delay = 60 # default: 0

  # used to build redis key, for counting job attempts.
  def self.retry_identifier(url, hook_id, hmac_key)
    "#{url}-#{hook_id}"
  end

  def self.perform(url, hook_id, hmac_key)
    heavy_lifting
  end
end

Defined Under Namespace

Modules: Logging Classes: AmbiguousRetryStrategyException, RetryConfigurationException

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Methods included from Logging

#log_message

Instance Attribute Details

#expire_retry_key_afterNumber (readonly)

This method is abstract.

The number of seconds to set the TTL to on the resque-retry key in redis

Returns:

  • (Number)

    number of seconds



280
281
282
# File 'lib/resque/plugins/retry.rb', line 280

def expire_retry_key_after
  @expire_retry_key_after
end

#fatal_exceptionsArray? (readonly)

This method is abstract.

Controls what exceptions may not be retried

Default: `nil` - this will retry all exceptions.

Returns:

  • (Array, nil)


256
257
258
# File 'lib/resque/plugins/retry.rb', line 256

def fatal_exceptions
  @fatal_exceptions
end

Class Method Details

.extended(receiver) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Fail fast, when extended, if the “receiver” is misconfigured



54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
# File 'lib/resque/plugins/retry.rb', line 54

def self.extended(receiver)
  retry_exceptions = receiver.instance_variable_get('@retry_exceptions')
  fatal_exceptions = receiver.instance_variable_get('@fatal_exceptions')
  ignore_exceptions = receiver.instance_variable_get('@ignore_exceptions')

  if fatal_exceptions && retry_exceptions
    raise AmbiguousRetryStrategyException.new(%{You can't define both "@fatal_exceptions" and "@retry_exceptions"})
  end

  # Check that ignore_exceptions is a subset of retry_exceptions
  if retry_exceptions.is_a?(Hash)
    exceptions = retry_exceptions.keys
  else
    exceptions = Array(retry_exceptions)
  end

  excess_exceptions = Array(ignore_exceptions) - exceptions
  unless excess_exceptions.empty?
    raise RetryConfigurationException, "The following exceptions are defined in @ignore_exceptions but not in @retry_exceptions: #{excess_exceptions.join(', ')}."
  end
end

Instance Method Details

#after_perform_retry(*args) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Resque after_perform hook

Deletes retry attempt count from Redis.



457
458
459
460
461
# File 'lib/resque/plugins/retry.rb', line 457

def after_perform_retry(*args)
  return if Resque.inline?
  log_message 'after_perform_retry, clearing retry key', args
  clean_retry_key(*args)
end

#before_perform_retry(*args) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Resque before_perform hook

Increments `@retry_attempt` count and updates the “retry_key” expiration time (if applicable)



434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
# File 'lib/resque/plugins/retry.rb', line 434

def before_perform_retry(*args)
  return if Resque.inline?
  log_message 'before_perform_retry', args
  @on_failure_retry_hook_already_called = false

  # store number of retry attempts.
  retry_key = redis_retry_key(*args)
  Resque.redis.setnx(retry_key, -1)
  @retry_attempt = Resque.redis.incr(retry_key)
  log_message "attempt: #{@retry_attempt} set in Redis", args

  # set/update the "retry_key" expiration
  if expire_retry_key_after
    log_message "updating expiration for retry key: #{retry_key}", args
    Resque.redis.expire(retry_key, retry_delay + expire_retry_key_after)
  end
end

#call_symbol_or_block(method, *args) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Helper to call functions that may be passed as Symbols or Procs. If a symbol, it is assumed to refer to a method that is already defined on this class.

Parameters:

  • method (Symbol|Proc)
  • *args (Object...)

Returns:

  • (Object)


636
637
638
639
640
641
642
# File 'lib/resque/plugins/retry.rb', line 636

def call_symbol_or_block(method, *args)
  if method.is_a?(Symbol)
    send(method, *args)
  elsif method.respond_to?(:call)
    instance_exec(*args, &method)
  end
end

#clean_retry_key(*args) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Clean up retry state from redis once done



520
521
522
523
# File 'lib/resque/plugins/retry.rb', line 520

def clean_retry_key(*args)
  log_message 'clean_retry_key', args
  Resque.redis.del(redis_retry_key(*args))
end

#give_up(exception, *args) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

We failed and we're not retrying.



422
423
424
425
426
# File 'lib/resque/plugins/retry.rb', line 422

def give_up(exception, *args)
  log_message 'retry criteria not sufficient for retry', args, exception
  run_give_up_callbacks(exception, *args)
  clean_retry_key(*args)
end

#give_up_callback(method = nil) {|exception, *args| ... } ⇒ Object

Register a give up callback that will be called when the job fails and is not retrying. Can be registered with a block or a symbol.

Examples:

Registering a callback with a block


give_up_callback do |exception, *args|
  logger.error(
    "Resque job received exception #{exception} and is giving up")
end

Registering a callback with a Symbol


give_up_callback :my_callback

Parameters:

  • method (Symbol?) (defaults to: nil)

Yields:

  • (exception, *args)

Yield Parameters:

  • exception (Exception)

    the exception that was raised

  • args (Array)

    job arguments



603
604
605
606
607
608
609
# File 'lib/resque/plugins/retry.rb', line 603

def give_up_callback(method=nil, &block)
  if method.is_a? Symbol
    give_up_callbacks << method
  elsif block_given?
    give_up_callbacks << block
  end
end

#give_up_callbacksArray<Proc>

Returns the give up callbacks.

Returns:

  • (Array<Proc>)


579
580
581
# File 'lib/resque/plugins/retry.rb', line 579

def give_up_callbacks
  @give_up_callbacks ||= []
end

#ignore_exceptionsObject



623
624
625
# File 'lib/resque/plugins/retry.rb', line 623

def ignore_exceptions
  @ignore_exceptions ||= []
end

#inherited(subclass) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Copy retry criteria checks, try again callbacks, and give up callbacks on inheritance.



80
81
82
83
84
85
# File 'lib/resque/plugins/retry.rb', line 80

def inherited(subclass)
  super(subclass)
  subclass.instance_variable_set('@retry_criteria_checks', retry_criteria_checks.dup)
  subclass.instance_variable_set('@try_again_callbacks', try_again_callbacks.dup)
  subclass.instance_variable_set('@give_up_callbacks', give_up_callbacks.dup)
end

#instance_exec(*args, &block) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Used to perform retry criteria check blocks under the job instance's context

Returns:

  • (Object)

    return value of the criteria check



506
507
508
509
510
511
512
513
514
515
# File 'lib/resque/plugins/retry.rb', line 506

def instance_exec(*args, &block)
  mname = "__instance_exec_#{Thread.current.object_id.abs}"
  class << self; self end.class_eval{ define_method(mname, &block) }
  begin
    ret = send(mname, *args)
  ensure
    class << self; self end.class_eval{ undef_method(mname) } rescue nil
  end
  ret
end

#on_failure_retry(exception, *args) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Note:

This hook will only allow execution once per job perform attempt. This was added because Resque v1.20.0 calls the hook twice. IMO; this isn't something resque-retry should have to worry about!

Resque on_failure hook

Checks if our retry criteria is valid, if it is we try again. Otherwise the retry attempt count is deleted from Redis.



473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
# File 'lib/resque/plugins/retry.rb', line 473

def on_failure_retry(exception, *args)
  return if Resque.inline?
  log_message 'on_failure_retry', args, exception
  if exception.is_a?(Resque::DirtyExit)
    # This hook is called from a worker processes, not the job process
    # that failed with a DirtyExit, so @retry_attempt wasn't set yet
    @retry_attempt = Resque.redis.get(redis_retry_key(*args)).to_i
  elsif @on_failure_retry_hook_already_called
    log_message 'on_failure_retry_hook_already_called', args, exception
    return
  end

  # If we are "ignoring" the exception, then we decrement the retry
  # counter, so that the current attempt didn't count toward the retry
  # counter.
  if ignore_exceptions.include?(exception.class)
    @retry_attempt = Resque.redis.decr(redis_retry_key(*args))
  end

  if retry_criteria_valid?(exception, *args)
    try_again(exception, *args)
  else
    give_up(exception, *args)
  end

  @on_failure_retry_hook_already_called = true
end

#redis_retry_key(*args) ⇒ String

Builds the redis key to be used for keeping state of the job attempts.

Returns:

  • (String)

    redis key



109
110
111
# File 'lib/resque/plugins/retry.rb', line 109

def redis_retry_key(*args)
  ['resque-retry', name, retry_identifier(*args)].compact.join(':').gsub(/\s/, '')
end

#retry_args(*args) ⇒ Array

This method is abstract.

Modify the arguments used to retry the job. Use this to do something other than try the exact same job again

Returns:

  • (Array)

    new job arguments



186
187
188
189
190
191
192
193
194
195
196
# File 'lib/resque/plugins/retry.rb', line 186

def retry_args(*args)
  # Here for backwards compatibility. If an "args_for_retry" method exists
  # invoke it, but warn that it is deprecated (and will be removed in a
  # future revision)
  if respond_to?(:args_for_retry)
    warn "`Resque::Plugins::Retry#args_for_retry` is deprecated, please use `Resque::Plugins::Retry#retry_args` instead."
    args_for_retry(*args)
  else
    args
  end
end

#retry_args_for_exception(exception, *args) ⇒ Array

This method is abstract.

Modify the arguments used to retry the job based on the exception. Use this to do something other than try the exact same job again.

Returns:

  • (Array)

    new job arguments



205
206
207
# File 'lib/resque/plugins/retry.rb', line 205

def retry_args_for_exception(exception, *args)
  retry_args(*args)
end

#retry_attemptFixnum

Number of retry attempts used to try and perform the job

The real value is kept in Redis, it is accessed and incremented using a before_perform hook.

Returns:

  • (Fixnum)

    number of attempts



133
134
135
# File 'lib/resque/plugins/retry.rb', line 133

def retry_attempt
  @retry_attempt ||= 0
end

#retry_criteria_check(method = nil) {|exception, *args| ... } ⇒ Object

Register a retry criteria check callback to be run before retrying the job again. Can be registered with a block or a symbol.

If any callback returns `true`, the job will be retried.

Examples:

Registering a custom retry criteria check.


retry_criteria_check do |exception, *args|
  if exception.message =~ /InvalidJobId/
    # don't retry if we got passed a invalid job id.
    false
  else
    true
  end
end

retry_criteria_check :my_check

Parameters:

  • method (Symbol?) (defaults to: nil)

Yields:

  • (exception, *args)

Yield Parameters:

  • exception (Exception)

    the exception that was raised

  • args (Array)

    job arguments

Yield Returns:

  • (Boolean)

    false == dont retry, true == can retry



361
362
363
364
365
366
367
# File 'lib/resque/plugins/retry.rb', line 361

def retry_criteria_check(method=nil, &block)
  if method.is_a? Symbol
    retry_criteria_checks << method
  elsif block_given?
    retry_criteria_checks << block
  end
end

#retry_criteria_checksArray

Retry criteria checks

Returns:

  • (Array)


315
316
317
# File 'lib/resque/plugins/retry.rb', line 315

def retry_criteria_checks
  @retry_criteria_checks ||= []
end

#retry_criteria_checks_pass?(exception, *args) ⇒ Boolean

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Returns true if any of the retry criteria checks pass. When a retry criteria check passes, the remaining ones are not executed.

Returns:

  • (Boolean)


375
376
377
378
379
380
# File 'lib/resque/plugins/retry.rb', line 375

def retry_criteria_checks_pass?(exception, *args)
  retry_criteria_checks.each do |criteria_check|
    return true if !!call_symbol_or_block(criteria_check, exception, *args)
  end
  false
end

#retry_criteria_valid?(exception, *args) ⇒ Boolean

Test if the retry criteria is valid

Parameters:

  • exception (Exception)
  • args (Array)

    job arguments

Returns:

  • (Boolean)


289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
# File 'lib/resque/plugins/retry.rb', line 289

def retry_criteria_valid?(exception, *args)
  # if the retry limit was reached, dont bother checking anything else.
  if retry_limit_reached?
    log_message 'retry limit reached', args, exception
    return false
  end

  # We always want to retry if the exception matches.
  retry_based_on_exception = retry_exception?(exception)
  log_message "Exception is #{retry_based_on_exception ? '' : 'not '}sufficient for a retry", args, exception

  retry_based_on_criteria = false
  unless retry_based_on_exception
    # call user retry criteria check blocks.
    retry_based_on_criteria = retry_criteria_checks_pass?(exception, *args)
    log_message "user retry criteria is #{retry_based_on_criteria ? '' : 'not '}sufficient for a retry", args, exception
  end

  retry_based_on_exception || retry_based_on_criteria
end

#retry_delay(exception_class = nil) ⇒ Number

This method is abstract.

Number of seconds to delay until the job is retried If @retry_exceptions is a Hash and there is no delay defined for exception_class, looks for closest superclass and assigns it's delay to @retry_exceptions

Returns:

  • (Number)

    number of seconds to delay



145
146
147
148
149
150
151
152
153
154
155
156
# File 'lib/resque/plugins/retry.rb', line 145

def retry_delay(exception_class = nil)
  if @retry_exceptions.is_a?(Hash)
    delay = @retry_exceptions[exception_class] ||= begin
      relevant_definitions = @retry_exceptions.select { |ex| exception_class <= ex }
      relevant_definitions.any? ? relevant_definitions.sort.first[1] : 0
    end
    # allow an array of delays.
    delay.is_a?(Array) ? delay[retry_attempt] || delay.last : delay
  else
    @retry_delay ||= 0
  end
end

#retry_exception?(exception) ⇒ Boolean

Convenience method to test whether you may retry on a given exception

also be a Class

Parameters:

  • an (Exception)

    instance of Exception. Deprecated: can

Returns:

  • (Boolean)


218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
# File 'lib/resque/plugins/retry.rb', line 218

def retry_exception?(exception)
  # If both "fatal_exceptions" and "retry_exceptions" are undefined we are
  # done (we should retry the exception)
  #
  # It is intentional that we check "retry_exceptions" first since it is
  # more likely that it will be defined (over "fatal_exceptions") as it
  # has been part of the API for quite a while
  return true if retry_exceptions.nil? && fatal_exceptions.nil?

  # If "fatal_exceptions" is undefined interrogate "retry_exceptions"
  if fatal_exceptions.nil?
    retry_exceptions.any? do |ex|
      if exception.is_a?(Class)
        ex >= exception
      else
        ex === exception
      end
    end
  # It is safe to assume we need to check "fatal_exceptions" at this point
  else
    fatal_exceptions.none? do |ex|
      if exception.is_a?(Class)
        ex >= exception
      else
        ex === exception
      end
    end
  end
end

#retry_exceptionsArray?

This method is abstract.

Controls what exceptions may be retried

Default: `nil` - this will retry all exceptions.

Returns:

  • (Array, nil)


266
267
268
269
270
271
272
# File 'lib/resque/plugins/retry.rb', line 266

def retry_exceptions
  if @retry_exceptions.is_a?(Hash)
    @retry_exceptions.keys
  else
    @retry_exceptions ||= nil
  end
end

#retry_identifier(*args) ⇒ String

This method is abstract.

You may override to implement a custom retry identifier, you should consider doing this if your job arguments are many/long or may not cleanly convert to strings.

Builds a retry identifier using the job arguments. This identifier is used as part of the redis key

Parameters:

  • args (Array)

    job arguments

Returns:

  • (String)

    job identifier



98
99
100
101
# File 'lib/resque/plugins/retry.rb', line 98

def retry_identifier(*args)
  args_string = args.join('-')
  args_string.empty? ? nil : Digest::SHA1.hexdigest(args_string)
end

#retry_job_delegateObject?

This method is abstract.

Specify another resque job (module or class) to delegate retry duties to upon failure

Returns:

  • (Object, nil)

    class or module if delegate on failure, otherwise nil



175
176
177
# File 'lib/resque/plugins/retry.rb', line 175

def retry_job_delegate
  @retry_job_delegate ||= nil
end

#retry_limitFixnum

Maximum number of retrys we can attempt to successfully perform the job

A retry limit of 0 will never retry. A retry limit of -1 or below will retry forever.

Returns:

  • (Fixnum)


121
122
123
# File 'lib/resque/plugins/retry.rb', line 121

def retry_limit
  @retry_limit ||= 1
end

#retry_limit_reached?Boolean

Test if the retry limit has been reached

Returns:

  • (Boolean)


324
325
326
327
328
329
330
331
332
# File 'lib/resque/plugins/retry.rb', line 324

def retry_limit_reached?
  if retry_limit == 0
    true
  elsif retry_limit > 0
    true if retry_attempt >= retry_limit
  else
    false
  end
end

#run_give_up_callbacks(exception, *args) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Runs all the give up callbacks.

Parameters:

  • exception (Exception)
  • args (Object...)


617
618
619
620
621
# File 'lib/resque/plugins/retry.rb', line 617

def run_give_up_callbacks(exception, *args)
  give_up_callbacks.each do |callback|
    call_symbol_or_block(callback, exception, *args)
  end
end

#run_try_again_callbacks(exception, *args) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Runs all the try again callbacks.

Parameters:

  • exception (Exception)
  • args (Object...)


568
569
570
571
572
# File 'lib/resque/plugins/retry.rb', line 568

def run_try_again_callbacks(exception, *args)
  try_again_callbacks.each do |callback|
    call_symbol_or_block(callback, exception, *args)
  end
end

#sleep_after_requeueNumber

This method is abstract.

Number of seconds to sleep after job is requeued

Returns:

  • (Number)

    number of seconds to sleep



164
165
166
# File 'lib/resque/plugins/retry.rb', line 164

def sleep_after_requeue
  @sleep_after_requeue ||= 0
end

#try_again(exception, *args) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Retries the job



385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
# File 'lib/resque/plugins/retry.rb', line 385

def try_again(exception, *args)
  log_message 'try_again', args, exception
  run_try_again_callbacks(exception, *args)

  # some plugins define retry_delay and have it take no arguments, so rather than break those,
  # we'll just check here to see whether it takes the additional exception class argument or not
  temp_retry_delay = ([-1, 1].include?(method(:retry_delay).arity) ? retry_delay(exception.class) : retry_delay)

  retry_in_queue = retry_job_delegate ? retry_job_delegate : self
  log_message "retry delay: #{temp_retry_delay} for class: #{retry_in_queue}", args, exception

  # remember that this job is now being retried. before_perform_retry will increment
  # this so it represents the retry count, and MultipleWithRetrySuppression uses
  # the existence of this to determine if the job should be sent to the
  # parent failure backend (e.g. failed queue) or not.  Removing this means
  # jobs that fail before ::perform will be both retried and sent to the failed queue.
  Resque.redis.setnx(redis_retry_key(*args), -1)

  retry_args = retry_args_for_exception(exception, *args)

  if temp_retry_delay <= 0
    # If the delay is 0, no point passing it through the scheduler
    Resque.enqueue(retry_in_queue, *retry_args)
  else
    Resque.enqueue_in(temp_retry_delay, retry_in_queue, *retry_args)
  end

  # remove retry key from redis if we handed retry off to another queue.
  clean_retry_key(*args) if retry_job_delegate

  # sleep after requeue if enabled.
  sleep(sleep_after_requeue) if sleep_after_requeue > 0
end

#try_again_callback(method = nil) {|exception, *args| ... } ⇒ Object

Register a try again callback that will be called when the job fails but is trying again. Can be registered with a block or a symbol.

Examples:

Registering a callback with a block


try_again_callback do |exception, *args|
  logger.error(
    "Resque job received exception #{exception} and is trying again")
end

Registering a callback with a Symbol


try_again_callback :my_callback

Parameters:

  • method (Symbol?) (defaults to: nil)

Yields:

  • (exception, *args)

Yield Parameters:

  • exception (Exception)

    the exception that was raised

  • args (Array)

    job arguments



554
555
556
557
558
559
560
# File 'lib/resque/plugins/retry.rb', line 554

def try_again_callback(method=nil, &block)
  if method.is_a? Symbol
    try_again_callbacks << method
  elsif block_given?
    try_again_callbacks << block
  end
end

#try_again_callbacksArray<Proc>

Returns the try again callbacks.

Returns:

  • (Array<Proc>)


530
531
532
# File 'lib/resque/plugins/retry.rb', line 530

def try_again_callbacks
  @try_again_callbacks ||= []
end