lib/gitlab/import_export/project/relation_saver.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

# frozen_string_literal: true

module Gitlab
  module ImportExport
    module Project
      class RelationSaver
        def initialize(project:, shared:, relation:)
          @project = project
          @relation = relation
          @shared = shared
        end

        def save
          if root_relation?
            serializer.serialize_root
          else
            serializer.serialize_relation(relation_schema)
          end

          true
        rescue StandardError => e
          shared.error(e)
          false
        end

        private

        attr_reader :project, :relation, :shared

        def serializer
          @serializer ||= ::Gitlab::ImportExport::Json::StreamingSerializer.new(
            project,
            reader.project_tree,
            json_writer,
            exportable_path: 'tree/project',
            current_user: nil
          )
        end

        def root_relation?
          relation == Projects::ImportExport::RelationExport::ROOT_RELATION
        end

        def relation_schema
          reader.project_tree[:include].find { |include| include[relation.to_sym] }
        end

        def reader
          @reader ||= ::Gitlab::ImportExport::Reader.new(shared: shared)
        end

        def json_writer
          @json_writer ||= ::Gitlab::ImportExport::Json::NdjsonWriter.new(shared.export_path)
        end
      end
    end
  end
end