Welcome to mirror list, hosted at ThFree Co, Russian Federation.

20230627115305_replace_p_ci_builds_metadata_foreign_key_v4.rb « post_migrate « db - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: cb5a5c96b3ce1bac50782da5ffd53a7d28c21ae6 (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
# frozen_string_literal: true

class ReplacePCiBuildsMetadataForeignKeyV4 < Gitlab::Database::Migration[2.1]
  include Gitlab::Database::MigrationHelpers::WraparoundAutovacuum

  disable_ddl_transaction!

  def up
    return unless should_run?
    return if foreign_key_exists?(:p_ci_builds_metadata, :p_ci_builds, name: :temp_fk_e20479742e_p)

    with_lock_retries do
      execute(<<~SQL.squish)
        LOCK TABLE ci_builds, p_ci_builds, p_ci_builds_metadata IN ACCESS EXCLUSIVE MODE;

        ALTER TABLE p_ci_builds_metadata
          ADD CONSTRAINT temp_fk_e20479742e_p
          FOREIGN KEY (partition_id, build_id)
          REFERENCES p_ci_builds (partition_id, id)
          ON  UPDATE CASCADE ON DELETE CASCADE;
      SQL
    end
  end

  def down
    return unless should_run?

    with_lock_retries do
      remove_foreign_key_if_exists :p_ci_builds_metadata, :p_ci_builds,
        name: :temp_fk_e20479742e_p,
        reverse_lock_order: true
    end
  end

  private

  def should_run?
    can_execute_on?(:ci_builds_metadata, :ci_builds)
  end
end