sidekiq-throttled 0.17.0 → 1.1.0

data/lib/sidekiq/throttled/communicator.rb

@@ -1,116 +0,0 @@
-# frozen_string_literal: true
-
-require "singleton"
-
-require "sidekiq/throttled/communicator/exception_handler"
-require "sidekiq/throttled/communicator/listener"
-require "sidekiq/throttled/communicator/callbacks"
-
-module Sidekiq
-  module Throttled
-    # Inter-process communication for sidekiq. It starts listener thread on
-    # sidekiq server and listens for incoming messages.
-    #
-    # @example
-    #
-    #   # Add incoming message handler for server
-    #   Communicator.instance.receive "knock" do |who|
-    #     puts "#{who}'s knocking on the door"
-    #   end
-    #
-    #   # Emit message from console
-    #   Sidekiq.redis do |conn|
-    #     Communicator.instance.transmit(conn, "knock", "ixti")
-    #   end
-    class Communicator
-      include Singleton
-      include ExceptionHandler
-
-      # Redis PUB/SUB channel name
-      #
-      # @see http://redis.io/topics/pubsub
-      CHANNEL_NAME = "sidekiq:throttled"
-      private_constant :CHANNEL_NAME
-
-      # Initializes singleton instance.
-      def initialize
-        @callbacks = Callbacks.new
-        @listener  = nil
-        @mutex     = Mutex.new
-      end
-
-      # Starts listener thread.
-      #
-      # @return [void]
-      def start_listener
-        @mutex.synchronize do
-          @listener ||= Listener.new(CHANNEL_NAME, @callbacks)
-        end
-      end
-
-      # Stops listener thread.
-      #
-      # @return [void]
-      def stop_listener
-        @mutex.synchronize do
-          @listener&.stop
-          @listener = nil
-        end
-      end
-
-      # Configures Sidekiq server to start/stop listener thread.
-      #
-      # @private
-      # @return [void]
-      def setup!
-        Sidekiq.configure_server do |config|
-          config.on(:startup) { start_listener }
-          config.on(:quiet) { stop_listener }
-        end
-      end
-
-      # Transmit message to listeners.
-      #
-      # @example
-      #
-      #   Sidekiq.redis do |conn|
-      #     Communicator.instance.transmit(conn, "knock")
-      #   end
-      #
-      # @param [Redis] redis Redis client
-      # @param [#to_s] message
-      # @param [Object] payload
-      # @return [void]
-      def transmit(redis, message, payload = nil)
-        redis.publish(CHANNEL_NAME, Marshal.dump([message.to_s, payload]))
-      end
-
-      # Add incoming message handler.
-      #
-      # @example
-      #
-      #   Communicator.instance.receive "knock" do |payload|
-      #     # do something upon `knock` message
-      #   end
-      #
-      # @param [#to_s] message
-      # @yield [payload] Runs given block everytime `message` being received.
-      # @yieldparam [Object, nil] payload Payload that was transmitted
-      # @yieldreturn [void]
-      # @return [void]
-      def receive(message, &handler)
-        @callbacks.on("message:#{message}", &handler)
-      end
-
-      # Communicator readiness hook.
-      #
-      # @yield Runs given block every time listener thread subscribes
-      #   to Redis pub/sub channel.
-      # @return [void]
-      def ready(&handler)
-        @callbacks.on("ready", &handler)
-        yield if @listener&.ready?
-      end
-    end
-  end
-end

data/lib/sidekiq/throttled/configuration.rb

