1 var RetryOperation = require('./retry_operation');
3 exports.operation = function(options) {
4 var timeouts = exports.timeouts(options);
5 return new RetryOperation(timeouts, {
6 forever: options && options.forever,
7 unref: options && options.unref
11 exports.timeouts = function(options) {
12 if (options instanceof Array) {
13 return [].concat(options);
23 for (var key in options) {
24 opts[key] = options[key];
27 if (opts.minTimeout > opts.maxTimeout) {
28 throw new Error('minTimeout is greater than maxTimeout');
32 for (var i = 0; i < opts.retries; i++) {
33 timeouts.push(this.createTimeout(i, opts));
36 if (options && options.forever && !timeouts.length) {
37 timeouts.push(this.createTimeout(i, opts));
40 // sort the array numerically ascending
41 timeouts.sort(function(a,b) {
48 exports.createTimeout = function(attempt, opts) {
49 var random = (opts.randomize)
53 var timeout = Math.round(random * opts.minTimeout * Math.pow(opts.factor, attempt));
54 timeout = Math.min(timeout, opts.maxTimeout);
59 exports.wrap = function(obj, options, methods) {
60 if (options instanceof Array) {
67 for (var key in obj) {
68 if (typeof obj[key] === 'function') {
74 for (var i = 0; i < methods.length; i++) {
75 var method = methods[i];
76 var original = obj[method];
78 obj[method] = function retryWrapper() {
79 var op = exports.operation(options);
80 var args = Array.prototype.slice.call(arguments);
81 var callback = args.pop();
83 args.push(function(err) {
88 arguments[0] = op.mainError();
90 callback.apply(this, arguments);
93 op.attempt(function() {
94 original.apply(obj, args);
97 obj[method].options = options;