tenebrous-dicebot/api/node_modules/p-retry/index.js

61 lines
1.5 KiB
JavaScript

'use strict';
const retry = require('retry');
class AbortError extends Error {
constructor(message) {
super();
if (message instanceof Error) {
this.originalError = message;
({message} = message);
} else {
this.originalError = new Error(message);
this.originalError.stack = this.stack;
}
this.name = 'AbortError';
this.message = message;
}
}
function decorateErrorWithCounts(error, attemptNumber, options) {
// Minus 1 from attemptNumber because the first attempt does not count as a retry
const retriesLeft = options.retries - (attemptNumber - 1);
error.attemptNumber = attemptNumber;
error.retriesLeft = retriesLeft;
return error;
}
module.exports = (input, options) => new Promise((resolve, reject) => {
options = Object.assign({
onFailedAttempt: () => {},
retries: 10
}, options);
const operation = retry.operation(options);
operation.attempt(attemptNumber => Promise.resolve(attemptNumber)
.then(input)
.then(resolve, error => {
if (error instanceof AbortError) {
operation.stop();
reject(error.originalError);
} else if (error instanceof TypeError) {
operation.stop();
reject(error);
} else if (operation.retry(error)) {
decorateErrorWithCounts(error, attemptNumber, options);
options.onFailedAttempt(error);
} else {
decorateErrorWithCounts(error, attemptNumber, options);
options.onFailedAttempt(error);
reject(operation.mainError());
}
})
);
});
module.exports.AbortError = AbortError;