Refactored initializer code to its own class and added tests

1 files changed, 21 insertions, 0 deletions

diff --git a/lib/gitlab/sidekiq_throttler.rb b/lib/gitlab/sidekiq_throttler.rb
new file mode 100644
index 00000000000..771736e7606
--- /dev/null
+++ b/lib/gitlab/sidekiq_throttler.rb
@@ -0,0 +1,21 @@
+module Gitlab
+  class SidekiqThrottler
+    class << self
+      def execute!
+        if Gitlab::CurrentSettings.sidekiq_throttling_enabled?
+          current_application_settings.sidekiq_throttling_queues.each do |queue|
+            Sidekiq::Queue[queue].limit = set_limit
+          end
+        end
+      end
+
+      private
+
+      def set_limit
+        factor = current_application_settings.sidekiq_throttling_factor
+
+        (factor * Sidekiq.options[:concurrency]).ceil
+      end
+    end
+  end
+end