summaryrefslogtreecommitdiff
path: root/db/post_migrate/20210824174615_prepare_ci_builds_metadata_and_ci_build_async_indexes.rb
blob: 0a0fda7e8708447143e46cd818651b5f47b96500 (plain)
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
# frozen_string_literal: true

class PrepareCiBuildsMetadataAndCiBuildAsyncIndexes < ActiveRecord::Migration[6.1]
  include Gitlab::Database::MigrationHelpers

  def up
    prepare_async_index :ci_builds_metadata, :id_convert_to_bigint, unique: true,
      name: :index_ci_builds_metadata_on_id_convert_to_bigint

    prepare_async_index :ci_builds_metadata, :build_id_convert_to_bigint, unique: true,
      name: :index_ci_builds_metadata_on_build_id_convert_to_bigint

    prepare_async_index :ci_builds_metadata, :build_id_convert_to_bigint, where: 'has_exposed_artifacts IS TRUE',
      name: :index_ci_builds_metadata_on_build_id_int8_and_exposed_artifacts

    prepare_async_index_from_sql(:ci_builds_metadata, :index_ci_builds_metadata_on_build_id_int8_where_interruptible, <<~SQL.squish)
        CREATE INDEX CONCURRENTLY "index_ci_builds_metadata_on_build_id_int8_where_interruptible"
        ON "ci_builds_metadata" ("build_id_convert_to_bigint") INCLUDE ("id_convert_to_bigint")
        WHERE interruptible = true
    SQL

    prepare_async_index :ci_builds, :id_convert_to_bigint, unique: true,
      name: :index_ci_builds_on_converted_id
  end

  def down
    unprepare_async_index_by_name :ci_builds, :index_ci_builds_on_converted_id

    unprepare_async_index_by_name :ci_builds_metadata, :index_ci_builds_metadata_on_build_id_int8_where_interruptible

    unprepare_async_index_by_name :ci_builds_metadata, :index_ci_builds_metadata_on_build_id_int8_and_exposed_artifacts

    unprepare_async_index_by_name :ci_builds_metadata, :index_ci_builds_metadata_on_build_id_convert_to_bigint

    unprepare_async_index_by_name :ci_builds_metadata, :index_ci_builds_metadata_on_id_convert_to_bigint
  end

  private

  def prepare_async_index_from_sql(table_name, index_name, definition)
    return unless async_index_creation_available?

    return if index_name_exists?(table_name, index_name)

    async_index = Gitlab::Database::AsyncIndexes::PostgresAsyncIndex.safe_find_or_create_by!(name: index_name) do |rec|
      rec.table_name = table_name
      rec.definition = definition
    end

    Gitlab::AppLogger.info(
      message: 'Prepared index for async creation',
      table_name: async_index.table_name,
      index_name: async_index.name)
  end
end