db/post_migrate/20230309020422_validate_fk_on_mr_user_mentions_note_id_convert_to_id_bigint.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

# frozen_string_literal: true

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

  TABLE_NAME = :merge_request_user_mentions
  COLUMN = :note_id_convert_to_bigint
  FK_NAME = :fk_merge_request_user_mentions_note_id_convert_to_bigint

  def up
    return unless should_run?

    prepare_async_foreign_key_validation TABLE_NAME, COLUMN, name: FK_NAME
  end

  def down
    return unless should_run?

    unprepare_async_foreign_key_validation TABLE_NAME, COLUMN, name: FK_NAME
  end

  private

  def should_run?
    com_or_dev_or_test_but_not_jh?
  end
end