Class: Gitlab::ImportExport::ImportFailureService

Inherits:
Object
  • Object
show all
Defined in:
lib/gitlab/import_export/import_failure_service.rb

Constant Summary collapse

RETRIABLE_EXCEPTIONS =
[GRPC::DeadlineExceeded, ActiveRecord::QueryCanceled].freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(importable) ⇒ ImportFailureService

Returns a new instance of ImportFailureService.


10
11
12
13
# File 'lib/gitlab/import_export/import_failure_service.rb', line 10

def initialize(importable)
  @importable = importable
  @association = importable.association(:import_failures)
end

Instance Attribute Details

#importableObject (readonly)

Returns the value of attribute importable


8
9
10
# File 'lib/gitlab/import_export/import_failure_service.rb', line 8

def importable
  @importable
end

Instance Method Details

#log_import_failure(source:, relation_key: nil, relation_index: nil, exception:, retry_count: 0) ⇒ Object


30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# File 'lib/gitlab/import_export/import_failure_service.rb', line 30

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

#with_retry(action:, relation_key: nil, relation_index: nil) ⇒ Object


15
16
17
18
19
20
21
22
23
24
25
26
27
28
# File 'lib/gitlab/import_export/import_failure_service.rb', line 15

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