gitlab-org--gitlab-foss/db/post_migrate/20210303165302_cleanup_clus...

26 lines
573 B
Ruby

# frozen_string_literal: true
class CleanupClusterTokensWithNullName < ActiveRecord::Migration[6.0]
include Gitlab::Database::MigrationHelpers
BATCH_SIZE = 1000
disable_ddl_transaction!
class AgentToken < ActiveRecord::Base
include EachBatch
self.table_name = 'cluster_agent_tokens'
end
def up
AgentToken.each_batch(of: BATCH_SIZE) do |relation|
relation.where(name: nil).update_all("name = 'agent-token-' || id")
end
end
def down
# no-op : can't go back to `NULL` without first dropping the `NOT NULL` constraint
end
end