summaryrefslogtreecommitdiff
path: root/app/services/ci/retry_pipeline_service.rb
blob: 1b931455fb8c22239c75ecca39becf6a1d2bf8b8 (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
module Ci
  class RetryPipelineService < ::BaseService
    def execute(pipeline)
      @pipeline = pipeline

      unless can?(current_user, :update_pipeline, pipeline)
        raise Gitlab::Access::AccessDeniedError
      end

      pipeline.mark_as_processable_after_stage(resume_stage.index)

      retryable_builds_in_subsequent_stages do |build|
        Ci::RetryBuildService.new(project, current_user)
          .reprocess(build)
      end

      retryable_builds_in_first_unsuccessful_stage do |build|
        Ci::RetryBuildService.new(project, current_user)
          .retry(build)
      end
    end

    private

    def retryable_builds_in_subsequent_stages
      relation = @pipeline.builds
        .after_stage(resume_stage.index)
        .failed_or_canceled

      each_retryable_build_with_locking(relation) do |build|
        yield build
      end
    end

    def retryable_builds_in_first_unsuccessful_stage
      relation = resume_stage.builds.failed_or_canceled

      each_retryable_build_with_locking(relation) do |build|
        yield build
      end
    end

    def each_retryable_build_with_locking(relation)
      Gitlab::OptimisticLocking.retry_lock(relation) do |builds|
        builds.find_each do |build|
          next unless build.retryable?
          yield build
        end
      end
    end

    def resume_stage
      @resume_stage ||= @pipeline.stages.find do |stage|
        stage.failed? || stage.canceled?
      end
    end
  end
end