92 lines
1.9 KiB
Ruby
92 lines
1.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module BulkImports
|
|
module Pipeline
|
|
extend ActiveSupport::Concern
|
|
include Gitlab::ClassAttributes
|
|
|
|
included do
|
|
include Runner
|
|
|
|
private
|
|
|
|
def extractor
|
|
@extractor ||= instantiate(self.class.get_extractor)
|
|
end
|
|
|
|
def transformers
|
|
@transformers ||= self.class.transformers.map(&method(:instantiate))
|
|
end
|
|
|
|
def loader
|
|
@loaders ||= instantiate(self.class.get_loader)
|
|
end
|
|
|
|
def after_run
|
|
@after_run ||= self.class.after_run_callback
|
|
end
|
|
|
|
def pipeline
|
|
@pipeline ||= self.class.name
|
|
end
|
|
|
|
def instantiate(class_config)
|
|
class_config[:klass].new(class_config[:options])
|
|
end
|
|
|
|
def abort_on_failure?
|
|
self.class.abort_on_failure?
|
|
end
|
|
end
|
|
|
|
class_methods do
|
|
def extractor(klass, options = nil)
|
|
class_attributes[:extractor] = { klass: klass, options: options }
|
|
end
|
|
|
|
def transformer(klass, options = nil)
|
|
add_attribute(:transformers, klass, options)
|
|
end
|
|
|
|
def loader(klass, options = nil)
|
|
class_attributes[:loader] = { klass: klass, options: options }
|
|
end
|
|
|
|
def after_run(&block)
|
|
class_attributes[:after_run] = block
|
|
end
|
|
|
|
def get_extractor
|
|
class_attributes[:extractor]
|
|
end
|
|
|
|
def transformers
|
|
class_attributes[:transformers]
|
|
end
|
|
|
|
def get_loader
|
|
class_attributes[:loader]
|
|
end
|
|
|
|
def after_run_callback
|
|
class_attributes[:after_run]
|
|
end
|
|
|
|
def abort_on_failure!
|
|
class_attributes[:abort_on_failure] = true
|
|
end
|
|
|
|
def abort_on_failure?
|
|
class_attributes[:abort_on_failure]
|
|
end
|
|
|
|
private
|
|
|
|
def add_attribute(sym, klass, options)
|
|
class_attributes[sym] ||= []
|
|
class_attributes[sym] << { klass: klass, options: options }
|
|
end
|
|
end
|
|
end
|
|
end
|