summaryrefslogtreecommitdiff
path: root/lib/gitlab/background_migration.rb
blob: 914a3b72abd0b56c06cc24693e8f7ab8f7276488 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
module Gitlab
  module BackgroundMigration
    # Begins stealing jobs from the background migrations queue, blocking the
    # caller until all jobs have been completed.
    #
    # steal_class - The name of the class for which to steal jobs.
    def self.steal(steal_class)
      queue = Sidekiq::Queue.
        new(BackgroundMigrationWorker.sidekiq_options['queue'])

      queue.each do |job|
        migration_class, migration_args = job.args

        next unless migration_class == steal_class

        perform(migration_class, migration_args)

        job.delete
      end
    end

    # class_name - The name of the background migration class as defined in the
    #              Gitlab::BackgroundMigration namespace.
    #
    # arguments - The arguments to pass to the background migration's "perform"
    #             method.
    def self.perform(class_name, arguments)
      const_get(class_name).new.perform(*arguments)
    end
  end
end