blob: aee6d5484d58e51e9178504c2b09e89b6e1c20cb (
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
37
|
# frozen_string_literal: true
class PopulateDismissalInformationForVulnerabilities < ActiveRecord::Migration[6.0]
DOWNTIME = false
BATCH_SIZE = 100
UPDATE_QUERY = <<~SQL
UPDATE
vulnerabilities
SET
dismissed_at = COALESCE(dismissed_at, updated_at),
dismissed_by_id = COALESCE(dismissed_by_id, updated_by_id, last_edited_by_id, author_id)
WHERE
vulnerabilities.id IN (%{ids})
SQL
class Vulnerability < ActiveRecord::Base
include EachBatch
self.table_name = 'vulnerabilities'
enum state: { detected: 1, confirmed: 4, resolved: 3, dismissed: 2 }
scope :broken, -> { dismissed.where('dismissed_at IS NULL OR dismissed_by_id IS NULL') }
end
def up
Vulnerability.broken.each_batch(of: BATCH_SIZE) do |batch|
query = format(UPDATE_QUERY, ids: batch.select(:id).to_sql)
connection.execute(query)
end
end
def down
# no-op
end
end
|