@@ -1,50 +0,0 @@
-# frozen_string_literal: true
-
-module Sidekiq
-  module Throttled
-    # Configuration holder.
-    class Configuration
-      # Class constructor.
-      def initialize
-        reset!
-      end
-
-      # Reset configuration to defaults.
-      #
-      # @return [self]
-      def reset!
-        @inherit_strategies = false
-
-        self
-      end
-
-      # Instructs throttler to lookup strategies in parent classes, if there's
-      # no own strategy:
-      #
-      #     class FooJob
-      #       include Sidekiq::Job
-      #       include Sidekiq::Throttled::Job
-      #
-      #       sidekiq_throttle :concurrency => { :limit => 42 }
-      #     end
-      #
-      #     class BarJob < FooJob
-      #     end
-      #
-      # By default in the example above, `Bar` won't have throttling options.
-      # Set this flag to `true` to enable this lookup in initializer, after
-      # that `Bar` will use `Foo` throttling bucket.
-      def inherit_strategies=(value)
-        @inherit_strategies = value ? true : false
-      end
-
-      # Whenever throttled workers should inherit parent's strategies or not.
-      # Default: `false`.
-      #
-      # @return [Boolean]
-      def inherit_strategies?
-        @inherit_strategies
-      end
-    end
-  end
-end

data/lib/sidekiq/throttled/expirable_list.rb

@@ -1,70 +0,0 @@
-# frozen_string_literal: true
-
-require "monitor"
-
-module Sidekiq
-  module Throttled
-    # List that tracks when elements were added and enumerates over those not
-    # older than `ttl` seconds ago.
-    #
-    # ## Implementation
-    #
-    # Internally list holds an array of arrays. Thus each element is a tuple of
-    # monotonic timestamp (when element was added) and element itself:
-    #
-    #     [
-    #       [ 123456.7890, "default" ],
-    #       [ 123456.7891, "urgent" ],
-    #       [ 123457.9621, "urgent" ],
-    #       ...
-    #     ]
-    #
-    # It does not deduplicates elements. Eviction happens only upon elements
-    # retrieval (see {#each}).
-    #
-    # @see https://ruby-doc.org/core/Process.html#method-c-clock_gettime
-    # @see https://linux.die.net/man/3/clock_gettime
-    #
-    # @private
-    class ExpirableList
-      include Enumerable
-
-      # @param ttl [Float] elements time-to-live in seconds
-      def initialize(ttl)
-        @ttl = ttl.to_f
-        @arr = []
-        @mon = Monitor.new
-      end
-
-      # Pushes given element into the list.
-      #
-      # @params element [Object]
-      # @return [ExpirableList] self
-      def <<(element)
-        @mon.synchronize { @arr << [::Process.clock_gettime(::Process::CLOCK_MONOTONIC), element] }
-        self
-      end
-
-      # Evicts expired elements and calls the given block once for each element
-      # left, passing that element as a parameter.
-      #
-      # @yield [element]
-      # @return [Enumerator] if no block given
-      # @return [ExpirableList] self if block given
-      def each
-        return to_enum __method__ unless block_given?
-
-        @mon.synchronize do
-          horizon = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC) - @ttl
-
-          # drop all elements older than horizon
-          @arr.shift while @arr[0] && @arr[0][0] < horizon
-
-          @arr.each { |x| yield x[1] }
-        end
-
-        self
-      end
-    end
-  end
-end

data/lib/sidekiq/throttled/fetch/unit_of_work.rb

