[6a3a178] | 1 | function RetryOperation(timeouts, options) {
|
---|
| 2 | // Compatibility for the old (timeouts, retryForever) signature
|
---|
| 3 | if (typeof options === 'boolean') {
|
---|
| 4 | options = { forever: options };
|
---|
| 5 | }
|
---|
| 6 |
|
---|
| 7 | this._originalTimeouts = JSON.parse(JSON.stringify(timeouts));
|
---|
| 8 | this._timeouts = timeouts;
|
---|
| 9 | this._options = options || {};
|
---|
| 10 | this._maxRetryTime = options && options.maxRetryTime || Infinity;
|
---|
| 11 | this._fn = null;
|
---|
| 12 | this._errors = [];
|
---|
| 13 | this._attempts = 1;
|
---|
| 14 | this._operationTimeout = null;
|
---|
| 15 | this._operationTimeoutCb = null;
|
---|
| 16 | this._timeout = null;
|
---|
| 17 | this._operationStart = null;
|
---|
| 18 |
|
---|
| 19 | if (this._options.forever) {
|
---|
| 20 | this._cachedTimeouts = this._timeouts.slice(0);
|
---|
| 21 | }
|
---|
| 22 | }
|
---|
| 23 | module.exports = RetryOperation;
|
---|
| 24 |
|
---|
| 25 | RetryOperation.prototype.reset = function() {
|
---|
| 26 | this._attempts = 1;
|
---|
| 27 | this._timeouts = this._originalTimeouts;
|
---|
| 28 | }
|
---|
| 29 |
|
---|
| 30 | RetryOperation.prototype.stop = function() {
|
---|
| 31 | if (this._timeout) {
|
---|
| 32 | clearTimeout(this._timeout);
|
---|
| 33 | }
|
---|
| 34 |
|
---|
| 35 | this._timeouts = [];
|
---|
| 36 | this._cachedTimeouts = null;
|
---|
| 37 | };
|
---|
| 38 |
|
---|
| 39 | RetryOperation.prototype.retry = function(err) {
|
---|
| 40 | if (this._timeout) {
|
---|
| 41 | clearTimeout(this._timeout);
|
---|
| 42 | }
|
---|
| 43 |
|
---|
| 44 | if (!err) {
|
---|
| 45 | return false;
|
---|
| 46 | }
|
---|
| 47 | var currentTime = new Date().getTime();
|
---|
| 48 | if (err && currentTime - this._operationStart >= this._maxRetryTime) {
|
---|
| 49 | this._errors.unshift(new Error('RetryOperation timeout occurred'));
|
---|
| 50 | return false;
|
---|
| 51 | }
|
---|
| 52 |
|
---|
| 53 | this._errors.push(err);
|
---|
| 54 |
|
---|
| 55 | var timeout = this._timeouts.shift();
|
---|
| 56 | if (timeout === undefined) {
|
---|
| 57 | if (this._cachedTimeouts) {
|
---|
| 58 | // retry forever, only keep last error
|
---|
| 59 | this._errors.splice(this._errors.length - 1, this._errors.length);
|
---|
| 60 | this._timeouts = this._cachedTimeouts.slice(0);
|
---|
| 61 | timeout = this._timeouts.shift();
|
---|
| 62 | } else {
|
---|
| 63 | return false;
|
---|
| 64 | }
|
---|
| 65 | }
|
---|
| 66 |
|
---|
| 67 | var self = this;
|
---|
| 68 | var timer = setTimeout(function() {
|
---|
| 69 | self._attempts++;
|
---|
| 70 |
|
---|
| 71 | if (self._operationTimeoutCb) {
|
---|
| 72 | self._timeout = setTimeout(function() {
|
---|
| 73 | self._operationTimeoutCb(self._attempts);
|
---|
| 74 | }, self._operationTimeout);
|
---|
| 75 |
|
---|
| 76 | if (self._options.unref) {
|
---|
| 77 | self._timeout.unref();
|
---|
| 78 | }
|
---|
| 79 | }
|
---|
| 80 |
|
---|
| 81 | self._fn(self._attempts);
|
---|
| 82 | }, timeout);
|
---|
| 83 |
|
---|
| 84 | if (this._options.unref) {
|
---|
| 85 | timer.unref();
|
---|
| 86 | }
|
---|
| 87 |
|
---|
| 88 | return true;
|
---|
| 89 | };
|
---|
| 90 |
|
---|
| 91 | RetryOperation.prototype.attempt = function(fn, timeoutOps) {
|
---|
| 92 | this._fn = fn;
|
---|
| 93 |
|
---|
| 94 | if (timeoutOps) {
|
---|
| 95 | if (timeoutOps.timeout) {
|
---|
| 96 | this._operationTimeout = timeoutOps.timeout;
|
---|
| 97 | }
|
---|
| 98 | if (timeoutOps.cb) {
|
---|
| 99 | this._operationTimeoutCb = timeoutOps.cb;
|
---|
| 100 | }
|
---|
| 101 | }
|
---|
| 102 |
|
---|
| 103 | var self = this;
|
---|
| 104 | if (this._operationTimeoutCb) {
|
---|
| 105 | this._timeout = setTimeout(function() {
|
---|
| 106 | self._operationTimeoutCb();
|
---|
| 107 | }, self._operationTimeout);
|
---|
| 108 | }
|
---|
| 109 |
|
---|
| 110 | this._operationStart = new Date().getTime();
|
---|
| 111 |
|
---|
| 112 | this._fn(this._attempts);
|
---|
| 113 | };
|
---|
| 114 |
|
---|
| 115 | RetryOperation.prototype.try = function(fn) {
|
---|
| 116 | console.log('Using RetryOperation.try() is deprecated');
|
---|
| 117 | this.attempt(fn);
|
---|
| 118 | };
|
---|
| 119 |
|
---|
| 120 | RetryOperation.prototype.start = function(fn) {
|
---|
| 121 | console.log('Using RetryOperation.start() is deprecated');
|
---|
| 122 | this.attempt(fn);
|
---|
| 123 | };
|
---|
| 124 |
|
---|
| 125 | RetryOperation.prototype.start = RetryOperation.prototype.try;
|
---|
| 126 |
|
---|
| 127 | RetryOperation.prototype.errors = function() {
|
---|
| 128 | return this._errors;
|
---|
| 129 | };
|
---|
| 130 |
|
---|
| 131 | RetryOperation.prototype.attempts = function() {
|
---|
| 132 | return this._attempts;
|
---|
| 133 | };
|
---|
| 134 |
|
---|
| 135 | RetryOperation.prototype.mainError = function() {
|
---|
| 136 | if (this._errors.length === 0) {
|
---|
| 137 | return null;
|
---|
| 138 | }
|
---|
| 139 |
|
---|
| 140 | var counts = {};
|
---|
| 141 | var mainError = null;
|
---|
| 142 | var mainErrorCount = 0;
|
---|
| 143 |
|
---|
| 144 | for (var i = 0; i < this._errors.length; i++) {
|
---|
| 145 | var error = this._errors[i];
|
---|
| 146 | var message = error.message;
|
---|
| 147 | var count = (counts[message] || 0) + 1;
|
---|
| 148 |
|
---|
| 149 | counts[message] = count;
|
---|
| 150 |
|
---|
| 151 | if (count >= mainErrorCount) {
|
---|
| 152 | mainError = error;
|
---|
| 153 | mainErrorCount = count;
|
---|
| 154 | }
|
---|
| 155 | }
|
---|
| 156 |
|
---|
| 157 | return mainError;
|
---|
| 158 | };
|
---|