Sha256: a6f86a03fe26e4464fc65f03dbaec6c68615de2a6e9aaff477fab452f40520fb

Contents?: true

Size: 1.71 KB

Versions: 7

Compression:

Stored size: 1.71 KB

Contents

# frozen_string_literal: true

module SidekiqUniqueJobs
  class Lock
    # Locks jobs while the job is executing in the server process
    # - Locks before yielding to the worker's perform method
    # - Unlocks after yielding to the worker's perform method
    #
    # See {#lock} for more information about the client.
    # See {#execute} for more information about the server
    #
    # @author Mikael Henriksson <mikael@zoolutions.se>
    class WhileExecuting < BaseLock
      RUN_SUFFIX ||= ":RUN"

      # @param [Hash] item the Sidekiq job hash
      # @param [Proc] callback callback to call after unlock
      # @param [Sidekiq::RedisConnection, ConnectionPool] redis_pool the redis connection
      #
      def initialize(item, callback, redis_pool = nil)
        super(item, callback, redis_pool)
        append_unique_key_suffix
      end

      # Simulate that a client lock was achieved.
      #   These locks should only ever be created in the server process.
      # @return [true] always returns true
      def lock
        true
      end

      # Executes in the Sidekiq server process.
      #   These jobs are locked in the server process not from the client
      # @yield to the worker class perform method
      def execute
        return strategy&.call unless locksmith.lock(item[LOCK_TIMEOUT_KEY])

        yield
        unlock_with_callback
      rescue Exception # rubocop:disable Lint/RescueException
        delete!
        raise
      end

      private

      # This is safe as the base_lock always creates a new digest
      #   The append there for needs to be done every time
      def append_unique_key_suffix
        item[UNIQUE_DIGEST_KEY] = item[UNIQUE_DIGEST_KEY] + RUN_SUFFIX
      end
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version Path
sidekiq-unique-jobs-6.0.25 lib/sidekiq_unique_jobs/lock/while_executing.rb
sidekiq-unique-jobs-6.0.24 lib/sidekiq_unique_jobs/lock/while_executing.rb
sidekiq-unique-jobs-6.0.23 lib/sidekiq_unique_jobs/lock/while_executing.rb
sidekiq-unique-jobs-6.0.22 lib/sidekiq_unique_jobs/lock/while_executing.rb
sidekiq-unique-jobs-6.0.21 lib/sidekiq_unique_jobs/lock/while_executing.rb
sidekiq-unique-jobs-6.0.20 lib/sidekiq_unique_jobs/lock/while_executing.rb
sidekiq-unique-jobs-6.0.19 lib/sidekiq_unique_jobs/lock/while_executing.rb