# frozen_string_literal: true

class AddIndexIdPartitionIdToCiBuild < Gitlab::Database::Migration[2.1]
  disable_ddl_transaction!

  TABLE_NAME = :ci_builds
  INDEX_NAME = :index_ci_builds_on_id_partition_id_unique
  COLUMNS = %i[id partition_id].freeze

  def up
    add_concurrent_index(TABLE_NAME, COLUMNS, unique: true, name: INDEX_NAME)
  end

  def down
    remove_concurrent_index_by_name(TABLE_NAME, INDEX_NAME)
  end
end