sidekiq-unique-jobs 6.0.24 → 7.0.4

data/lib/sidekiq_unique_jobs/redis.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module SidekiqUniqueJobs
+  #
+  # Augments and enhances redis with object oriented methods
+  #
+  # @author Mikael Henriksson <mikael@mhenrixon.com>
+  #
+  module Redis
+  end
+end

data/lib/sidekiq_unique_jobs/redis/entity.rb

@@ -0,0 +1,106 @@
+# frozen_string_literal: true
+
+module SidekiqUniqueJobs
+  module Redis
+    #
+    # Class Entity functions as a base class for redis types
+    #
+    # @author Mikael Henriksson <mikael@mhenrixon.com>
+    #
+    class Entity
+      # includes "SidekiqUniqueJobs::Logging"
+      # @!parse include SidekiqUniqueJobs::Logging
+      include SidekiqUniqueJobs::Logging
+
+      # includes "SidekiqUniqueJobs::Script::Caller"
+      # @!parse include SidekiqUniqueJobs::Script::Caller
+      include SidekiqUniqueJobs::Script::Caller
+
+      # includes "SidekiqUniqueJobs::JSON"
+      # @!parse include SidekiqUniqueJobs::JSON
+      include SidekiqUniqueJobs::JSON
+
+      # includes "SidekiqUniqueJobs::Timing"
+      # @!parse include SidekiqUniqueJobs::Timing
+      include SidekiqUniqueJobs::Timing
+
+      #
+      # @!attribute [r] key
+      #   @return [String] the redis key for this entity
+      attr_reader :key
+
+      #
+      # Initialize a new Entity
+      #
+      # @param [String] key the redis key for this entity
+      #
+      def initialize(key)
+        @key = key
+      end
+
+      #
+      # Checks if the key for this entity exists in redis
+      #
+      #
+      # @return [true] when exists
+      # @return [false] when not exists
+      #
+      def exist?
+        redis do |conn|
+          value = conn.exists(key)
+
+          return value if boolean?(value)
+
+          value.to_i.positive?
+        end
+      end
+
+      #
+      # The number of microseconds until the key expires
+      #
+      #
+      # @return [Integer] expiration in milliseconds
+      #
+      def pttl
+        redis { |conn| conn.pttl(key) }
+      end
+
+      #
+      # The number of seconds until the key expires
+      #
+      #
+      # @return [Integer] expiration in seconds
+      #
+      def ttl
+        redis { |conn| conn.ttl(key) }
+      end
+
+      #
+      # Check if the entity has expiration
+      #
+      #
+      # @return [true] when entity is set to exire
+      # @return [false] when entity isn't expiring
+      #
+      def expires?
+        pttl.positive? || ttl.positive?
+      end
+
+      #
+      # Returns the number of entries in this entity
+      #
+      #
+      # @return [Integer] 0
+      #
+      def count
+        0
+      end
+
+      private
+
+      def boolean?(value)
+        [TrueClass, FalseClass].any? { |klazz| value.is_a?(klazz) }
+      end
+    end
+  end
+end

data/lib/sidekiq_unique_jobs/redis/hash.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+module SidekiqUniqueJobs
+  module Redis
+    #
+    # Class Hash provides convenient access to redis hashes
+    #
+    # @author Mikael Henriksson <mikael@mhenrixon.com>
+    #
+    class Hash < Entity
+      #
+      # Return entries for this hash
+      #
+      # @param [true,false] with_values false return hash
+      #
+      # @return [Array<Object>] when given with_values: false
+      # @return [Hash<String, String>] when given with_values: true
+      #
+      def entries(with_values: false)
+        if with_values
+          redis { |conn| conn.hgetall(key) }
+        else
+          redis { |conn| conn.hkeys(key) }
+        end
+      end
+
+      #
+      # Removes the key from redis
+      #
+      def del(*fields)
+        redis { |conn| conn.hdel(key, *fields) }
+      end
+
+      #
+      # Get a members value
+      #
+      # @param [String] member the member who's value to get
+      #
+      # @return [Object] whatever is stored on this hash member
+      #
+      def [](member)
+        redis { |conn| conn.hget(key, member) }
+      end
+
+      #
+      # Returns the count for this hash
+      #
+      #
+      # @return [Integer] the length of this hash
+      #
+      def count
+        redis { |conn| conn.hlen(key) }
+      end
+    end
+  end
+end

data/lib/sidekiq_unique_jobs/redis/list.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module SidekiqUniqueJobs
+  module Redis
+    #
+    # Class List provides convenient access to redis hashes
+    #
+    # @author Mikael Henriksson <mikael@mhenrixon.com>
+    #
+    class List < Entity
+      #
+      # Entries in this list
+      #
+      #
+      # @return [Array<Object>] the elements in this list
+      #
+      def entries
+        redis { |conn| conn.lrange(key, 0, -1) }
+      end
+
+      #
+      # The number of entries in this list
+      #
+      #
+      # @return [Integer] the total number of entries
+      #
+      def count
+        redis { |conn| conn.llen(key) }
+      end
+    end
+  end
+end

data/lib/sidekiq_unique_jobs/redis/set.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module SidekiqUniqueJobs
+  module Redis
+    #
+    # Class Set provides convenient access to redis sets
+    #
+    # @author Mikael Henriksson <mikael@mhenrixon.com>
+    #
+    class Set < Entity
+      #
+      # Return entries for this set
+      #
+      #
+      # @return [Array<String>]
+      #
+      def entries
+        redis { |conn| conn.smembers(key) }
+      end
+
+      #
+      # Returns the count for this sorted set
+      #
+      #
+      # @return [Integer] the number of entries
+      #
+      def count
+        redis { |conn| conn.scard(key) }
+      end
+    end
+  end
+end

