diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2021-11-30 08:04:12 +0000 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2021-11-30 08:04:12 +0000 |
commit | 20755ace290946895c0a3e2f4d2f7916069b2a41 (patch) | |
tree | c5c16305a83a3af54bb964e09f6bbfb18a27447d | |
parent | b543e84093a004afc4a23a0971271075e15df682 (diff) | |
download | gitlab-ce-20755ace290946895c0a3e2f4d2f7916069b2a41.tar.gz |
Add latest changes from gitlab-org/gitlab@14-4-stable-ee
-rw-r--r-- | config/initializers/sidekiq.rb | 2 | ||||
-rw-r--r-- | lib/gitlab/patch/sidekiq_cron_poller.rb | 17 | ||||
-rw-r--r-- | lib/gitlab/sidekiq_enq.rb | 10 |
3 files changed, 29 insertions, 0 deletions
diff --git a/config/initializers/sidekiq.rb b/config/initializers/sidekiq.rb index d33550b82d1..86ddb45b6b9 100644 --- a/config/initializers/sidekiq.rb +++ b/config/initializers/sidekiq.rb @@ -114,3 +114,5 @@ Sidekiq.configure_client do |config| config.client_middleware(&Gitlab::SidekiqMiddleware.client_configurator) end + +Sidekiq::Cron::Poller.prepend Gitlab::Patch::SidekiqCronPoller diff --git a/lib/gitlab/patch/sidekiq_cron_poller.rb b/lib/gitlab/patch/sidekiq_cron_poller.rb new file mode 100644 index 00000000000..56ca24c68f5 --- /dev/null +++ b/lib/gitlab/patch/sidekiq_cron_poller.rb @@ -0,0 +1,17 @@ +# frozen_string_literal: true + +module Gitlab + module Patch + module SidekiqCronPoller + def enqueue + Rails.application.reloader.wrap do + ::Gitlab::WithRequestStore.with_request_store do + super + ensure + ::Gitlab::Database::LoadBalancing.release_hosts + end + end + end + end + end +end diff --git a/lib/gitlab/sidekiq_enq.rb b/lib/gitlab/sidekiq_enq.rb index d8a01ac8ef4..8095eba9708 100644 --- a/lib/gitlab/sidekiq_enq.rb +++ b/lib/gitlab/sidekiq_enq.rb @@ -11,6 +11,16 @@ 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| |