34 lines
845 B
Ruby
34 lines
845 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module ImportExport
|
||
|
|
module Project
|
||
|
|
module Sample
|
||
|
|
class RelationTreeRestorer < ImportExport::RelationTreeRestorer
|
||
|
|
def initialize(*args)
|
||
|
|
super
|
||
|
|
|
||
|
|
@date_calculator = Gitlab::ImportExport::Project::Sample::DateCalculator.new(dates)
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def relation_factory_params(*args)
|
||
|
|
super.merge(date_calculator: @date_calculator)
|
||
|
|
end
|
||
|
|
|
||
|
|
def dates
|
||
|
|
return [] if relation_reader.legacy?
|
||
|
|
|
||
|
|
RelationFactory::DATE_MODELS.flat_map do |tag|
|
||
|
|
relation_reader.consume_relation(@importable_path, tag, mark_as_consumed: false).map do |model|
|
||
|
|
model.first['due_date']
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|