1 files changed, 46 insertions, 0 deletions
diff --git a/lib/bulk_imports/common/extractors/graphql_extractor.rb b/lib/bulk_imports/common/extractors/graphql_extractor.rb
new file mode 100644
index 00000000000..7d58032cfcc
--- /dev/null
+++ b/lib/bulk_imports/common/extractors/graphql_extractor.rb
@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module BulkImports
+  module Common
+    module Extractors
+      class GraphqlExtractor
+        def initialize(query)
+          @query = query[:query]
+          @query_string = @query.to_s
+          @variables = @query.variables
+        end
+
+        def extract(context)
+          @context = context
+
+          Enumerator.new do |yielder|
+            result = graphql_client.execute(parsed_query, query_variables(context.entity))
+
+            yielder << result.original_hash.deep_dup
+          end
+        end
+
+        private
+
+        def graphql_client
+          @graphql_client ||= BulkImports::Clients::Graphql.new(
+            url: @context.configuration.url,
+            token: @context.configuration.access_token
+          )
+        end
+
+        def parsed_query
+          @parsed_query ||= graphql_client.parse(@query.to_s)
+        end
+
+        def query_variables(entity)
+          return unless @variables
+
+          @variables.transform_values do |entity_attribute|
+            entity.public_send(entity_attribute) # rubocop:disable GitlabSecurity/PublicSend
+          end
+        end
+      end
+    end
+  end
+end