Contents

require 'digest'

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

          set_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 set_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)
          if redis_pool
            redis_pool.with { |conn| conn.del(payload_hash) }
          else
            Sidekiq.redis { |conn| conn.del(payload_hash) }
          end
        end

        def logger
          Sidekiq.logger
        end
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
sidekiq-unique-jobs-3.0.2	lib/sidekiq-unique-jobs/middleware/server/unique_jobs.rb
sidekiq-unique-jobs-3.0.1	lib/sidekiq-unique-jobs/middleware/server/unique_jobs.rb
sidekiq-unique-jobs-3.0.0	lib/sidekiq-unique-jobs/middleware/server/unique_jobs.rb