db/post_migrate/20230515142300_add_unique_index_for_ml_model_packages.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

# frozen_string_literal: true

class AddUniqueIndexForMlModelPackages < Gitlab::Database::Migration[2.1]
  INDEX_NAME = 'uniq_idx_packages_packages_on_project_id_name_version_ml_model'
  PACKAGE_TYPE_ML_MODEL = 14

  disable_ddl_transaction!

  def up
    add_concurrent_index :packages_packages, [:project_id, :name, :version],
      unique: true,
      where: "package_type = #{PACKAGE_TYPE_ML_MODEL}",
      name: INDEX_NAME
  end

  def down
    remove_concurrent_index_by_name(:packages_packages, INDEX_NAME)
  end
end