db/post_migrate/20210825182303_remove_duplicate_dast_site_tokens_with_same_token.rb


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

# frozen_string_literal: true

class RemoveDuplicateDastSiteTokensWithSameToken < ActiveRecord::Migration[6.1]
  include Gitlab::Database::MigrationHelpers

  INDEX_NAME = 'index_dast_site_token_on_token'

  # rubocop: disable Migration/AddIndex
  def up
    execute("WITH duplicate_tokens AS(
                  SELECT id, rank() OVER (PARTITION BY token ORDER BY id) r FROM dast_site_tokens
                )
                DELETE FROM dast_site_tokens c USING duplicate_tokens t
                WHERE c.id = t.id AND t.r > 1;")

    add_index :dast_site_tokens, :token, name: INDEX_NAME, unique: true
  end

  # rubocop: disable Migration/RemoveIndex
  def down
    remove_index :dast_site_tokens, :token, name: INDEX_NAME
  end
end