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



275
276
277
# File 'lib/resque/plugins/retry.rb', line 275

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)


251
252
253
# File 'lib/resque/plugins/retry.rb', line 251

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.



452
453
454
455
456
# File 'lib/resque/plugins/retry.rb', line 452

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)



429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
# File 'lib/resque/plugins/retry.rb', line 429

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)


631
632
633
634
635
636
637
# File 'lib/resque/plugins/retry.rb', line 631

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



515
516
517
518
# File 'lib/resque/plugins/retry.rb', line 515

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.



417
418
419
420
421
# File 'lib/resque/plugins/retry.rb', line 417

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



598
599
600
601
602
603
604
# File 'lib/resque/plugins/retry.rb', line 598

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>)


574
575
576
# File 'lib/resque/plugins/retry.rb', line 574

def give_up_callbacks
  @give_up_callbacks ||= []
end

#ignore_exceptionsObject



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

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



501
502
503
504
505
506
507
508
509
510
# File 'lib/resque/plugins/retry.rb', line 501

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.



468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
# File 'lib/resque/plugins/retry.rb', line 468

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



181
182
183
184
185
186
187
188
189
190
191
# File 'lib/resque/plugins/retry.rb', line 181

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



200
201
202
# File 'lib/resque/plugins/retry.rb', line 200

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



356
357
358
359
360
361
362
# File 'lib/resque/plugins/retry.rb', line 356

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)


310
311
312
# File 'lib/resque/plugins/retry.rb', line 310

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)


370
371
372
373
374
375
# File 'lib/resque/plugins/retry.rb', line 370

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)


284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
# File 'lib/resque/plugins/retry.rb', line 284

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

Returns:

  • (Number)

    number of seconds to delay



143
144
145
146
147
148
149
150
151
# File 'lib/resque/plugins/retry.rb', line 143

def retry_delay(exception_class = nil)
  if @retry_exceptions.is_a?(Hash)
    delay = @retry_exceptions[exception_class] || 0
    # 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)


213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
# File 'lib/resque/plugins/retry.rb', line 213

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)


261
262
263
264
265
266
267
# File 'lib/resque/plugins/retry.rb', line 261

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



170
171
172
# File 'lib/resque/plugins/retry.rb', line 170

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)


319
320
321
322
323
324
325
326
327
# File 'lib/resque/plugins/retry.rb', line 319

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...)


612
613
614
615
616
# File 'lib/resque/plugins/retry.rb', line 612

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...)


563
564
565
566
567
# File 'lib/resque/plugins/retry.rb', line 563

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



159
160
161
# File 'lib/resque/plugins/retry.rb', line 159

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



380
381
382
383
384
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
# File 'lib/resque/plugins/retry.rb', line 380

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



549
550
551
552
553
554
555
# File 'lib/resque/plugins/retry.rb', line 549

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>)


525
526
527
# File 'lib/resque/plugins/retry.rb', line 525

def try_again_callbacks
  @try_again_callbacks ||= []
end