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

1 files changed, 29 insertions, 0 deletions

diff --git a/db/post_migrate/20230317080137_add_unique_packages_index_when_debian.rb b/db/post_migrate/20230317080137_add_unique_packages_index_when_debian.rb
new file mode 100644
index 00000000000..b079e860c0f
--- /dev/null
+++ b/db/post_migrate/20230317080137_add_unique_packages_index_when_debian.rb
@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+class AddUniquePackagesIndexWhenDebian < Gitlab::Database::Migration[2.1]
+  TABLE_NAME = :packages_packages
+  PACKAGE_TYPE_DEBIAN = 9
+  PACKAGE_STATUS_PENDING_DESTRUCTION = 4
+  TMP_DEBIAN_UNIQUE_INDEX_NAME = 'tmp_unique_packages_project_id_and_name_and_version_when_debian'
+  DEBIAN_UNIQUE_INDEX_NAME = 'unique_packages_project_id_and_name_and_version_when_debian'
+
+  disable_ddl_transaction!
+
+  def up
+    add_concurrent_index TABLE_NAME, [:project_id, :name, :version],
+      where: "package_type = #{PACKAGE_TYPE_DEBIAN} AND status != #{PACKAGE_STATUS_PENDING_DESTRUCTION}",
+      unique: true,
+      name: DEBIAN_UNIQUE_INDEX_NAME
+    remove_concurrent_index_by_name TABLE_NAME, TMP_DEBIAN_UNIQUE_INDEX_NAME
+  end
+
+  def down
+    # This index will disallow further duplicates while we're deduplicating the data.
+    add_concurrent_index TABLE_NAME, [:project_id, :name, :version],
+      where: "package_type = #{PACKAGE_TYPE_DEBIAN} AND status != #{PACKAGE_STATUS_PENDING_DESTRUCTION} AND
+        created_at > TIMESTAMP WITH TIME ZONE '#{Time.now.utc}'",
+      unique: true,
+      name: TMP_DEBIAN_UNIQUE_INDEX_NAME
+    remove_concurrent_index_by_name TABLE_NAME, DEBIAN_UNIQUE_INDEX_NAME
+  end
+end