summaryrefslogtreecommitdiff
path: root/db/post_migrate/20201130103926_schedule_populate_dismissed_state_for_vulnerabilities.rb
diff options
context:
space:
mode:
Diffstat (limited to 'db/post_migrate/20201130103926_schedule_populate_dismissed_state_for_vulnerabilities.rb')
-rw-r--r--db/post_migrate/20201130103926_schedule_populate_dismissed_state_for_vulnerabilities.rb62
1 files changed, 0 insertions, 62 deletions
diff --git a/db/post_migrate/20201130103926_schedule_populate_dismissed_state_for_vulnerabilities.rb b/db/post_migrate/20201130103926_schedule_populate_dismissed_state_for_vulnerabilities.rb
deleted file mode 100644
index 5e8da532251..00000000000
--- a/db/post_migrate/20201130103926_schedule_populate_dismissed_state_for_vulnerabilities.rb
+++ /dev/null
@@ -1,62 +0,0 @@
-# frozen_string_literal: true
-
-class SchedulePopulateDismissedStateForVulnerabilities < ActiveRecord::Migration[6.0]
- include Gitlab::Database::MigrationHelpers
-
- TMP_INDEX_NAME = 'tmp_index_on_vulnerabilities_non_dismissed'
-
- DOWNTIME = false
- BATCH_SIZE = 1_000
- VULNERABILITY_BATCH_SIZE = 5_000
- DELAY_INTERVAL = 3.minutes.to_i
- MIGRATION_CLASS = 'PopulateDismissedStateForVulnerabilities'
-
- VULNERABILITY_JOIN_CONDITION = 'JOIN "vulnerability_occurrences" ON "vulnerability_occurrences"."vulnerability_id" = "vulnerabilities"."id"'
- FEEDBACK_WHERE_CONDITION = <<~SQL
- EXISTS (SELECT 1 FROM vulnerability_feedback
- WHERE "vulnerability_occurrences"."project_id" = "vulnerability_feedback"."project_id"
- AND "vulnerability_occurrences"."report_type" = "vulnerability_feedback"."category"
- AND ENCODE("vulnerability_occurrences"."project_fingerprint", 'hex') = "vulnerability_feedback"."project_fingerprint"
- AND "vulnerability_feedback"."feedback_type" = 0
- )
- SQL
-
- class Vulnerability < ActiveRecord::Base # rubocop:disable Style/Documentation
- include EachBatch
-
- self.table_name = 'vulnerabilities'
- end
-
- disable_ddl_transaction!
-
- def up
- add_concurrent_index(:vulnerabilities, :id, where: 'state <> 2', name: TMP_INDEX_NAME)
-
- batch = []
- index = 1
-
- Vulnerability.where('state <> 2').each_batch(of: VULNERABILITY_BATCH_SIZE) do |relation|
- ids = relation
- .joins(VULNERABILITY_JOIN_CONDITION)
- .where(FEEDBACK_WHERE_CONDITION)
- .pluck('vulnerabilities.id')
-
- ids.each do |id|
- batch << id
-
- if batch.size == BATCH_SIZE
- migrate_in(index * DELAY_INTERVAL, MIGRATION_CLASS, batch)
- index += 1
-
- batch.clear
- end
- end
- end
-
- migrate_in(index * DELAY_INTERVAL, MIGRATION_CLASS, batch) unless batch.empty?
- end
-
- def down
- remove_concurrent_index_by_name(:vulnerabilities, TMP_INDEX_NAME)
- end
-end