1 files changed, 77 insertions, 0 deletions
diff --git a/lib/gitlab/ci/trace/archive.rb b/lib/gitlab/ci/trace/archive.rb
new file mode 100644
index 00000000000..5047cf04562
--- /dev/null
+++ b/lib/gitlab/ci/trace/archive.rb
@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Ci
+    class Trace
+      class Archive
+        include ::Gitlab::Utils::StrongMemoize
+        include Checksummable
+
+        def initialize(job, trace_metadata, metrics = ::Gitlab::Ci::Trace::Metrics.new)
+          @job = job
+          @trace_metadata = trace_metadata
+          @metrics = metrics
+        end
+
+        def execute!(stream)
+          clone_file!(stream, JobArtifactUploader.workhorse_upload_path) do |clone_path|
+            md5_checksum    = self.class.md5_hexdigest(clone_path)
+            sha256_checksum = self.class.sha256_hexdigest(clone_path)
+
+            job.transaction do
+              self.trace_artifact = create_build_trace!(clone_path, sha256_checksum)
+              trace_metadata.track_archival!(trace_artifact.id, md5_checksum)
+            end
+          end
+
+          validate_archived_trace
+        end
+
+        private
+
+        attr_reader :job, :trace_metadata, :metrics
+        attr_accessor :trace_artifact
+
+        def clone_file!(src_stream, temp_dir)
+          FileUtils.mkdir_p(temp_dir)
+          Dir.mktmpdir("tmp-trace-#{job.id}", temp_dir) do |dir_path|
+            temp_path = File.join(dir_path, "job.log")
+            FileUtils.touch(temp_path)
+            size = IO.copy_stream(src_stream, temp_path)
+            raise ::Gitlab::Ci::Trace::ArchiveError, 'Failed to copy stream' unless size == src_stream.size
+
+            yield(temp_path)
+          end
+        end
+
+        def create_build_trace!(path, file_sha256)
+          File.open(path) do |stream|
+            # TODO: Set `file_format: :raw` after we've cleaned up legacy traces migration
+            # https://gitlab.com/gitlab-org/gitlab-foss/merge_requests/20307
+            job.create_job_artifacts_trace!(
+              project: job.project,
+              file_type: :trace,
+              file: stream,
+              file_sha256: file_sha256)
+          end
+        end
+
+        def validate_archived_trace
+          return unless remote_checksum
+
+          trace_metadata.update!(remote_checksum: remote_checksum)
+
+          unless trace_metadata.remote_checksum_valid?
+            metrics.increment_error_counter(type: :archive_invalid_checksum)
+          end
+        end
+
+        def remote_checksum
+          strong_memoize(:remote_checksum) do
+            ::Gitlab::Ci::Trace::RemoteChecksum.new(trace_artifact).md5_checksum
+          end
+        end
+      end
+    end
+  end
+end