Contents

require 'digest'
require 'sidekiq_unique_jobs/connectors'

module SidekiqUniqueJobs
  module Middleware
    module Server
      class UniqueJobs
        attr_reader :unlock_order, :redis_pool

        def call(worker, item, _queue, redis_pool = nil)
          @redis_pool = redis_pool

          decide_unlock_order(worker.class)
          lock_key = payload_hash(item)
          unlocked = before_yield? ? unlock(lock_key).inspect : 0

          yield
        ensure
          if after_yield? || !defined? unlocked || unlocked != 1
            unlock(lock_key)
          end
        end

        def decide_unlock_order(klass)
          @unlock_order = if unlock_order_configured?(klass)
                            klass.get_sidekiq_options['unique_unlock_order']
                          else
                            default_unlock_order
                          end
        end

        def unlock_order_configured?(klass)
          klass.respond_to?(:get_sidekiq_options) &&
            !klass.get_sidekiq_options['unique_unlock_order'].nil?
        end

        def default_unlock_order
          SidekiqUniqueJobs.config.default_unlock_order
        end

        def before_yield?
          unlock_order == :before_yield
        end

        def after_yield?
          unlock_order == :after_yield
        end

        protected

        def payload_hash(item)
          SidekiqUniqueJobs::PayloadHelper.get_payload(item['class'], item['queue'], item['args'])
        end

        def unlock(payload_hash)
          conn.del(payload_hash)
        end

        def logger
          Sidekiq.logger
        end

        def conn
          SidekiqUniqueJobs::Connectors.conn(redis_pool)
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
sidekiq-unique-jobs-3.0.10	lib/sidekiq_unique_jobs/middleware/server/unique_jobs.rb
sidekiq-unique-jobs-3.0.9	lib/sidekiq_unique_jobs/middleware/server/unique_jobs.rb
sidekiq-unique-jobs-3.0.8	lib/sidekiq_unique_jobs/middleware/server/unique_jobs.rb
sidekiq-unique-jobs-3.0.7	lib/sidekiq_unique_jobs/middleware/server/unique_jobs.rb
sidekiq-unique-jobs-3.0.5	lib/sidekiq_unique_jobs/middleware/server/unique_jobs.rb