summaryrefslogtreecommitdiff
path: root/app/workers/bulk_imports/entity_worker.rb
blob: 5c04cdc96a04a3ff220c31d0b3f967ea50defec1 (plain)
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
# frozen_string_literal: true

module BulkImports
  class EntityWorker # rubocop:disable Scalability/IdempotentWorker
    include ApplicationWorker

    data_consistency :always

    feature_category :importers

    sidekiq_options retry: false, dead: false

    worker_has_external_dependencies!

    def perform(entity_id, current_stage = nil)
      return if stage_running?(entity_id, current_stage)

      logger.info(
        worker: self.class.name,
        entity_id: entity_id,
        current_stage: current_stage
      )

      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(
        worker: self.class.name,
        entity_id: entity_id,
        current_stage: current_stage,
        error_message: e.message
      )

      Gitlab::ErrorTracking.track_exception(e, entity_id: entity_id)
    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)
    end

    def logger
      @logger ||= Gitlab::Import::Logger.build
    end
  end
end