1  'use strict';
 2  
 3  Object.defineProperty(exports, "__esModule", {
 4      value: true
 5  });
 6  exports.default = retryable;
 7  
 8  var _retry = require('./retry');
 9  
10  var _retry2 = _interopRequireDefault(_retry);
11  
12  var _initialParams = require('./internal/initialParams');
13  
14  var _initialParams2 = _interopRequireDefault(_initialParams);
15  
16  var _wrapAsync = require('./internal/wrapAsync');
17  
18  var _wrapAsync2 = _interopRequireDefault(_wrapAsync);
19  
20  var _promiseCallback = require('./internal/promiseCallback');
21  
22  function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
23  
24  /**
25   * A close relative of [`retry`]{@link module:ControlFlow.retry}.  This method
26   * wraps a task and makes it retryable, rather than immediately calling it
27   * with retries.
28   *
29   * @name retryable
30   * @static
31   * @memberOf module:ControlFlow
32   * @method
33   * @see [async.retry]{@link module:ControlFlow.retry}
34   * @category Control Flow
35   * @param {Object|number} [opts = {times: 5, interval: 0}| 5] - optional
36   * options, exactly the same as from `retry`, except for a `opts.arity` that
37   * is the arity of the `task` function, defaulting to `task.length`
38   * @param {AsyncFunction} task - the asynchronous function to wrap.
39   * This function will be passed any arguments passed to the returned wrapper.
40   * Invoked with (...args, callback).
41   * @returns {AsyncFunction} The wrapped function, which when invoked, will
42   * retry on an error, based on the parameters specified in `opts`.
43   * This function will accept the same parameters as `task`.
44   * @example
45   *
46   * async.auto({
47   *     dep1: async.retryable(3, getFromFlakyService),
48   *     process: ["dep1", async.retryable(3, function (results, cb) {
49   *         maybeProcessData(results.dep1, cb);
50   *     })]
51   * }, callback);
52   */
53  function retryable(opts, task) {
54      if (!task) {
55          task = opts;
56          opts = null;
57      }
58      let arity = opts && opts.arity || task.length;
59      if ((0, _wrapAsync.isAsync)(task)) {
60          arity += 1;
61      }
62      var _task = (0, _wrapAsync2.default)(task);
63      return (0, _initialParams2.default)((args, callback) => {
64          if (args.length < arity - 1 || callback == null) {
65              args.push(callback);
66              callback = (0, _promiseCallback.promiseCallback)();
67          }
68          function taskFn(cb) {
69              _task(...args, cb);
70          }
71  
72          if (opts) (0, _retry2.default)(opts, taskFn, callback);else (0, _retry2.default)(taskFn, callback);
73  
74          return callback[_promiseCallback.PROMISE_SYMBOL];
75      });
76  }
77  module.exports = exports['default'];