-
Notifications
You must be signed in to change notification settings - Fork 308
/
retry.ts
84 lines (72 loc) · 2.25 KB
/
retry.ts
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
const MAX_REQUEST_RETRY_JITTER = 250;
const MAX_REQUEST_RETRY_DELAY = 10000;
const DEFAULT_NUMBER_RETRIES = 3;
const MAX_NUMBER_RETRIES = 10;
const BASE_DELAY = 500;
/**
* @private
* Function that returns a random int between a configurable min and max.
* @param min The min value
* @param max The max value
* @returns {number} The random generated value
*/
function getRandomInt(min: number, max: number) {
min = Math.ceil(min);
max = Math.floor(max);
return Math.floor(Math.random() * (max - min) + min); // The maximum is exclusive and the minimum is inclusive
}
/**
* @private
* Function that returns a promise which resolves after a configurable amount of milliseconds
* @param delay value to be used for the delay
* @returns {Promise} A delayed promise
*/
async function pause(delay: number) {
return new Promise((resolve) => setTimeout(resolve, delay));
}
/**
* Configure the retry logic for http calls.
* By default, this retries any request that returns a 429 3 times.
*/
export interface RetryConfiguration {
/**
* Configure the usage of retries.
* Defaults to true on the Management Client and false on the Authentication Client.
*/
enabled?: boolean;
/**
* Configure the max amount of retries the SDK should do.
* Defaults to 5.
*/
maxRetries?: number;
/**
* Status Codes on which the SDK should trigger retries.
* Defaults to [429].
*/
retryWhen?: number[];
}
/**
* @private
* Function that retries the provided action callback for a configurable amount of time, defaults to 3.
*/
export function retry(
action: () => Promise<Response>,
{ maxRetries, retryWhen }: RetryConfiguration
) {
const nrOfTriesToAttempt = Math.min(MAX_NUMBER_RETRIES, maxRetries ?? DEFAULT_NUMBER_RETRIES);
let nrOfTries = 0;
const retryAndWait = async () => {
let result: Response;
result = await action();
if ((retryWhen || [429]).includes(result.status) && nrOfTries < nrOfTriesToAttempt) {
nrOfTries++;
let wait = BASE_DELAY * Math.pow(2, nrOfTries - 1);
wait = getRandomInt(wait + 1, wait + MAX_REQUEST_RETRY_JITTER);
wait = Math.min(wait, MAX_REQUEST_RETRY_DELAY);
await pause(wait);
result = await retryAndWait();
}
return result;
};
return retryAndWait();
}