1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
# frozen_string_literal: true
module BulkImports
class EntityWorker # rubocop:disable Scalability/IdempotentWorker
include ApplicationWorker
idempotent!
deduplicate :until_executing
data_consistency :always
feature_category :importers
sidekiq_options retry: false, dead: false
worker_has_external_dependencies!
def perform(entity_id, current_stage = nil)
if stage_running?(entity_id, current_stage)
logger.info(
structured_payload(
bulk_import_entity_id: entity_id,
bulk_import_id: bulk_import_id(entity_id),
current_stage: current_stage,
message: 'Stage running',
importer: 'gitlab_migration'
)
)
return
end
logger.info(
structured_payload(
bulk_import_entity_id: entity_id,
bulk_import_id: bulk_import_id(entity_id),
current_stage: current_stage,
message: 'Stage starting',
importer: 'gitlab_migration'
)
)
next_pipeline_trackers_for(entity_id).each do |pipeline_tracker|
BulkImports::PipelineWorker.perform_async(
pipeline_tracker.id,
pipeline_tracker.stage,
entity_id
)
end
rescue StandardError => e
logger.error(
structured_payload(
bulk_import_entity_id: entity_id,
bulk_import_id: bulk_import_id(entity_id),
current_stage: current_stage,
message: e.message,
importer: 'gitlab_migration'
)
)
Gitlab::ErrorTracking.track_exception(
e, bulk_import_entity_id: entity_id, bulk_import_id: bulk_import_id(entity_id), importer: 'gitlab_migration'
)
end
private
def stage_running?(entity_id, stage)
return unless stage
BulkImports::Tracker.stage_running?(entity_id, stage)
end
def next_pipeline_trackers_for(entity_id)
BulkImports::Tracker.next_pipeline_trackers_for(entity_id).update(status_event: 'enqueue')
end
def bulk_import_id(entity_id)
@bulk_import_id ||= Entity.find(entity_id).bulk_import_id
end
def logger
@logger ||= Gitlab::Import::Logger.build
end
end
end
|
