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.
20 lines
503 B
20 lines
503 B
# frozen_string_literal: true
|
|
|
|
require_relative '../mastodon/snowflake'
|
|
|
|
module ActiveRecord
|
|
module Tasks
|
|
module DatabaseTasks
|
|
original_load_schema = instance_method(:load_schema)
|
|
|
|
define_method(:load_schema) do |db_config, *args|
|
|
ActiveRecord::Base.establish_connection(db_config)
|
|
Mastodon::Snowflake.define_timestamp_id
|
|
|
|
original_load_schema.bind(self).call(db_config, *args)
|
|
|
|
Mastodon::Snowflake.ensure_id_sequences_exist
|
|
end
|
|
end
|
|
end
|
|
end
|