From 486e29c86769f8e08bdb4d2e5d0fb3c4892132dd Mon Sep 17 00:00:00 2001
From: Tianwen Chen <tchen@gitlab.com>
Date: Thu, 16 Feb 2023 17:07:21 +1100
Subject: [PATCH] Prepare async foreign key validation for
 ci_build_trace_metadata

As titled, see https://gitlab.com/gitlab-org/gitlab/-/issues/391671

Changelog: changed
---
 ..._key_validation_for_ci_build_trace_metadata.rb | 15 +++++++++++++++
 db/schema_migrations/20230216060333               |  1 +
 2 files changed, 16 insertions(+)
 create mode 100644 db/post_migrate/20230216060333_prepare_async_foreign_key_validation_for_ci_build_trace_metadata.rb
 create mode 100644 db/schema_migrations/20230216060333

diff --git a/db/post_migrate/20230216060333_prepare_async_foreign_key_validation_for_ci_build_trace_metadata.rb b/db/post_migrate/20230216060333_prepare_async_foreign_key_validation_for_ci_build_trace_metadata.rb
new file mode 100644
index 00000000000000..1d5999acc61a88
--- /dev/null
+++ b/db/post_migrate/20230216060333_prepare_async_foreign_key_validation_for_ci_build_trace_metadata.rb
@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+class PrepareAsyncForeignKeyValidationForCiBuildTraceMetadata < Gitlab::Database::Migration[2.1]
+  TABLE_NAME = :ci_build_trace_metadata
+  COLUMN_NAMES = [:partition_id, :build_id]
+  FOREIGN_KEY_NAME = :fk_rails_aebc78111f_p
+
+  def up
+    prepare_async_foreign_key_validation(TABLE_NAME, COLUMN_NAMES, name: FOREIGN_KEY_NAME)
+  end
+
+  def down
+    unprepare_async_foreign_key_validation(TABLE_NAME, COLUMN_NAMES, name: FOREIGN_KEY_NAME)
+  end
+end
diff --git a/db/schema_migrations/20230216060333 b/db/schema_migrations/20230216060333
new file mode 100644
index 00000000000000..5f9fa26b1f0828
--- /dev/null
+++ b/db/schema_migrations/20230216060333
@@ -0,0 +1 @@
+c06d80c04bd661805f066412d750a651a0ead053cfff0e3314c03b3846a3f36d
\ No newline at end of file
-- 
GitLab