summaryrefslogtreecommitdiff
path: root/lib/gitlab/ci/build/auto_retry.rb
blob: 6ab567dff7c6772fb13bfb641bc617b7dc4b06a5 (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
59
60
61
62
63
64
65
66
67
68
# frozen_string_literal: true

class Gitlab::Ci::Build::AutoRetry
  include Gitlab::Utils::StrongMemoize

  DEFAULT_RETRIES = {
    scheduler_failure: 2
  }.freeze

  RETRY_OVERRIDES = {
    ci_quota_exceeded: 0,
    no_matching_runner: 0,
    missing_dependency_failure: 0
  }.freeze

  def initialize(build)
    @build = build
  end

  def allowed?
    return false unless @build.retryable?

    within_max_retry_limit?
  end

  private

  delegate :failure_reason, to: :@build

  def within_max_retry_limit?
    max_allowed_retries > 0 && max_allowed_retries > @build.retries_count
  end

  def max_allowed_retries
    strong_memoize(:max_allowed_retries) do
      RETRY_OVERRIDES[failure_reason.to_sym] ||
        options_retry_max ||
        DEFAULT_RETRIES[failure_reason.to_sym] ||
        0
    end
  end

  def options_retry_max
    Integer(options_retry[:max], exception: false) if retry_on_reason_or_always?
  end

  def options_retry_when
    options_retry.fetch(:when, ['always'])
  end

  def retry_on_reason_or_always?
    options_retry_when.include?(failure_reason.to_s) ||
      options_retry_when.include?('always')
  end

  # The format of the retry option changed in GitLab 11.5: Before it was
  # integer only, after it is a hash. New builds are created with the new
  # format, but builds created before GitLab 11.5 and saved in database still
  # have the old integer only format. This method returns the retry option
  # normalized as a hash in 11.5+ format.
  def options_retry
    strong_memoize(:options_retry) do
      value = @build.options&.dig(:retry)
      value = value.is_a?(Integer) ? { max: value } : value.to_h
      value.with_indifferent_access
    end
  end
end