1 files changed, 41 insertions, 0 deletions
diff --git a/lib/bulk_imports/pipeline/attributes.rb b/lib/bulk_imports/pipeline/attributes.rb
new file mode 100644
index 00000000000..ebfbaf6f6ba
--- /dev/null
+++ b/lib/bulk_imports/pipeline/attributes.rb
@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module BulkImports
+  module Pipeline
+    module Attributes
+      extend ActiveSupport::Concern
+      include Gitlab::ClassAttributes
+
+      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 add_attribute(sym, klass, options)
+          class_attributes[sym] ||= []
+          class_attributes[sym] << { klass: klass, options: options }
+        end
+
+        def extractors
+          class_attributes[:extractors]
+        end
+
+        def transformers
+          class_attributes[:transformers]
+        end
+
+        def loaders
+          class_attributes[:loaders]
+        end
+      end
+    end
+  end
+end