db/post_migrate/20230510013754_cleanup_notes_bigint_preparation.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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

# frozen_string_literal: true

class CleanupNotesBigintPreparation < Gitlab::Database::Migration[2.1]
  include Gitlab::Database::MigrationHelpers::ConvertToBigint

  disable_ddl_transaction!

  def up
    return unless should_run?

    remove_concurrent_index_by_name(
      :merge_request_user_mentions,
      :index_merge_request_user_mentions_note_id_convert_to_bigint,
      if_exists: true
    )

    remove_concurrent_index_by_name(
      :issue_user_mentions,
      :index_issue_user_mentions_on_note_id_convert_to_bigint,
      if_exists: true
    )

    with_lock_retries do
      remove_foreign_key_if_exists(
        :issue_user_mentions,
        :notes,
        name: :fk_issue_user_mentions_note_id_convert_to_bigint,
        reverse_lock_order: true
      )
    end

    with_lock_retries do
      remove_foreign_key_if_exists(
        :merge_request_user_mentions,
        :notes,
        name: :fk_merge_request_user_mentions_note_id_convert_to_bigint,
        reverse_lock_order: true
      )
    end
  end

  def down
    # No-op
  end

  private

  def should_run?
    com_or_dev_or_test_but_not_jh?
  end
end