db/post_migrate/20220603173103_update_vulnerabilities_project_id_id_index.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

# frozen_string_literal: true

class UpdateVulnerabilitiesProjectIdIdIndex < Gitlab::Database::Migration[2.0]
  disable_ddl_transaction!

  NEW_INDEX_NAME = 'idx_vulnerabilities_partial_devops_adoption_and_default_branch'
  OLD_INDEX_NAME = 'idx_vulnerabilities_partial_devops_adoption'

  def up
    add_concurrent_index :vulnerabilities, [:project_id, :created_at, :present_on_default_branch],
                         where: 'state != 1',
                         name: NEW_INDEX_NAME

    remove_concurrent_index_by_name(:vulnerabilities, OLD_INDEX_NAME)
  end

  def down
    add_concurrent_index :vulnerabilities, [:project_id, :created_at], where: 'state != 1', name: OLD_INDEX_NAME

    remove_concurrent_index_by_name(:vulnerabilities, NEW_INDEX_NAME)
  end
end