69 lines
1.7 KiB
Ruby
69 lines
1.7 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module Ci
|
||
|
|
class Trace
|
||
|
|
##
|
||
|
|
# RemoteChecksum class is responsible for fetching the MD5 checksum of
|
||
|
|
# an uploaded build trace.
|
||
|
|
#
|
||
|
|
class RemoteChecksum
|
||
|
|
include Gitlab::Utils::StrongMemoize
|
||
|
|
|
||
|
|
def initialize(trace_artifact)
|
||
|
|
@trace_artifact = trace_artifact
|
||
|
|
end
|
||
|
|
|
||
|
|
def md5_checksum
|
||
|
|
strong_memoize(:md5_checksum) do
|
||
|
|
fetch_md5_checksum
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
attr_reader :trace_artifact
|
||
|
|
delegate :aws?, :google?, to: :object_store_config, prefix: :provider
|
||
|
|
|
||
|
|
def fetch_md5_checksum
|
||
|
|
return unless Feature.enabled?(:ci_archived_build_trace_checksum, trace_artifact.project, default_enabled: :yaml)
|
||
|
|
return unless object_store_config.enabled?
|
||
|
|
return if trace_artifact.local_store?
|
||
|
|
|
||
|
|
remote_checksum_value
|
||
|
|
end
|
||
|
|
|
||
|
|
def remote_checksum_value
|
||
|
|
strong_memoize(:remote_checksum_value) do
|
||
|
|
if provider_google?
|
||
|
|
checksum_from_google
|
||
|
|
elsif provider_aws?
|
||
|
|
checksum_from_aws
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def object_store_config
|
||
|
|
strong_memoize(:object_store_config) do
|
||
|
|
trace_artifact.file.class.object_store_config
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def checksum_from_google
|
||
|
|
Base64.decode64(upload_attributes[:content_md5].to_s).unpack1('H*')
|
||
|
|
end
|
||
|
|
|
||
|
|
def checksum_from_aws
|
||
|
|
upload_attributes[:etag]
|
||
|
|
end
|
||
|
|
|
||
|
|
def upload_attributes
|
||
|
|
strong_memoize(:upload_attributes) do
|
||
|
|
trace_artifact.file.file.attributes
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|