blob: 49b0f638387d8542e24ffdf8f9545780002c9dce (
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
# frozen_string_literal: true
module Bundler
# General purpose class for retrying code that may fail
class Retry
attr_accessor :name, :total_runs, :current_run
class << self
attr_accessor :default_base_delay
def default_attempts
default_retries + 1
end
alias_method :attempts, :default_attempts
def default_retries
Bundler.settings[:retry]
end
end
# Set default base delay for exponential backoff
self.default_base_delay = 1.0
def initialize(name, exceptions = nil, retries = self.class.default_retries, opts = {})
@name = name
@retries = retries
@exceptions = Array(exceptions) || []
@total_runs = @retries + 1 # will run once, then upto attempts.times
@base_delay = opts[:base_delay] || self.class.default_base_delay
@max_delay = opts[:max_delay] || 60.0
@jitter = opts[:jitter] || 0.5
end
def attempt(&block)
@current_run = 0
@failed = false
@error = nil
run(&block) while keep_trying?
@result
end
alias_method :attempts, :attempt
private
def run(&block)
@failed = false
@current_run += 1
@result = block.call
rescue StandardError => e
fail_attempt(e)
end
def fail_attempt(e)
@failed = true
if last_attempt? || @exceptions.any? {|k| e.is_a?(k) }
Bundler.ui.info "" unless Bundler.ui.debug?
raise e
end
if name
Bundler.ui.info "" unless Bundler.ui.debug? # Add new line in case dots preceded this
Bundler.ui.warn "Retrying #{name} due to error (#{current_run.next}/#{total_runs}): #{e.class} #{e.message}", true
end
backoff_sleep if @base_delay > 0
true
end
def backoff_sleep
# Exponential backoff: delay = base_delay * 2^(attempt - 1)
# Add jitter to prevent thundering herd: random value between 0 and jitter seconds
delay = @base_delay * (2**(@current_run - 1))
delay = [@max_delay, delay].min
jitter_amount = rand * @jitter
total_delay = delay + jitter_amount
Bundler.ui.debug "Sleeping for #{total_delay.round(2)} seconds before retry"
sleep(total_delay)
end
def sleep(duration)
Kernel.sleep(duration)
end
def keep_trying?
return true if current_run.zero?
return false if last_attempt?
true if @failed
end
def last_attempt?
current_run >= total_runs
end
end
end
|
