Module: Puppet::Util::RetryAction

Defined in:
lib/vendor/puppet/util/retryaction.rb

Defined Under Namespace

Classes: RetryException

Class Method Summary collapse

Class Method Details

.retry_action(parameters = { :retry_exceptions => nil, :retries => nil }) ⇒ Object

Raises:



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
 
# File 'lib/vendor/puppet/util/retryaction.rb', line 7

def self.retry_action( parameters = { :retry_exceptions => nil, :retries => nil } )
  # Retry actions for a specified amount of time. This method will allow the final
  # retry to complete even if that extends beyond the timeout period.
  unless block_given?
    raise RetryException::NoBlockGiven
  end

  raise RetryException::NoRetriesGiven if parameters[:retries].nil?
  parameters[:retry_exceptions] ||= Hash.new

  start = Time.now
  failures = 0

  begin
    yield
  rescue Exception => e
    # If we were giving exceptions to catch,
    # catch the excptions we care about and retry.
    # All others fail hard

    raise RetryException::RetriesExceeded if parameters[:retries] == 0

    if (not parameters[:retry_exceptions].keys.empty?) and parameters[:retry_exceptions].keys.include?(e.class)
      Puppet.info("Caught exception #{e.class}:#{e}")
      Puppet.info(parameters[:retry_exceptions][e.class])
    elsif (not parameters[:retry_exceptions].keys.empty?)
      # If the exceptions is not in the list of retry_exceptions re-raise.
      raise e
    end

    failures += 1
    parameters[:retries] -= 1

    # Increase the amount of time that we sleep after every
    # failed retry attempt.
    sleep (((2 ** failures) -1) * 0.1)

    retry

  end
end