mirror of
https://github.com/discourse/discourse.git
synced 2024-11-24 08:32:26 +08:00
44cf3cf975
If sidekiq is paused or Discourse is in readonly continue to queue heartbeats If we do not do that then a master process can end up reaping sidekiq workers and causing various badness This also impacts restore which can do weird stuff TM in cases like this
89 lines
1.5 KiB
Ruby
89 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(Rails.env.test? ? 0.01 : 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? && !(Jobs::RunHeartbeat === worker)
|
|
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
|