db/post_migrate/20210330130420_drop_finding_fingerprint_table.rb


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
38
39
40

# frozen_string_literal: true

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

  DOWNTIME = false

  FINGERPRINT_IDX = :idx_vuln_fingerprints_on_occurrences_id_and_fingerprint_sha256
  UNIQ_IDX = :idx_vuln_fingerprints_uniqueness_fingerprint_sha256

  def up
    with_lock_retries do
      drop_table :vulnerability_finding_fingerprints
    end
  end

  def down
    with_lock_retries do
      create_table :vulnerability_finding_fingerprints do |t|
        t.references :finding,
          index: true,
          null: false,
          foreign_key: { to_table: :vulnerability_occurrences, column: :finding_id, on_delete: :cascade }

        t.timestamps_with_timezone null: false

        t.integer :algorithm_type, null: false, limit: 2
        t.binary :fingerprint_sha256, null: false

        t.index %i[finding_id fingerprint_sha256],
          name: FINGERPRINT_IDX,
          unique: true # only one link should exist between occurrence and the fingerprint

        t.index %i[finding_id algorithm_type fingerprint_sha256],
          name: UNIQ_IDX,
          unique: true # these should be unique
      end
    end
  end
end