@@ -1,83 +0,0 @@
-# frozen_string_literal: true
-
-require "sidekiq"
-
-require "sidekiq/throttled/queue_name"
-
-module Sidekiq
-  module Throttled
-    class Fetch
-      # BRPOP response envelope.
-      #
-      # @see Throttled::Fetch
-      # @private
-      class UnitOfWork
-        # @return [String] Redis key where job was pulled from
-        attr_reader :queue
-
-        # @return [String] Job's JSON payload
-        attr_reader :job
-
-        # @param [String] queue Redis key where job was pulled from
-        # @param [String] job Job's JSON payload
-        def initialize(queue, job)
-          @queue = queue
-          @job   = job
-        end
-
-        # Callback that is called by `Sidekiq::Processor` when job was
-        # succeccfully processed. Most likely this is used by `ReliableFetch`
-        # of Sidekiq Pro/Enterprise to remove job from running queue.
-        #
-        # @return [void]
-        def acknowledge
-          # do nothing
-        end
-
-        # Normalized `queue` name.
-        #
-        # @see QueueName.normalize
-        # @return [String]
-        def queue_name
-          @queue_name ||= QueueName.normalize queue
-        end
-
-        # Pushes job back to the tail of the queue, so that it will be popped
-        # first next time fetcher will pull job.
-        #
-        # @note This is triggered when job was not finished and Sidekiq server
-        #   process was terminated. It is a reverse of whatever fetcher was
-        #   doing to pull the job out of queue.
-        #
-        # @param [Redis] pipelined connection for requeing via Redis#pipelined
-        # @return [void]
-        def requeue(pipeline = nil)
-          if pipeline
-            pipeline.rpush(QueueName.expand(queue_name), job)
-          else
-            Sidekiq.redis { |conn| conn.rpush(QueueName.expand(queue_name), job) }
-          end
-        end
-
-        # Pushes job back to the head of the queue, so that job won't be tried
-        # immediately after it was requeued (in most cases).
-        #
-        # @note This is triggered when job is throttled. So it is same operation
-        #   Sidekiq performs upon `Sidekiq::Worker.perform_async` call.
-        #
-        # @return [void]
-        def requeue_throttled
-          Sidekiq.redis { |conn| conn.lpush(QueueName.expand(queue_name), job) }
-        end
-
-        # Tells whenever job should be pushed back to queue (throttled) or not.
-        #
-        # @see Sidekiq::Throttled.throttled?
-        # @return [Boolean]
-        def throttled?
-          Throttled.throttled? job
-        end
-      end
-    end
-  end
-end

data/lib/sidekiq/throttled/fetch.rb

@@ -1,94 +0,0 @@
-# frozen_string_literal: true
-
-require "sidekiq"
-require "sidekiq/throttled/expirable_list"
-require "sidekiq/throttled/fetch/unit_of_work"
-require "sidekiq/throttled/queues_pauser"
-require "sidekiq/throttled/queue_name"
-
-module Sidekiq
-  module Throttled
-    # Throttled fetch strategy.
-    #
-    # @private
-    class Fetch
-      # Timeout to sleep between fetch retries in case of no job received,
-      # as well as timeout to wait for redis to give us something to work.
-      TIMEOUT = 2
-
-      # Initializes fetcher instance.
-      # @param options [Hash]
-      # @option options [Integer] :throttled_queue_cooldown (TIMEOUT)
-      #   Min delay in seconds before queue will be polled again after
-      #   throttled job.
-      # @option options [Boolean] :strict (false)
-      # @option options [Array<#to_s>] :queue
-      def initialize(options)
-        @paused = ExpirableList.new(options.fetch(:throttled_queue_cooldown, TIMEOUT))
-
-        @strict = options.fetch(:strict, false)
-        @queues = options.fetch(:queues).map { |q| QueueName.expand q }
-
-        raise ArgumentError, "empty :queues" if @queues.empty?
-
-        @queues.uniq! if @strict
-      end
-
-      # Retrieves job from redis.
-      #
-      # @return [Sidekiq::Throttled::UnitOfWork, nil]
-      def retrieve_work
-        work = brpop
-        return unless work
-
-        work = UnitOfWork.new(*work)
-        return work unless work.throttled?
-
-        work.requeue_throttled
-        @paused << QueueName.expand(work.queue_name)
-
-        nil
-      end
-
-      def bulk_requeue(units, _options)
-        return if units.empty?
-
-        Sidekiq.logger.debug { "Re-queueing terminated jobs" }
-        Sidekiq.redis do |conn|
-          conn.pipelined do |pipeline|
-            units.each { |unit| unit.requeue(pipeline) }
-          end
-        end
-        Sidekiq.logger.info("Pushed #{units.size} jobs back to Redis")
-      rescue => e
-        Sidekiq.logger.warn("Failed to requeue #{units.size} jobs: #{e}")
-      end
-
-      private
-
-      # Tries to pop pair of `queue` and job `message` out of sidekiq queues.
-      #
-      # @see http://redis.io/commands/brpop
-      # @return [Array(String, String), nil]
-      def brpop
-        queues = filter_queues(@strict ? @queues : @queues.shuffle.uniq)
-
-        if queues.empty?
-          sleep TIMEOUT
-          return
-        end
-
-        Sidekiq.redis { |conn| conn.brpop(*queues, :timeout => TIMEOUT) }
-      end
-
-      # Returns list of queues to try to fetch jobs from.
-      #
-      # @note It may return an empty array.
-      # @param [Array<String>] queues
-      # @return [Array<String>]
-      def filter_queues(queues)
-        QueuesPauser.instance.filter(queues) - @paused.to_a
-      end
-    end
-  end
-end

