index.js 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. // Packages
  2. var retrier = require('retry');
  3. function retry(fn, opts) {
  4. function run(resolve, reject) {
  5. var options = opts || {};
  6. var op;
  7. // Default `randomize` to true
  8. if (!('randomize' in options)) {
  9. options.randomize = true;
  10. }
  11. op = retrier.operation(options);
  12. // We allow the user to abort retrying
  13. // this makes sense in the cases where
  14. // knowledge is obtained that retrying
  15. // would be futile (e.g.: auth errors)
  16. function bail(err) {
  17. reject(err || new Error('Aborted'));
  18. }
  19. function onError(err, num) {
  20. if (err.bail) {
  21. bail(err);
  22. return;
  23. }
  24. if (!op.retry(err)) {
  25. reject(op.mainError());
  26. } else if (options.onRetry) {
  27. options.onRetry(err, num);
  28. }
  29. }
  30. function runAttempt(num) {
  31. var val;
  32. try {
  33. val = fn(bail, num);
  34. } catch (err) {
  35. onError(err, num);
  36. return;
  37. }
  38. Promise.resolve(val)
  39. .then(resolve)
  40. .catch(function catchIt(err) {
  41. onError(err, num);
  42. });
  43. }
  44. op.attempt(runAttempt);
  45. }
  46. return new Promise(run);
  47. }
  48. module.exports = retry;