data/lib/sidekiq_unique_jobs/redis/sorted_set.rb

@@ -0,0 +1,86 @@
+# frozen_string_literal: true
+
+module SidekiqUniqueJobs
+  module Redis
+    #
+    # Class SortedSet provides convenient access to redis sorted sets
+    #
+    # @author Mikael Henriksson <mikael@mhenrixon.com>
+    #
+    class SortedSet < Entity
+      #
+      # Return entries for this sorted set
+      #
+      # @param [true,false] with_scores true return
+      #
+      # @return [Array<Object>] when given with_scores: false
+      # @return [Hash] when given with_scores: true
+      #
+      def entries(with_scores: true)
+        entrys = redis { |conn| conn.zrange(key, 0, -1, with_scores: with_scores) }
+        return entrys unless with_scores
+
+        entrys.each_with_object({}) { |pair, hash| hash[pair[0]] = pair[1] }
+      end
+
+      #
+      # Adds a value to the sorted set
+      #
+      # @param [Array<Float, String>, String] the values to add
+      #
+      # @return [Boolean, Integer] <description>
+      #
+      def add(values)
+        redis do |conn|
+          if values.is_a?(Array)
+            conn.zadd(key, values)
+          else
+            conn.zadd(key, now_f, values)
+          end
+        end
+      end
+
+      #
+      # Return the zrak of the member
+      #
+      # @param [String] member the member to pull rank on
+      #
+      # @return [Integer]
+      #
+      def rank(member)
+        redis { |conn| conn.zrank(key, member) }
+      end
+
+      #
+      # Return score for a member
+      #
+      # @param [String] member the member for which score
+      #
+      # @return [Integer, Float]
+      #
+      def score(member)
+        redis { |conn| conn.zscore(key, member) }
+      end
+
+      #
+      # Clears the sorted set from all entries
+      #
+      #
+      # @return [Integer] number of entries removed
+      #
+      def clear
+        redis { |conn| conn.zremrangebyrank(key, 0, count) }
+      end
+
+      #
+      # Returns the count for this sorted set
+      #
+      #
+      # @return [Integer] the number of entries
+      #
+      def count
+        redis { |conn| conn.zcard(key) }
+      end
+    end
+  end
+end

data/lib/sidekiq_unique_jobs/redis/string.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+module SidekiqUniqueJobs
+  module Redis
+    #
+    # Class String provides convenient access to redis strings
+    #
+    # @author Mikael Henriksson <mikael@mhenrixon.com>
+    #
+    class String < Entity
+      #
+      # Returns the value of the key
+      #
+      #
+      # @return [String]
+      #
+      def value
+        redis { |conn| conn.get(key) }
+      end
+
+      #
+      # Sets the value of the key to given object
+      #
+      # @param [String] obj the object to update the key with
+      #
+      # @return [true, false]
+      #
+      def set(obj)
+        redis { |conn| conn.set(key, obj) }
+      end
+
+      #
+      # Removes the key from redis
+      #
+      def del(*)
+        redis { |conn| conn.del(key) }
+      end
+
+      #
+      # Used only for compatibility with other keys
+      #
+      # @return [1] when key exists
+      # @return [0] when key does not exists
+      def count
+        exist? ? 1 : 0
+      end
+    end
+  end
+end

data/lib/sidekiq_unique_jobs/rspec/matchers.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module SidekiqUniqueJobs
+  #
+  # We all know what the hell RSpec is no?
+  #
+  # @author Mikael Henriksson <mikael@mhenrixon.com>
+  #
+  module RSpec
+    #
+    # Module Matchers provides RSpec matcher for your workers
+    #
+    # @author Mikael Henriksson <mikael@mhenrixon.com>
+    #
+    module Matchers
+    end
+  end
+end
+
+if defined?(RSpec)
+  require_relative "matchers/have_valid_sidekiq_options"
+
+  RSpec.configure do |config|
+    config.include SidekiqUniqueJobs::RSpec::Matchers
+  end
+end

data/lib/sidekiq_unique_jobs/rspec/matchers/have_valid_sidekiq_options.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+module SidekiqUniqueJobs
+  module RSpec
+    #
+    # Module Matchers provides RSpec matcher for your workers
+    #
+    # @author Mikael Henriksson <mikael@mhenrixon.com>
+    #
+    module Matchers
+      #
+      # Class HaveValidSidekiqOptions validates the unique/lock configuration for a worker.
+      #
+      # @author Mikael Henriksson <mikael@mhenrixon.com>
+      #
+      class HaveValidSidekiqOptions
+        attr_reader :worker, :lock_config, :sidekiq_options
+
+        def matches?(worker)
+          @worker          = worker
+          @sidekiq_options = worker.get_sidekiq_options
+          @lock_config     = SidekiqUniqueJobs.validate_worker(sidekiq_options)
+          lock_config.valid?
+        end
+
+        # :nodoc:
+        def failure_message
+          <<~FAILURE_MESSAGE
+            Expected #{worker} to have valid sidekiq options but found the following problems:
+            #{lock_config.errors_as_string}
+          FAILURE_MESSAGE
+        end
+
+        # :nodoc:
+        def description
+          "have valid sidekiq options"
+        end
+      end
+
+      #
+      # RSpec matcher method for validating that a sidekiq worker has valid unique/lock configuration
+      #
+      #
+      # @return [HaveValidSidekiqOptions] an RSpec matcher
+      #
+      def have_valid_sidekiq_options(*args) # rubocop:disable Naming/PredicateName
+        HaveValidSidekiqOptions.new(*args)
+      end
+    end
+  end
+end