sidekiq-throttled 0.18.0 → 1.0.0.alpha

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/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/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

data/lib/sidekiq/throttled/web/queues.html.erb

@@ -1,49 +0,0 @@
-<h3><%= t('Queues') %></h3>
-
-<div class="table_container">
-  <table class="queues table table-hover table-bordered table-striped table-white">
-    <thead>
-      <th><%= t('Queue') %></th>
-      <th><%= t('Size') %></th>
-      <th><%= t('Actions') %></th>
-    </thead>
-    <% @queues.each do |queue| %>
-      <tr>
-        <td>
-          <a href="<%= root_path %>queues/<%= CGI.escape(queue.name) %>"><%= queue.name %></a>
-        </td>
-        <td><%= number_with_delimiter(queue.size) %> </td>
-        <td width="20%">
-          <form action="<%=root_path %>enhanced-queues/<%= CGI.escape(queue.name) %>" method="post" style="display:inline">
-            <%= csrf_tag %>
-            <% if queue.paused? %>
-              <button class="btn btn-primary btn-pauser-resume btn-xs" type="submit" name="action" value="resume">Resume</button>
-            <% else %>
-              <button class="btn btn-danger btn-pauser-pause btn-xs" type="submit" name="action" value="pause">Pause</button>
-            <% end %>
-
-            <button class="btn btn-danger btn-xs" type="submit" name="action" value="delete" data-confirm="<%= t('AreYouSureDeleteQueue', :queue => h(queue.name)) %>"><%= t('Delete') %></button>
-          </form>
-        </td>
-      </tr>
-    <% end %>
-  </table>
-</div>
-
-<style>
-.btn-pauser-pause {
-  background-image: -webkit-gradient(linear, 50% 0%, 50% 100%, color-stop(0%, #b13e00), color-stop(100%, #983500));
-  background-image: -webkit-linear-gradient(#b13e00, #983500);
-  background-image: -moz-linear-gradient(#b13e00, #983500);
-  background-image: -o-linear-gradient(#b13e00, #983500);
-  background-image: linear-gradient(#b13e00, #983500);
-}
-
-.btn-pauser-resume {
-  background-image: -webkit-gradient(linear, 50% 0%, 50% 100%, color-stop(0%, #00b13e), color-stop(100%, #009835));
-  background-image: -webkit-linear-gradient(#00b13e, #009835);
-  background-image: -moz-linear-gradient(#00b13e, #009835);
-  background-image: -o-linear-gradient(#00b13e, #009835);
-  background-image: linear-gradient(#00b13e, #009835);
-}
-</style>

data/lib/sidekiq/throttled/web/summary_fix.js

@@ -1,10 +0,0 @@
-document.addEventListener("DOMContentLoaded", function () {
-  var elem = document.querySelector(".summary li.enqueued > a"), href;
-
-  if (!elem) {
-    console.warn("[Sidekiq::Threshold] cannot find summary bar link to fix");
-  } else {
-    href = elem.getAttribute("href").toString();
-    elem.setAttribute("href", href.replace(/\/queues$/, "/enhanced-queues"));
-  }
-});

data/lib/sidekiq/throttled/web/summary_fix.rb

@@ -1,35 +0,0 @@
-# frozen_string_literal: true
-
-module Sidekiq
-  module Throttled
-    module Web
-      module SummaryFix
-        JAVASCRIPT = [File.read(File.expand_path("summary_fix.js", __dir__)).freeze].freeze
-        HEADERS    = { "Content-Type" => "application/javascript" }.freeze
-
-        class << self
-          attr_accessor :enabled
-
-          def apply!(app)
-            Sidekiq::WebAction.prepend SummaryFix
-
-            app.get("/throttled/summary_fix") do
-              [200, HEADERS.dup, JAVASCRIPT.dup]
-            end
-          end
-        end
-
-        def display_custom_head
-          "#{super}#{summary_fix_script if SummaryFix.enabled}"
-        end
-
-        private
-
-        def summary_fix_script
-          src = "#{root_path}throttled/summary_fix"
-          %(<script type="text/javascript" src="#{src}"></script>)
-        end
-      end
-    end
-  end
-end

data/rubocop/layout.yml

@@ -1,24 +0,0 @@
-Layout/ArgumentAlignment:
-  EnforcedStyle: with_fixed_indentation
-
-Layout/EmptyLinesAroundAttributeAccessor:
-  Enabled: true
-
-Layout/FirstArrayElementIndentation:
-  EnforcedStyle: consistent
-
-Layout/FirstHashElementIndentation:
-  Enabled: true
-  EnforcedStyle: consistent
-
-Layout/HashAlignment:
-  Enabled: true
-  EnforcedHashRocketStyle: table
-  EnforcedColonStyle: table
-
-Layout/SpaceAroundMethodCallOperator:
-  Enabled: true
-
-Layout/MultilineMethodCallIndentation:
-  Enabled: true
-  EnforcedStyle: indented

data/rubocop/lint.yml

@@ -1,41 +0,0 @@
-Lint/BinaryOperatorWithIdenticalOperands:
-  Enabled: true
-
-Lint/DeprecatedOpenSSLConstant:
-  Enabled: true
-
-Lint/DuplicateElsifCondition:
-  Enabled: true
-
-Lint/DuplicateRescueException:
-  Enabled: true
-
-Lint/EmptyConditionalBody:
-  Enabled: true
-
-Lint/FloatComparison:
-  Enabled: true
-
-Lint/MissingSuper:
-  Enabled: true
-
-Lint/MixedRegexpCaptureTypes:
-  Enabled: true
-
-Lint/OutOfRangeRegexpRef:
-  Enabled: true
-
-Lint/RaiseException:
-  Enabled: true
-
-Lint/SelfAssignment:
-  Enabled: true
-
-Lint/StructNewOverride:
-  Enabled: true
-
-Lint/TopLevelReturnWithArgument:
-  Enabled: true
-
-Lint/UnreachableLoop:
-  Enabled: true

data/rubocop/metrics.yml

@@ -1,4 +0,0 @@
-Metrics/BlockLength:
-  Exclude:
-    - "spec/**/*_spec.rb"
-    - "**/*.gemspec"