mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-05-24 16:41:21 +08:00
58558849ac
Examples: * Run only the "badges" and "users" steps in the converter: ``` migrations/bin/cli convert discourse --reset --only=badges,users ``` * Run all steps except "users" in the converter: ``` migrations/bin/cli convert discourse --reset --skip=users ``` * Run only the "badges" step in importer: This also runs all steps that "badges" depends on (e.g. "users" and "uploads"). ``` migrations/bin/cli import --reset --only=badges ``` * Run only the "badges" step in importer and skip the "users" on which "badges" depends: ``` migrations/bin/cli import --reset --only=badges --skip=users ```
78 lines
2.2 KiB
Ruby
Vendored
78 lines
2.2 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
module Migrations::Importer
|
|
class Executor
|
|
def initialize(config, options)
|
|
@intermediate_db = ::Migrations::Database.connect(config[:intermediate_db])
|
|
@discourse_db = DiscourseDB.new
|
|
@shared_data = SharedData.new(@discourse_db)
|
|
@options = options
|
|
|
|
attach_mappings_db(config[:mappings_db], options[:reset])
|
|
attach_uploads_db(config[:uploads_db])
|
|
end
|
|
|
|
def start
|
|
runtime =
|
|
::Migrations::DateHelper.track_time do
|
|
optimize_intermediate_db
|
|
execute_steps
|
|
ensure
|
|
cleanup
|
|
end
|
|
|
|
puts I18n.t("importer.done", runtime: ::Migrations::DateHelper.human_readable_time(runtime))
|
|
end
|
|
|
|
private
|
|
|
|
def attach_mappings_db(db_path, reset)
|
|
::Migrations::Database.reset!(db_path) if reset
|
|
migrate_and_attach(db_path, ::Migrations::Database::MAPPINGS_DB_SCHEMA_PATH, "mapped")
|
|
end
|
|
|
|
def attach_uploads_db(db_path)
|
|
migrate_and_attach(db_path, ::Migrations::Database::UPLOADS_DB_SCHEMA_PATH, "files")
|
|
end
|
|
|
|
def migrate_and_attach(db_path, schema_path, alias_name)
|
|
::Migrations::Database.migrate(db_path, migrations_path: schema_path)
|
|
@intermediate_db.execute("ATTACH DATABASE ? AS #{alias_name}", db_path)
|
|
end
|
|
|
|
def optimize_intermediate_db
|
|
@intermediate_db.execute("PRAGMA optimize=0x10002")
|
|
end
|
|
|
|
def step_classes
|
|
steps_module = ::Migrations::Importer::Steps
|
|
classes =
|
|
steps_module
|
|
.constants
|
|
.map { |c| steps_module.const_get(c) }
|
|
.select { |klass| klass.is_a?(Class) && klass < ::Migrations::Importer::Step }
|
|
|
|
filtered_classes =
|
|
::Migrations::ClassFilter.filter(classes, only: @options[:only], skip: @options[:skip])
|
|
::Migrations::TopologicalSorter.sort(filtered_classes)
|
|
end
|
|
|
|
def execute_steps
|
|
max = step_classes.size
|
|
|
|
step_classes
|
|
.each
|
|
.with_index(1) do |step_class, index|
|
|
puts "#{step_class.title} [#{index}/#{max}]"
|
|
step = step_class.new(@intermediate_db, @discourse_db, @shared_data)
|
|
step.execute
|
|
puts ""
|
|
end
|
|
end
|
|
|
|
def cleanup
|
|
@intermediate_db.close
|
|
@discourse_db.close
|
|
end
|
|
end
|
|
end
|
