You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
3.7 KiB
119 lines
3.7 KiB
require Rails.root.join('lib', 'mastodon', 'migration_helpers') |
|
|
|
class IdsToBigints < ActiveRecord::Migration[5.1] |
|
include Mastodon::MigrationHelpers |
|
|
|
disable_ddl_transaction! |
|
|
|
INCLUDED_COLUMNS = [ |
|
[: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], |
|
] |
|
INCLUDED_COLUMNS << [:deprecated_preview_cards, :id] if table_exists?(:deprecated_preview_cards) |
|
|
|
def migrate_columns(to_type) |
|
# Print out a warning that this will probably take a while. |
|
say '' |
|
say 'WARNING: This migration may take a *long* time for large instances' |
|
say 'It will *not* lock tables for any significant time, but it may run' |
|
say 'for a very long time. We will pause for 10 seconds to allow you to' |
|
say 'interrupt this migration if you are not ready.' |
|
say '' |
|
say 'This migration has some sections that can be safely interrupted' |
|
say 'and restarted later, and will tell you when those are occurring.' |
|
say '' |
|
say 'For more information, see https://github.com/tootsuite/mastodon/pull/5088' |
|
|
|
10.downto(1) do |i| |
|
say "Continuing in #{i} second#{i == 1 ? '' : 's'}...", true |
|
sleep 1 |
|
end |
|
|
|
tables = INCLUDED_COLUMNS.map(&:first).uniq |
|
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 up |
|
migrate_columns(:bigint) |
|
end |
|
|
|
def down |
|
migrate_columns(:integer) |
|
end |
|
end
|
|
|