data/lib/sidekiq/throttled/middleware.rb

@@ -1,22 +0,0 @@
-# frozen_string_literal: true
-
-# internal
-require "sidekiq/throttled/registry"
-
-module Sidekiq
-  module Throttled
-    # Server middleware that notifies strategy that job was finished.
-    #
-    # @private
-    class Middleware
-      # Called within Sidekiq job processing
-      def call(_worker, msg, _queue)
-        yield
-      ensure
-        Registry.get msg["class"] do |strategy|
-          strategy.finalize!(msg["jid"], *msg["args"])
-        end
-      end
-    end
-  end
-end

data/lib/sidekiq/throttled/patches/queue.rb

@@ -1,18 +0,0 @@
-# frozen_string_literal: true
-
-module Sidekiq
-  module Throttled
-    module Patches
-      module Queue
-        def paused?
-          QueuesPauser.instance.paused? name
-        end
-
-        def self.apply!
-          require "sidekiq/api"
-          ::Sidekiq::Queue.send(:prepend, self)
-        end
-      end
-    end
-  end
-end

data/lib/sidekiq/throttled/queue_name.rb

@@ -1,46 +0,0 @@
-# frozen_string_literal: true
-
-module Sidekiq
-  module Throttled
-    # Queue name utility belt.
-    #
-    # @private
-    module QueueName
-      # RegExp used to stip out any redisr-namespace prefixes with `queue:`.
-      QUEUE_NAME_PREFIX_RE = %r{.*queue:}.freeze
-      private_constant :QUEUE_NAME_PREFIX_RE
-
-      class << self
-        # Strips redis-namespace and `queue:` prefix from given queue name.
-        #
-        # @example
-        #
-        #   QueueName.normalize "queue:default"
-        #   # => "default"
-        #
-        #   QueueName.normalize "queue:queue:default"
-        #   # => "default"
-        #
-        #   QueueName.normalize "foo:bar:queue:default"
-        #   # => "default"
-        #
-        # @param [#to_s]
-        # @return [String]
-        def normalize(queue)
-          -queue.to_s.sub(QUEUE_NAME_PREFIX_RE, "")
-        end
-
-        # Prepends `queue:` prefix to given `queue` name.
-        #
-        # @note It does not normalizes queue before expanding it, thus
-        #   double-call of this method will potentially do some harm.
-        #
-        # @param [#to_s] queue Queue name
-        # @return [String]
-        def expand(queue)
-          -"queue:#{queue}"
-        end
-      end
-    end
-  end
-end

data/lib/sidekiq/throttled/queues_pauser.rb

