61 lines
2.6 KiB
Ruby
61 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# This is a copy of https://github.com/mperham/sidekiq/blob/32c55e31659a1e6bd42f98334cca5eef2863de8d/lib/sidekiq/scheduled.rb#L11-L34
|
|
#
|
|
# It effectively reverts
|
|
# https://github.com/mperham/sidekiq/commit/9b75467b33759888753191413eddbc15c37a219e
|
|
# because we observe that the extra ZREMs caused by this change can lead to high
|
|
# CPU usage on Redis at peak times:
|
|
# https://gitlab.com/gitlab-com/gl-infra/scalability/-/issues/1179
|
|
#
|
|
module Gitlab
|
|
class SidekiqEnq
|
|
def enqueue_jobs(now = Time.now.to_f.to_s, sorted_sets = Sidekiq::Scheduled::SETS)
|
|
Rails.application.reloader.wrap do
|
|
::Gitlab::WithRequestStore.with_request_store do
|
|
find_jobs_and_enqueue(now, sorted_sets)
|
|
end
|
|
ensure
|
|
::Gitlab::Database::LoadBalancing.release_hosts
|
|
end
|
|
end
|
|
|
|
def find_jobs_and_enqueue(now, sorted_sets)
|
|
# A job's "score" in Redis is the time at which it should be processed.
|
|
# Just check Redis for the set of jobs with a timestamp before now.
|
|
Sidekiq.redis do |conn|
|
|
sorted_sets.each do |sorted_set|
|
|
start_time = ::Gitlab::Metrics::System.monotonic_time
|
|
jobs = redundant_jobs = 0
|
|
|
|
Sidekiq.logger.info(message: 'Enqueuing scheduled jobs', status: 'start', sorted_set: sorted_set)
|
|
|
|
# Get the next item in the queue if it's score (time to execute) is <= now.
|
|
# We need to go through the list one at a time to reduce the risk of something
|
|
# going wrong between the time jobs are popped from the scheduled queue and when
|
|
# they are pushed onto a work queue and losing the jobs.
|
|
while (job = conn.zrangebyscore(sorted_set, "-inf", now, limit: [0, 1]).first)
|
|
|
|
# Pop item off the queue and add it to the work queue. If the job can't be popped from
|
|
# the queue, it's because another process already popped it so we can move on to the
|
|
# next one.
|
|
if conn.zrem(sorted_set, job)
|
|
jobs += 1
|
|
Sidekiq::Client.push(Sidekiq.load_json(job))
|
|
else
|
|
redundant_jobs += 1
|
|
end
|
|
end
|
|
|
|
end_time = ::Gitlab::Metrics::System.monotonic_time
|
|
Sidekiq.logger.info(message: 'Enqueuing scheduled jobs',
|
|
status: 'done',
|
|
sorted_set: sorted_set,
|
|
jobs_count: jobs,
|
|
redundant_jobs_count: redundant_jobs,
|
|
duration_s: end_time - start_time)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|