summaryrefslogtreecommitdiff
path: root/db/post_migrate/20210415155043_move_container_registry_enabled_to_project_features3.rb
blob: 6fd8d280c979165606cc67efdfba97e9571a1f98 (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
32
33
34
35
36
# frozen_string_literal: true

class MoveContainerRegistryEnabledToProjectFeatures3 < ActiveRecord::Migration[6.0]
  include Gitlab::Database::MigrationHelpers
  include Gitlab::Database::DynamicModelHelpers

  BATCH_SIZE = 21_000
  MIGRATION = 'MoveContainerRegistryEnabledToProjectFeature'

  disable_ddl_transaction!

  def up
    # Delete any existing jobs from the queue
    delete_queued_jobs(MIGRATION)

    # Delete existing rows in background_migration_jobs table
    bg_migration_job_class = define_model('background_migration_jobs')
    bg_migration_job_class.where(class_name: MIGRATION).delete_all

    batchable_project_class = define_batchable_model('projects')
    queue_background_migration_jobs_by_range_at_intervals(batchable_project_class, MIGRATION, 2.minutes, batch_size: BATCH_SIZE, track_jobs: true)
  end

  def down
    # no-op
  end

  private

  def define_model(table_name)
    Class.new(ActiveRecord::Base) do
      self.table_name = table_name
      self.inheritance_column = :_type_disabled
    end
  end
end