@@ -1,152 +0,0 @@
-# frozen_string_literal: true
-
-require "set"
-require "singleton"
-require "concurrent/timer_task"
-
-require "sidekiq/throttled/patches/queue"
-require "sidekiq/throttled/communicator"
-require "sidekiq/throttled/queue_name"
-
-module Sidekiq
-  module Throttled
-    # Singleton class used to pause queues from being processed.
-    # For the sake of efficiency it uses {Communicator} behind the scene
-    # to notify all processes about paused/resumed queues.
-    #
-    # @private
-    class QueuesPauser
-      include Singleton
-
-      # Redis key of Set with paused queues.
-      #
-      # @return [String]
-      PAUSED_QUEUES = "throttled:X:paused_queues"
-      private_constant :PAUSED_QUEUES
-
-      # {Communicator} message used to notify that queue needs to be paused.
-      #
-      # @return [String]
-      PAUSE_MESSAGE = "pause"
-      private_constant :PAUSE_MESSAGE
-
-      # {Communicator} message used to notify that queue needs to be resumed.
-      #
-      # @return [String]
-      RESUME_MESSAGE = "resume"
-      private_constant :RESUME_MESSAGE
-
-      # Initializes singleton instance.
-      def initialize
-        @paused_queues = Set.new
-        @communicator  = Communicator.instance
-        @mutex         = Mutex.new
-      end
-
-      # Configures Sidekiq server to keep actual list of paused queues.
-      #
-      # @private
-      # @return [void]
-      def setup!
-        Patches::Queue.apply!
-
-        Sidekiq.configure_server do |config|
-          config.on(:startup) { start_watcher }
-          config.on(:quiet) { stop_watcher }
-
-          @communicator.receive(PAUSE_MESSAGE, &method(:add))
-          @communicator.receive(RESUME_MESSAGE, &method(:delete))
-          @communicator.ready { sync! }
-        end
-      end
-
-      # Returns queues list with paused queues being stripped out.
-      #
-      # @private
-      # @return [Array<String>]
-      def filter(queues)
-        @mutex.synchronize { queues - @paused_queues.to_a }
-      rescue => e
-        Sidekiq.logger.error { "[#{self.class}] Failed filter queues: #{e}" }
-        queues
-      end
-
-      # Returns list of paused queues.
-      #
-      # @return [Array<String>]
-      def paused_queues
-        Sidekiq.redis { |conn| conn.smembers(PAUSED_QUEUES).to_a }
-      end
-
-      # Pauses given `queue`.
-      #
-      # @param [#to_s] queue
-      # @return [void]
-      def pause!(queue)
-        queue = QueueName.normalize queue.to_s
-
-        Sidekiq.redis do |conn|
-          conn.sadd(PAUSED_QUEUES, queue)
-          @communicator.transmit(conn, PAUSE_MESSAGE, queue)
-        end
-      end
-
-      # Checks if given `queue` is paused.
-      #
-      # @param queue [#to_s]
-      # @return [Boolean]
-      def paused?(queue)
-        queue = QueueName.normalize queue.to_s
-        Sidekiq.redis { |conn| conn.sismember(PAUSED_QUEUES, queue) }
-      end
-
-      # Resumes given `queue`.
-      #
-      # @param [#to_s] queue
-      # @return [void]
-      def resume!(queue)
-        queue = QueueName.normalize queue.to_s
-
-        Sidekiq.redis do |conn|
-          conn.srem(PAUSED_QUEUES, queue)
-          @communicator.transmit(conn, RESUME_MESSAGE, queue)
-        end
-      end
-
-      private
-
-      def add(queue)
-        @mutex.synchronize do
-          @paused_queues << QueueName.expand(queue)
-        end
-      end
-
-      def delete(queue)
-        @mutex.synchronize do
-          @paused_queues.delete QueueName.expand(queue)
-        end
-      end
-
-      def sync!
-        @mutex.synchronize do
-          @paused_queues.replace(paused_queues.map { |q| QueueName.expand q })
-        end
-      end
-
-      def start_watcher
-        @mutex.synchronize do
-          @watcher ||= Concurrent::TimerTask.execute({
-            :run_now            => true,
-            :execution_interval => 60
-          }) { sync! }
-        end
-      end
-
-      def stop_watcher
-        @mutex.synchronize do
-          defined?(@watcher) && @watcher&.shutdown
-        end
-      end
-    end
-  end
-end

data/lib/sidekiq/throttled/testing.rb

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-require "sidekiq/throttled/registry"
-
-RSpec.configure do |config|
-  config.before :example do
-    Sidekiq::Throttled::Registry.instance_eval do
-      @strategies.clear
-      @aliases.clear
-    end
-  end
-end

data/lib/sidekiq/throttled/utils.rb

@@ -1,19 +0,0 @@
-# frozen_string_literal: true
-
-module Sidekiq
-  module Throttled
-    module Utils
-      module_function
-
-      # Resolve constant from it's name
-      # @param name [#to_s] Constant name
-      # @return [Object, nil] Resolved constant or nil if failed.
-      def constantize(name)
-        name.to_s.sub(%r{^::}, "").split("::").inject(Object, &:const_get)
-      rescue NameError
-        Sidekiq.logger.warn { "Failed to constantize: #{name}" }
-        nil
-      end
-    end
-  end
-end