Contents

require 'pathname'
require 'digest/sha1'
require 'concurrent/map'
require 'sidekiq_unique_jobs/scripts/acquire_lock'
require 'sidekiq_unique_jobs/scripts/release_lock'

module SidekiqUniqueJobs
  ScriptError         = Class.new(StandardError)
  UniqueKeyMissing    = Class.new(ArgumentError)
  JidMissing          = Class.new(ArgumentError)
  MaxLockTimeMissing  = Class.new(ArgumentError)
  UnexpectedValue     = Class.new(StandardError)

  module Scripts
    LUA_PATHNAME ||= Pathname.new(__FILE__).dirname.join('../../redis').freeze
    SOURCE_FILES ||= Dir[LUA_PATHNAME.join('**/*.lua')].compact.freeze
    DEFINED_METHODS ||= [].freeze
    SCRIPT_SHAS ||= Concurrent::Map.new

    module_function

    extend SingleForwardable
    def_delegators :SidekiqUniqueJobs, :connection, :logger

    def call(file_name, redis_pool, options = {}) # rubocop:disable MethodLength
      connection(redis_pool) do |redis|
        if SCRIPT_SHAS[file_name].nil?
          SCRIPT_SHAS[file_name] = redis.script(:load, script_source(file_name))
        end
        redis.evalsha(SCRIPT_SHAS[file_name], options)
      end
    rescue Redis::CommandError => ex
      if ex.message == 'NOSCRIPT No matching script. Please use EVAL.'
        SCRIPT_SHAS.delete(file_name)
        call(file_name, redis_pool, options)
        raise
      else
        raise ScriptError,
              "#{file_name}.lua\n\n#{ex.message}\n\n#{script_source(file_name)}" \
              "#{ex.backtrace.join("\n")}"
      end
    end

    def script_source(file_name)
      script_path(file_name).read
    end

    def script_path(file_name)
      LUA_PATHNAME.join("#{file_name}.lua")
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
sidekiq-unique-jobs-5.0.2	lib/sidekiq_unique_jobs/scripts.rb
sidekiq-unique-jobs-5.0.1	lib/sidekiq_unique_jobs/scripts.rb