summaryrefslogtreecommitdiff
path: root/db/post_migrate/20201028160832_schedule_populate_missing_dismissal_information_for_vulnerabilities.rb
diff options
context:
space:
mode:
Diffstat (limited to 'db/post_migrate/20201028160832_schedule_populate_missing_dismissal_information_for_vulnerabilities.rb')
-rw-r--r--db/post_migrate/20201028160832_schedule_populate_missing_dismissal_information_for_vulnerabilities.rb23
1 files changed, 23 insertions, 0 deletions
diff --git a/db/post_migrate/20201028160832_schedule_populate_missing_dismissal_information_for_vulnerabilities.rb b/db/post_migrate/20201028160832_schedule_populate_missing_dismissal_information_for_vulnerabilities.rb
new file mode 100644
index 00000000000..f358ea863db
--- /dev/null
+++ b/db/post_migrate/20201028160832_schedule_populate_missing_dismissal_information_for_vulnerabilities.rb
@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+class SchedulePopulateMissingDismissalInformationForVulnerabilities < ActiveRecord::Migration[6.0]
+ include Gitlab::Database::MigrationHelpers
+
+ DOWNTIME = false
+ BATCH_SIZE = 1_000
+ DELAY_INTERVAL = 3.minutes.to_i
+ MIGRATION_CLASS = 'PopulateMissingVulnerabilityDismissalInformation'
+
+ disable_ddl_transaction!
+
+ def up
+ ::Gitlab::BackgroundMigration::PopulateMissingVulnerabilityDismissalInformation::Vulnerability.broken.each_batch(of: BATCH_SIZE) do |batch, index|
+ vulnerability_ids = batch.pluck(:id)
+ migrate_in(index * DELAY_INTERVAL, MIGRATION_CLASS, vulnerability_ids)
+ end
+ end
+
+ def down
+ # no-op
+ end
+end