mirror of
https://github.com/discourse/discourse.git
synced 2024-12-16 11:48:34 +08:00
88 lines
1.5 KiB
Ruby
88 lines
1.5 KiB
Ruby
require 'thread'
|
|
|
|
class SidekiqPauser
|
|
def initialize
|
|
@mutex = Mutex.new
|
|
end
|
|
|
|
def pause!
|
|
redis.setex paused_key, 60, "paused"
|
|
@mutex.synchronize do
|
|
@extend_lease_thread ||= extend_lease_thread
|
|
sleep 0.001 while !paused?
|
|
end
|
|
|
|
true
|
|
end
|
|
|
|
def paused?
|
|
!!redis.get(paused_key)
|
|
end
|
|
|
|
def unpause!
|
|
@mutex.synchronize do
|
|
@extend_lease_thread = nil
|
|
end
|
|
|
|
redis.del(paused_key)
|
|
true
|
|
end
|
|
|
|
private
|
|
|
|
def extend_lease_thread
|
|
Thread.new do
|
|
while true do
|
|
break unless @mutex.synchronize { @extend_lease_thread }
|
|
redis.expire paused_key, 60
|
|
sleep 30
|
|
end
|
|
end
|
|
end
|
|
|
|
def redis
|
|
$redis.without_namespace
|
|
end
|
|
|
|
def paused_key
|
|
"sidekiq_is_paused_v2"
|
|
end
|
|
|
|
end
|
|
|
|
module Sidekiq
|
|
@pauser = SidekiqPauser.new
|
|
def self.pause!
|
|
@pauser.pause!
|
|
end
|
|
|
|
def self.paused?
|
|
@pauser.paused?
|
|
end
|
|
|
|
def self.unpause!
|
|
@pauser.unpause!
|
|
end
|
|
end
|
|
|
|
# server middleware that will reschedule work whenever Sidekiq is paused
|
|
class Sidekiq::Pausable
|
|
|
|
def initialize(delay = 5.seconds)
|
|
@delay = delay
|
|
end
|
|
|
|
def call(worker, msg, queue)
|
|
if Sidekiq.paused?
|
|
worker.class.perform_in(@delay, *msg['args'])
|
|
else
|
|
start = Process.clock_gettime(Process::CLOCK_MONOTONIC)
|
|
result = yield
|
|
duration = Process.clock_gettime(Process::CLOCK_MONOTONIC) - start
|
|
DiscourseEvent.trigger(:sidekiq_job_ran, worker, msg, queue, duration)
|
|
result
|
|
end
|
|
end
|
|
|
|
end
|