mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-05-28 09:05:50 +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 ```
90 lines
2.1 KiB
Ruby
Vendored
90 lines
2.1 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
module Migrations::Converters::Base
|
|
class Converter
|
|
attr_accessor :settings
|
|
|
|
def initialize(settings)
|
|
@settings = settings
|
|
end
|
|
|
|
def run(only_steps: [], skip_steps: [])
|
|
if respond_to?(:setup)
|
|
puts "Initializing..."
|
|
setup
|
|
end
|
|
|
|
create_database
|
|
|
|
filter_steps(steps, only_steps, skip_steps).each do |step_class|
|
|
step = create_step(step_class)
|
|
before_step_execution(step)
|
|
execute_step(step)
|
|
after_step_execution(step)
|
|
end
|
|
rescue SignalException
|
|
STDERR.puts "\nAborted"
|
|
exit(1)
|
|
ensure
|
|
::Migrations::Database::IntermediateDB.close
|
|
end
|
|
|
|
def steps
|
|
step_class = ::Migrations::Converters::Base::Step
|
|
current_module = self.class.name.deconstantize.constantize
|
|
|
|
current_module
|
|
.constants
|
|
.map { |c| current_module.const_get(c) }
|
|
.select { |klass| klass.is_a?(Class) && klass < step_class }
|
|
.sort_by(&:to_s)
|
|
end
|
|
|
|
def before_step_execution(step)
|
|
# do nothing
|
|
end
|
|
|
|
def execute_step(step)
|
|
executor =
|
|
if step.is_a?(ProgressStep)
|
|
ProgressStepExecutor
|
|
else
|
|
StepExecutor
|
|
end
|
|
|
|
executor.new(step).execute
|
|
end
|
|
|
|
def after_step_execution(step)
|
|
# do nothing
|
|
end
|
|
|
|
def step_args(step_class)
|
|
{}
|
|
end
|
|
|
|
private
|
|
|
|
def create_database
|
|
db_path = File.expand_path(settings[:intermediate_db][:path], ::Migrations.root_path)
|
|
::Migrations::Database.migrate(
|
|
db_path,
|
|
migrations_path: ::Migrations::Database::INTERMEDIATE_DB_SCHEMA_PATH,
|
|
)
|
|
|
|
db = ::Migrations::Database.connect(db_path)
|
|
::Migrations::Database::IntermediateDB.setup(db)
|
|
end
|
|
|
|
def create_step(step_class)
|
|
default_args = { settings: }
|
|
|
|
args = default_args.merge(step_args(step_class))
|
|
step_class.new(StepTracker.new, args)
|
|
end
|
|
|
|
def filter_steps(step_classes, only_steps, skip_steps)
|
|
::Migrations::ClassFilter.filter(step_classes, only: only_steps, skip: skip_steps)
|
|
end
|
|
end
|
|
end
|
