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
|
'use strict';
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.default = retry;
var _series = require('./series');
var _series2 = _interopRequireDefault(_series);
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
function retry(times, task, callback) {
var DEFAULT_TIMES = 5;
var DEFAULT_INTERVAL = 0;
var attempts = [];
var opts = {
times: DEFAULT_TIMES,
interval: DEFAULT_INTERVAL
};
function parseTimes(acc, t) {
if (typeof t === 'number') {
acc.times = parseInt(t, 10) || DEFAULT_TIMES;
} else if (typeof t === 'object') {
acc.times = parseInt(t.times, 10) || DEFAULT_TIMES;
acc.interval = parseInt(t.interval, 10) || DEFAULT_INTERVAL;
} else {
throw new Error('Unsupported argument type for \'times\': ' + typeof t);
}
}
var length = arguments.length;
if (length < 1 || length > 3) {
throw new Error('Invalid arguments - must be either (task), (task, callback), (times, task) or (times, task, callback)');
} else if (length <= 2 && typeof times === 'function') {
callback = task;
task = times;
}
if (typeof times !== 'function') {
parseTimes(opts, times);
}
opts.callback = callback;
opts.task = task;
function wrappedTask(wrappedCallback, wrappedResults) {
function retryAttempt(task, finalAttempt) {
return function (seriesCallback) {
task(function (err, result) {
seriesCallback(!err || finalAttempt, {
err: err,
result: result
});
}, wrappedResults);
};
}
function retryInterval(interval) {
return function (seriesCallback) {
setTimeout(function () {
seriesCallback(null);
}, interval);
};
}
while (opts.times) {
var finalAttempt = !(opts.times -= 1);
attempts.push(retryAttempt(opts.task, finalAttempt));
if (!finalAttempt && opts.interval > 0) {
attempts.push(retryInterval(opts.interval));
}
}
(0, _series2.default)(attempts, function (done, data) {
data = data[data.length - 1];
(wrappedCallback || opts.callback)(data.err, data.result);
});
}
// If a callback is passed, run this as a controll flow
return opts.callback ? wrappedTask() : wrappedTask;
}
module.exports = exports['default'];
|