about summary refs log tree commit diff
path: root/db/migrate/20210306164523_account_ids_to_timestamp_ids.rb
blob: 39cd4cdea4f11248d2c9ed7864c38f673a6dfeef (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class AccountIdsToTimestampIds < ActiveRecord::Migration[5.1]
  def up
    # Set up the accounts.id column to use our timestamp-based IDs.
    safety_assured do
      execute("ALTER TABLE accounts ALTER COLUMN id SET DEFAULT timestamp_id('accounts')")
    end

    # Make sure we have a sequence to use.
    Mastodon::Snowflake.ensure_id_sequences_exist
  end

  def down
    execute("LOCK accounts")
    execute("SELECT setval('accounts_id_seq', (SELECT MAX(id) FROM accounts))")
    execute("ALTER TABLE accounts ALTER COLUMN id SET DEFAULT nextval('accounts_id_seq')")
  end
end