db/post_migrate/20230117113719_add_tmp_index_to_ci_sources_pipelines_on_partition_id.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

# frozen_string_literal: true

class AddTmpIndexToCiSourcesPipelinesOnPartitionId < Gitlab::Database::Migration[2.1]
  disable_ddl_transaction!

  INDEX_NAME = :tmp_index_ci_sources_pipelines_on_partition_id_and_id
  SOURCE_INDEX_NAME = :tmp_index_ci_sources_pipelines_on_source_partition_id_and_id
  TABLE_NAME = :ci_sources_pipelines

  def up
    return unless Gitlab.com?

    add_concurrent_index TABLE_NAME,
      [:partition_id, :id],
      name: INDEX_NAME, where: 'partition_id = 101'

    add_concurrent_index TABLE_NAME,
      [:source_partition_id, :id],
      name: SOURCE_INDEX_NAME,
      where: 'source_partition_id = 101'
  end

  def down
    return unless Gitlab.com?

    remove_concurrent_index_by_name TABLE_NAME, INDEX_NAME
    remove_concurrent_index_by_name TABLE_NAME, SOURCE_INDEX_NAME
  end
end