96 lines
3.2 KiB
Ruby
96 lines
3.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Ci
|
|
class Config
|
|
module Entry
|
|
##
|
|
# Entry that represents a retry config for a job.
|
|
#
|
|
class Retry < ::Gitlab::Config::Entry::Simplifiable
|
|
strategy :SimpleRetry, if: -> (config) { config.is_a?(Integer) }
|
|
strategy :FullRetry, if: -> (config) { config.is_a?(Hash) }
|
|
|
|
class SimpleRetry < ::Gitlab::Config::Entry::Node
|
|
include ::Gitlab::Config::Entry::Validatable
|
|
|
|
validations do
|
|
validates :config, numericality: { only_integer: true,
|
|
greater_than_or_equal_to: 0,
|
|
less_than_or_equal_to: 2 }
|
|
end
|
|
|
|
def value
|
|
{
|
|
max: config
|
|
}
|
|
end
|
|
|
|
def location
|
|
'retry'
|
|
end
|
|
end
|
|
|
|
class FullRetry < ::Gitlab::Config::Entry::Node
|
|
include ::Gitlab::Config::Entry::Validatable
|
|
include ::Gitlab::Config::Entry::Attributable
|
|
|
|
ALLOWED_KEYS = %i[max when exit_codes].freeze
|
|
attributes ALLOWED_KEYS
|
|
|
|
validations do
|
|
validates :config, allowed_keys: ALLOWED_KEYS
|
|
|
|
with_options allow_nil: true do
|
|
validates :max, numericality: { only_integer: true,
|
|
greater_than_or_equal_to: 0,
|
|
less_than_or_equal_to: 2 }
|
|
|
|
validates :when, array_of_strings_or_string: true
|
|
validates :when,
|
|
allowed_array_values: { in: FullRetry.possible_retry_when_values },
|
|
if: -> (config) { config.when.is_a?(Array) }
|
|
validates :when,
|
|
inclusion: { in: FullRetry.possible_retry_when_values },
|
|
if: -> (config) { config.when.is_a?(String) }
|
|
validates :exit_codes, array_of_integers_or_integer: true
|
|
end
|
|
end
|
|
|
|
def self.possible_retry_when_values
|
|
@possible_retry_when_values ||= ::Ci::Build.failure_reasons.keys.map(&:to_s) + ['always']
|
|
end
|
|
|
|
def value
|
|
super.tap do |config|
|
|
# make sure that `when` and `exit_codes` are arrays, because we allow them to
|
|
# be passed as a String/Integer in config for simplicity
|
|
config[:when] = Array.wrap(config[:when]) if config[:when]
|
|
if config[:exit_codes] && Feature.enabled?(:ci_retry_on_exit_codes, Feature.current_request)
|
|
config[:exit_codes] = Array.wrap(config[:exit_codes])
|
|
else
|
|
config.delete(:exit_codes)
|
|
end
|
|
end
|
|
end
|
|
|
|
def location
|
|
'retry'
|
|
end
|
|
end
|
|
|
|
class UnknownStrategy < ::Gitlab::Config::Entry::Node
|
|
def errors
|
|
["#{location} has to be either an integer or a hash"]
|
|
end
|
|
|
|
def location
|
|
'retry config'
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|