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

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

  INDEX_NAME = 'index_vulnerability_reads_on_location_image_trigram'
  REPORT_TYPES = { container_scanning: 2, cluster_image_scanning: 7 }.freeze

  def up
    remove_concurrent_index_by_name :vulnerability_reads, INDEX_NAME

    prepare_async_index :vulnerability_reads, :location_image,
      name: INDEX_NAME,
      using: :gin, opclass: { location_image: :gin_trgm_ops },
      where: "report_type = ANY (ARRAY[#{REPORT_TYPES.values.join(', ')}]) AND location_image IS NOT NULL"
  end

  def down
    unprepare_async_index :vulnerability_reads, :location_image, name: INDEX_NAME
  end
end