diff options
author | Yorick Peterse <yorickpeterse@gmail.com> | 2019-04-09 13:17:40 +0300 |
---|---|---|
committer | Yorick Peterse <yorickpeterse@gmail.com> | 2019-04-09 13:17:40 +0300 |
commit | 7e22054a361a33f65a7bd5c544a41642841dcb64 (patch) | |
tree | f0fc2b63bf3ad7a439ceef1e1684a504f0ebd74b /lib | |
parent | e38fe79cd278ef642340de4046b97290a111e885 (diff) | |
parent | 285fcb4744ed5464aab7d2ff248e7824dd77315c (diff) |
Merge branch 'osw-schedule-multiple-mr-assignees-ff-auto-enabling' into 'master'
Add methods to check dead and retrying jobs
See merge request gitlab-org/gitlab-ce!27149
Diffstat (limited to 'lib')
-rw-r--r-- | lib/gitlab/background_migration.rb | 28 |
1 files changed, 22 insertions, 6 deletions
diff --git a/lib/gitlab/background_migration.rb b/lib/gitlab/background_migration.rb index 5251e0fadf9..2e3a4f3b869 100644 --- a/lib/gitlab/background_migration.rb +++ b/lib/gitlab/background_migration.rb @@ -58,11 +58,31 @@ module Gitlab migration_class_for(class_name).new.perform(*arguments) end - def self.exists?(migration_class) + def self.exists?(migration_class, additional_queues = []) enqueued = Sidekiq::Queue.new(self.queue) scheduled = Sidekiq::ScheduledSet.new - [enqueued, scheduled].each do |queue| + enqueued_job?([enqueued, scheduled], migration_class) + end + + def self.dead_jobs?(migration_class) + dead_set = Sidekiq::DeadSet.new + + enqueued_job?([dead_set], migration_class) + end + + def self.retrying_jobs?(migration_class) + retry_set = Sidekiq::RetrySet.new + + enqueued_job?([retry_set], migration_class) + end + + def self.migration_class_for(class_name) + const_get(class_name) + end + + def self.enqueued_job?(queues, migration_class) + queues.each do |queue| queue.each do |job| return true if job.queue == self.queue && job.args.first == migration_class end @@ -70,9 +90,5 @@ module Gitlab false end - - def self.migration_class_for(class_name) - const_get(class_name) - end end end |