24 lines
682 B
Ruby
24 lines
682 B
Ruby
# frozen_string_literal: true
|
|
|
|
class AddRequirementTestReportsForeignKey < ActiveRecord::Migration[6.1]
|
|
include Gitlab::Database::MigrationHelpers
|
|
|
|
disable_ddl_transaction!
|
|
|
|
TARGET_TABLE = :requirements_management_test_reports
|
|
CONSTRAINT_NAME = 'requirements_test_reports_requirement_id_xor_issue_id'
|
|
|
|
def up
|
|
add_concurrent_foreign_key TARGET_TABLE, :issues, column: :issue_id
|
|
|
|
add_check_constraint(TARGET_TABLE, 'num_nonnulls(requirement_id, issue_id) = 1', CONSTRAINT_NAME)
|
|
end
|
|
|
|
def down
|
|
remove_check_constraint TARGET_TABLE, CONSTRAINT_NAME
|
|
|
|
with_lock_retries do
|
|
remove_foreign_key_if_exists(TARGET_TABLE, column: :issue_id)
|
|
end
|
|
end
|
|
end
|