mirror of
https://github.com/mastodon/mastodon.git
synced 2024-08-20 21:08:15 -07:00
20 lines
503 B
Ruby
20 lines
503 B
Ruby
# 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
|