summaryrefslogtreecommitdiff
path: root/db/post_migrate/20220201193033_add_unique_index_to_vulnerability_finding_links_with_truncate.rb
blob: cc9dabdf6248c9ed2a09ba33ea3beab513808364 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# frozen_string_literal: true

class AddUniqueIndexToVulnerabilityFindingLinksWithTruncate < Gitlab::Database::Migration[1.0]
  disable_ddl_transaction!

  NAME_URL_INDEX_NAME = 'finding_link_name_url_idx'
  URL_INDEX_NAME = 'finding_link_url_idx'

  def up
    execute('TRUNCATE TABLE vulnerability_finding_links')

    add_concurrent_index :vulnerability_finding_links, [:vulnerability_occurrence_id, :name, :url], unique: true, name: NAME_URL_INDEX_NAME
    add_concurrent_index :vulnerability_finding_links, [:vulnerability_occurrence_id, :url], unique: true, where: 'name is null', name: URL_INDEX_NAME
  end

  def down
    remove_concurrent_index :vulnerability_finding_links, [:vulnerability_occurrence_id, :name, :url], name: NAME_URL_INDEX_NAME
    remove_concurrent_index :vulnerability_finding_links, [:vulnerability_occurrence_id, :url], name: URL_INDEX_NAME
  end
end