66 lines
1.7 KiB
Ruby
66 lines
1.7 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Ci
|
||
|
|
class AppendBuildTraceService
|
||
|
|
Result = Struct.new(:status, :stream_size, keyword_init: true)
|
||
|
|
TraceRangeError = Class.new(StandardError)
|
||
|
|
|
||
|
|
attr_reader :build, :params
|
||
|
|
|
||
|
|
def initialize(build, params)
|
||
|
|
@build = build
|
||
|
|
@params = params
|
||
|
|
end
|
||
|
|
|
||
|
|
def execute(body_data)
|
||
|
|
# TODO:
|
||
|
|
# it seems that `Content-Range` as formatted by runner is wrong,
|
||
|
|
# the `byte_end` should point to final byte, but it points byte+1
|
||
|
|
# that means that we have to calculate end of body,
|
||
|
|
# as we cannot use `content_length[1]`
|
||
|
|
# Issue: https://gitlab.com/gitlab-org/gitlab-runner/issues/3275
|
||
|
|
|
||
|
|
content_range = stream_range.split('-')
|
||
|
|
body_start = content_range[0].to_i
|
||
|
|
body_end = body_start + body_data.bytesize
|
||
|
|
|
||
|
|
stream_size = build.trace.append(body_data, body_start)
|
||
|
|
|
||
|
|
unless stream_size == body_end
|
||
|
|
log_range_error(stream_size, body_end)
|
||
|
|
|
||
|
|
return Result.new(status: 416, stream_size: stream_size)
|
||
|
|
end
|
||
|
|
|
||
|
|
Result.new(status: 202, stream_size: stream_size)
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def stream_range
|
||
|
|
params.fetch(:content_range)
|
||
|
|
end
|
||
|
|
|
||
|
|
def log_range_error(stream_size, body_end)
|
||
|
|
extra = {
|
||
|
|
build_id: build.id,
|
||
|
|
body_end: body_end,
|
||
|
|
stream_size: stream_size,
|
||
|
|
stream_class: stream_size.class,
|
||
|
|
stream_range: stream_range
|
||
|
|
}
|
||
|
|
|
||
|
|
build.trace_chunks.last.try do |chunk|
|
||
|
|
extra.merge!(
|
||
|
|
chunk_index: chunk.chunk_index,
|
||
|
|
chunk_store: chunk.data_store,
|
||
|
|
chunks_count: build.trace_chunks.count
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
::Gitlab::ErrorTracking
|
||
|
|
.log_exception(TraceRangeError.new, extra)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|