discourse/db/migrate/20240826121506_copy_user_badges_notification_id_values.rb

# frozen_string_literal: true

class CopyUserBadgesNotificationIdValues < ActiveRecord::Migration[7.0]
  disable_ddl_transaction!

  def up
    min_id, max_id = execute("SELECT MIN(id), MAX(id) FROM user_badges")[0].values
    batch_size = 10_000

    (min_id..max_id).step(batch_size) { |start_id| execute <<~SQL.squish } if min_id && max_id
        UPDATE user_badges
        SET new_notification_id = notification_id
        WHERE id >= #{start_id} AND id < #{start_id + batch_size} AND notification_id IS NOT NULL AND new_notification_id IS NULL
      SQL
  end

  def down
    raise ActiveRecord::IrreversibleMigration
  end
end