Browse Source

Do not execute the job with the same arguments as the retry job (#4814)

pull/4/head
abcang 6 years ago
committed by Eugen Rochko
parent
commit
be7ffa2d75
4 changed files with 25 additions and 1 deletions
  1. +1
    -1
      app/workers/pubsubhubbub/subscribe_worker.rb
  2. +1
    -0
      config/application.rb
  3. +3
    -0
      config/initializers/sidekiq.rb
  4. +20
    -0
      lib/mastodon/unique_retry_job_middleware.rb

+ 1
- 1
app/workers/pubsubhubbub/subscribe_worker.rb View File

@ -3,7 +3,7 @@
class Pubsubhubbub::SubscribeWorker
include Sidekiq::Worker
sidekiq_options queue: 'push', retry: 10, unique: :until_executed, dead: false
sidekiq_options queue: 'push', retry: 10, unique: :until_executed, dead: false, unique_retry: true
sidekiq_retry_in do |count|
case count

+ 1
- 0
config/application.rb View File

@ -10,6 +10,7 @@ require_relative '../app/lib/exceptions'
require_relative '../lib/paperclip/gif_transcoder'
require_relative '../lib/paperclip/video_transcoder'
require_relative '../lib/mastodon/version'
require_relative '../lib/mastodon/unique_retry_job_middleware'
Dotenv::Railtie.load

+ 3
- 0
config/initializers/sidekiq.rb View File

@ -13,4 +13,7 @@ end
Sidekiq.configure_client do |config|
config.redis = redis_params
config.client_middleware do |chain|
chain.add Mastodon::UniqueRetryJobMiddleware
end
end

+ 20
- 0
lib/mastodon/unique_retry_job_middleware.rb View File

@ -0,0 +1,20 @@
# frozen_string_literal: true
class Mastodon::UniqueRetryJobMiddleware
def call(_worker_class, item, _queue, _redis_pool)
return if item['unique_retry'] && retried?(item)
yield
end
private
def retried?(item)
# Use unique digest key of SidekiqUniqueJobs
unique_key = SidekiqUniqueJobs::UNIQUE_DIGEST_KEY
unique_digest = item[unique_key]
class_name = item['class']
retries = Sidekiq::RetrySet.new
retries.any? { |job| job.item['class'] == class_name && job.item[unique_key] == unique_digest }
end
end

Loading…
Cancel
Save