1 files changed, 80 insertions, 1 deletions
diff --git a/lib/bulk_imports/pipeline.rb b/lib/bulk_imports/pipeline.rb
index 70e6030ea2c..a44f8fc7193 100644
--- a/lib/bulk_imports/pipeline.rb
+++ b/lib/bulk_imports/pipeline.rb
@@ -3,10 +3,89 @@
 module BulkImports
   module Pipeline
     extend ActiveSupport::Concern
+    include Gitlab::ClassAttributes
 
     included do
-      include Attributes
       include Runner
+
+      private
+
+      def extractors
+        @extractors ||= self.class.extractors.map(&method(:instantiate))
+      end
+
+      def transformers
+        @transformers ||= self.class.transformers.map(&method(:instantiate))
+      end
+
+      def loaders
+        @loaders ||= self.class.loaders.map(&method(:instantiate))
+      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)
+        add_attribute(:extractors, klass, options)
+      end
+
+      def transformer(klass, options = nil)
+        add_attribute(:transformers, klass, options)
+      end
+
+      def loader(klass, options = nil)
+        add_attribute(:loaders, klass, options)
+      end
+
+      def after_run(&block)
+        class_attributes[:after_run] = block
+      end
+
+      def extractors
+        class_attributes[:extractors]
+      end
+
+      def transformers
+        class_attributes[:transformers]
+      end
+
+      def loaders
+        class_attributes[:loaders]
+      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