diff options
Diffstat (limited to 'lib/gitlab/background_migration.rb')
-rw-r--r-- | lib/gitlab/background_migration.rb | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/lib/gitlab/background_migration.rb b/lib/gitlab/background_migration.rb index ddd6b11eebb..6a16c37e880 100644 --- a/lib/gitlab/background_migration.rb +++ b/lib/gitlab/background_migration.rb @@ -58,6 +58,14 @@ module Gitlab migration_class_for(class_name).new.perform(*arguments) end + def self.remaining + scheduled = Sidekiq::ScheduledSet.new.count do |job| + job.queue == self.queue + end + + scheduled + Sidekiq::Queue.new(self.queue).size + end + def self.exists?(migration_class, additional_queues = []) enqueued = Sidekiq::Queue.new(self.queue) scheduled = Sidekiq::ScheduledSet.new |