Welcome to mirror list, hosted at ThFree Co, Russian Federation.

finish_batched_pipeline_worker.rb « bulk_imports « workers « app - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 4200d0e4a0f994b86d48ff4a9ccbd1f0bd0db8d9 (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
# frozen_string_literal: true

module BulkImports
  class FinishBatchedPipelineWorker
    include ApplicationWorker
    include ExceptionBacktrace

    REQUEUE_DELAY = 5.seconds

    idempotent!
    deduplicate :until_executing
    data_consistency :always # rubocop:disable SidekiqLoadBalancing/WorkerDataConsistency
    feature_category :importers

    def perform(pipeline_tracker_id)
      @tracker = Tracker.find(pipeline_tracker_id)

      return unless tracker.batched?
      return unless tracker.started?
      return re_enqueue if import_in_progress?

      if tracker.stale?
        tracker.batches.map(&:fail_op!)
        tracker.fail_op!
      else
        tracker.finish!
      end

    ensure
      ::BulkImports::EntityWorker.perform_async(tracker.entity.id, tracker.stage)
    end

    private

    attr_reader :tracker

    def re_enqueue
      self.class.perform_in(REQUEUE_DELAY, tracker.id)
    end

    def import_in_progress?
      tracker.batches.any?(&:started?)
    end
  end
end