blob: ee39007efaae09b74c707a7f357a8214560d31b0 (
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
|
# Copyright 2018 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
__all__ = (
'ExponentialBackoff',
'RandomExponentialBackoff',
)
import random
import sys
class ExponentialBackoff(object):
"""
An object that when called with number of previous tries, calculates
an exponential delay for the next try.
"""
def __init__(self, multiplier=1, base=2, limit=sys.maxsize):
"""
@param multiplier: constant multiplier
@type multiplier: int or float
@param base: maximum number of tries
@type base: int or float
@param limit: maximum number of seconds to delay
@type limit: int or float
"""
self._multiplier = multiplier
self._base = base
self._limit = limit
def __call__(self, tries):
"""
Given a number of previous tries, calculate the amount of time
to delay the next try.
@param tries: number of previous tries
@type tries: int
@return: amount of time to delay the next try
@rtype: int
"""
try:
return min(self._limit, self._multiplier * (self._base ** tries))
except OverflowError:
return self._limit
class RandomExponentialBackoff(ExponentialBackoff):
"""
Equivalent to ExponentialBackoff, with an extra multiplier that uses
a random distribution between 0 and 1.
"""
def __call__(self, tries):
return random.random() * super(RandomExponentialBackoff, self).__call__(tries)
|