instrumental/db/migrate/20170918125918_ids_to_bigints.rb

# frozen_string_literal: true

require_relative '../../lib/mastodon/migration_helpers'
require_relative '../../lib/mastodon/migration_warning'

class IdsToBigints < ActiveRecord::Migration[5.1]
  include Mastodon::MigrationHelpers
  include Mastodon::MigrationWarning

  TABLE_COLUMN_MAPPING = [
    [:account_domain_blocks, :account_id],
    [:account_domain_blocks, :id],
    [:accounts, :id],
    [:blocks, :account_id],
    [:blocks, :id],
    [:blocks, :target_account_id],
    [:conversation_mutes, :account_id],
    [:conversation_mutes, :id],
    [:domain_blocks, :id],
    [:favourites, :account_id],
    [:favourites, :id],
    [:favourites, :status_id],
    [:follow_requests, :account_id],
    [:follow_requests, :id],
    [:follow_requests, :target_account_id],
    [:follows, :account_id],
    [:follows, :id],
    [:follows, :target_account_id],
    [:imports, :account_id],
    [:imports, :id],
    [:media_attachments, :account_id],
    [:media_attachments, :id],
    [:mentions, :account_id],
    [:mentions, :id],
    [:mutes, :account_id],
    [:mutes, :id],
    [:mutes, :target_account_id],
    [:notifications, :account_id],
    [:notifications, :from_account_id],
    [:notifications, :id],
    [:oauth_access_grants, :application_id],
    [:oauth_access_grants, :id],
    [:oauth_access_grants, :resource_owner_id],
    [:oauth_access_tokens, :application_id],
    [:oauth_access_tokens, :id],
    [:oauth_access_tokens, :resource_owner_id],
    [:oauth_applications, :id],
    [:oauth_applications, :owner_id],
    [:reports, :account_id],
    [:reports, :action_taken_by_account_id],
    [:reports, :id],
    [:reports, :target_account_id],
    [:session_activations, :access_token_id],
    [:session_activations, :user_id],
    [:session_activations, :web_push_subscription_id],
    [:settings, :id],
    [:settings, :thing_id],
    [:statuses, :account_id],
    [:statuses, :application_id],
    [:statuses, :in_reply_to_account_id],
    [:stream_entries, :account_id],
    [:stream_entries, :id],
    [:subscriptions, :account_id],
    [:subscriptions, :id],
    [:tags, :id],
    [:users, :account_id],
    [:users, :id],
    [:web_settings, :id],
    [:web_settings, :user_id],
  ].freeze

  disable_ddl_transaction!

  def migrate_columns(to_type)
    display_warning

    table_sizes = {}

    # Sort tables by their size
    tables.each do |table|
      table_sizes[table] = estimate_rows_in_table(table)
    end

    ordered_columns = included_columns.sort_by do |col_parts|
      [-table_sizes[col_parts.first], col_parts.last]
    end

    ordered_columns.each do |column_parts|
      table, column = column_parts

      # Skip this if we're resuming and already did this one.
      next if column_for(table, column).sql_type == to_type.to_s

      change_column_type_concurrently table, column, to_type
      cleanup_concurrent_column_type_change table, column
    end
  end

  def display_warning
    migration_duration_warning(<<~EXPLANATION)
      This migration has some sections that can be safely interrupted
      and restarted later, and will tell you when those are occurring.

      For more information, see https://github.com/mastodon/mastodon/pull/5088
    EXPLANATION
  end

  def tables
    included_columns.map(&:first).uniq
  end

  def included_columns
    TABLE_COLUMN_MAPPING.dup.tap do |included_columns|
      included_columns << [:deprecated_preview_cards, :id] if table_exists?(:deprecated_preview_cards)
    end
  end

  def up
    migrate_columns(:bigint)
  end

  def down
    migrate_columns(:integer)
  end
end