Use sidekiq event to determine when it is stopping

fatkodima · May 4, 2024 · 87def54 · 87def54
1 parent 51a9496
commit 87def54
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/lib/sidekiq_iteration.rb b/lib/sidekiq_iteration.rb
@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
 require "sidekiq"
+require_relative "sidekiq_iteration/iteration"
+require_relative "sidekiq_iteration/job_retry_patch"
 require_relative "sidekiq_iteration/version"
 
 module SidekiqIteration
@@ -44,8 +46,14 @@ class << self
     def logger
       @logger ||= Sidekiq.logger
     end
+
+    # @private
+    attr_accessor :stopping
   end
 end
 
-require_relative "sidekiq_iteration/iteration"
-require_relative "sidekiq_iteration/job_retry_patch"
+Sidekiq.configure_server do |config|
+  config.on(:quiet) do
+    SidekiqIteration.stopping = true
+  end
+end
diff --git a/lib/sidekiq_iteration/iteration.rb b/lib/sidekiq_iteration/iteration.rb
@@ -20,8 +20,7 @@ def self.included(base)
         end
 
         throttle_on(backoff: SidekiqIteration.default_retry_backoff) do
-          defined?(Sidekiq::CLI) &&
-            Sidekiq::CLI.instance.launcher.stopping?
+          SidekiqIteration.stopping
         end
       end