Add latest changes from gitlab-org/gitlab@16-0-stable-eev16.0.0-rc42

1 files changed, 66 insertions, 0 deletions

diff --git a/db/post_migrate/20230306143322_prepare_ci_builds_primary_key_for_partitioning.rb b/db/post_migrate/20230306143322_prepare_ci_builds_primary_key_for_partitioning.rb
new file mode 100644
index 00000000000..99b52e4c443
--- /dev/null
+++ b/db/post_migrate/20230306143322_prepare_ci_builds_primary_key_for_partitioning.rb
@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+class PrepareCiBuildsPrimaryKeyForPartitioning < Gitlab::Database::Migration[2.1]
+  include Gitlab::Database::PartitioningMigrationHelpers
+
+  disable_ddl_transaction!
+
+  TABLE_NAME = :ci_builds
+  PRIMARY_KEY = :ci_builds_pkey
+  NEW_INDEX_NAME = :index_ci_builds_on_id_partition_id_unique
+  OLD_INDEX_NAME = :index_ci_builds_on_id_unique
+
+  def up
+    swap_primary_key(TABLE_NAME, PRIMARY_KEY, NEW_INDEX_NAME)
+  end
+
+  def down
+    add_concurrent_index(TABLE_NAME, :id, unique: true, name: OLD_INDEX_NAME)
+    add_concurrent_index(TABLE_NAME, [:id, :partition_id], unique: true, name: NEW_INDEX_NAME)
+
+    unswap_primary_key(TABLE_NAME, PRIMARY_KEY, OLD_INDEX_NAME)
+
+    recreate_partitioned_foreign_keys
+  end
+
+  private
+
+  def recreate_partitioned_foreign_keys
+    add_partitioned_fk(:ci_job_variables, :fk_rails_fbf3b34792_p, column: :job_id)
+    add_partitioned_fk(:ci_job_artifacts, :fk_rails_c5137cb2c1_p, column: :job_id)
+    add_partitioned_fk(:ci_running_builds, :fk_rails_da45cfa165_p)
+    add_partitioned_fk(:ci_build_pending_states, :fk_861cd17da3_p)
+    add_partitioned_fk(:ci_build_trace_chunks, :fk_89e29fa5ee_p)
+    add_partitioned_fk(:ci_unit_test_failures, :fk_9e0fc58930_p)
+    add_partitioned_fk(:ci_build_trace_metadata, :fk_rails_aebc78111f_p)
+    add_partitioned_fk(:ci_pending_builds, :fk_rails_725a2644a3_p)
+    add_partitioned_fk(:ci_builds_runner_session, :fk_rails_70707857d3_p)
+    add_partitioned_fk(:ci_build_needs, :fk_rails_3cf221d4ed_p)
+    add_partitioned_fk(:ci_build_report_results, :fk_rails_16cb1ff064_p)
+    add_partitioned_fk(:ci_resources, :fk_e169a8e3d5_p, delete: :nullify)
+    add_partitioned_fk(:ci_sources_pipelines, :fk_be5624bf37_p, columns: %i[source_partition_id source_job_id])
+
+    add_routing_table_fk(:p_ci_builds_metadata, :fk_e20479742e_p)
+    add_routing_table_fk(:p_ci_runner_machine_builds, :fk_bb490f12fe_p)
+  end
+
+  def add_partitioned_fk(source_table, name, column: :build_id, columns: nil, delete: :cascade)
+    add_concurrent_foreign_key(source_table, :ci_builds,
+      column: columns || [:partition_id, column],
+      target_column: [:partition_id, :id],
+      reverse_lock_order: true,
+      on_update: :cascade,
+      on_delete: delete,
+      name: name)
+  end
+
+  def add_routing_table_fk(source_table, name)
+    add_concurrent_partitioned_foreign_key(source_table, :ci_builds,
+      column: [:partition_id, :build_id],
+      target_column: [:partition_id, :id],
+      reverse_lock_order: true,
+      on_update: :cascade,
+      on_delete: :cascade,
+      name: name)
+  end
+end