From ef3ee17b1866269858b72fb11a2f9a84f264a228 Mon Sep 17 00:00:00 2001
From: Tianwen Chen <tchen@gitlab.com>
Date: Tue, 21 Feb 2023 12:13:35 +1100
Subject: [PATCH] Prepare async foreign key validation for ci_sources_pipelines

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

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

diff --git a/db/post_migrate/20230221010522_prepare_async_foreign_key_validation_for_ci_sources_pipelines.rb b/db/post_migrate/20230221010522_prepare_async_foreign_key_validation_for_ci_sources_pipelines.rb
new file mode 100644
index 00000000000000..e25bbca5c807db
--- /dev/null
+++ b/db/post_migrate/20230221010522_prepare_async_foreign_key_validation_for_ci_sources_pipelines.rb
@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+class PrepareAsyncForeignKeyValidationForCiSourcesPipelines < Gitlab::Database::Migration[2.1]
+  TABLE_NAME = :ci_sources_pipelines
+  COLUMN_NAMES = [:source_partition_id, :source_job_id]
+  FOREIGN_KEY_NAME = :fk_be5624bf37_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/20230221010522 b/db/schema_migrations/20230221010522
new file mode 100644
index 00000000000000..57727695f5ec49
--- /dev/null
+++ b/db/schema_migrations/20230221010522
@@ -0,0 +1 @@
+474f2b46179134270dc65c2b32ef9acfb01edb976f5efefe9ab49352d3fc390f
\ No newline at end of file
-- 
GitLab