lib/gitlab/sidekiq_middleware/pause_control/workers_map.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

# frozen_string_literal: true

module Gitlab
  module SidekiqMiddleware
    module PauseControl
      class WorkersMap
        class << self
          attr_reader :workers

          def set_strategy_for(strategy:, worker:)
            raise ArgumentError, "Unknown strategy: #{strategy}" unless PauseControl::STRATEGIES.key?(strategy)

            @workers ||= Hash.new { |h, k| h[k] = [] }
            @workers[strategy].push(worker)
          end

          def strategy_for(worker:)
            return unless @workers

            @workers.find { |_, v| v.include?(worker) }&.first
          end
        end
      end
    end
  end
end