mirror of
https://github.com/discourse/discourse.git
synced 2024-11-24 15:49:55 +08:00
43 lines
645 B
Ruby
43 lines
645 B
Ruby
module Sidekiq
|
|
|
|
def self.pause!
|
|
Sidekiq.redis { |r| r.set(paused_key, 1) }
|
|
true
|
|
end
|
|
|
|
def self.paused?
|
|
Sidekiq.redis { |r| !!r.get(paused_key) }
|
|
end
|
|
|
|
def self.unpause!
|
|
Sidekiq.redis { |r| r.del(paused_key) }
|
|
true
|
|
end
|
|
|
|
private
|
|
|
|
def self.paused_key
|
|
"sidekiq_is_paused"
|
|
end
|
|
|
|
end
|
|
|
|
# server middleware that will reschedule work whenever Sidekiq is paused
|
|
class Sidekiq::Pausable
|
|
|
|
attr_reader :delay
|
|
|
|
def initialize(delay = 5.seconds)
|
|
@delay = delay
|
|
end
|
|
|
|
def call(worker, msg, queue)
|
|
if Sidekiq.paused?
|
|
worker.class.perform_in(delay, *msg['args'])
|
|
else
|
|
yield
|
|
end
|
|
end
|
|
|
|
end
|