58 lines
1.6 KiB
Ruby
58 lines
1.6 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
||
|
module Gitlab
|
||
|
module ImportExport
|
||
|
class ImportFailureService
|
||
|
RETRIABLE_EXCEPTIONS = [GRPC::DeadlineExceeded, ActiveRecord::QueryCanceled].freeze
|
||
|
|
||
|
attr_reader :importable
|
||
|
|
||
|
def initialize(importable)
|
||
|
@importable = importable
|
||
|
@association = importable.association(:import_failures)
|
||
|
end
|
||
|
|
||
|
def with_retry(action:, relation_key: nil, relation_index: nil)
|
||
|
on_retry = -> (exception, retry_count, *_args) do
|
||
|
log_import_failure(
|
||
|
source: action,
|
||
|
relation_key: relation_key,
|
||
|
relation_index: relation_index,
|
||
|
exception: exception,
|
||
|
retry_count: retry_count)
|
||
|
end
|
||
|
|
||
|
Retriable.with_context(:relation_import, on_retry: on_retry) do
|
||
|
yield
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def log_import_failure(source:, relation_key: nil, relation_index: nil, exception:, retry_count: 0)
|
||
|
extra = {
|
||
|
source: source,
|
||
|
relation_key: relation_key,
|
||
|
relation_index: relation_index,
|
||
|
retry_count: retry_count
|
||
|
}
|
||
|
extra[importable_column_name] = importable.id
|
||
|
|
||
|
Gitlab::ErrorTracking.track_exception(exception, extra)
|
||
|
|
||
|
attributes = {
|
||
|
exception_class: exception.class.to_s,
|
||
|
exception_message: exception.message.truncate(255),
|
||
|
correlation_id_value: Labkit::Correlation::CorrelationId.current_or_new_id
|
||
|
}.merge(extra)
|
||
|
|
||
|
ImportFailure.create(attributes)
|
||
|
end
|
||
|
|
||
|
private
|
||
|
|
||
|
def importable_column_name
|
||
|
@importable_column_name ||= @association.reflection.foreign_key.to_sym
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
end
|