new method retry.wrap about node-retry HOT 10 CLOSED

// wrap all methods of myLibrary object
retry.wrap({retries: 3}, myLibrary);
// wrap only named methods of myLibrary object
retry.wrap({retries: 3}, myLibrary, ['method1', 'method2', 'method3']);

// wrap all mehtods of myLibrary using default options
retry.wrap(myLibrary);

// wrap all methods of myLibrary object
retry.wrap(myLibrary, {retries: 3});

// wrap only named methods of myLibrary object
retry.wrap(myLibrary, {retries: 3}, ['method1', 'method2', 'method3']);

function wrap(obj, options, methods) {
    if (options instanceof Array) {
        methods = options;
        options = null;
    }

    if (!methods) {
        methods = Object.keys(obj);
    }

    methods.forEach(function(method) {
        var origin = obj[method];

        if (typeof origin != 'function') {
            return;
        }

        obj[method] = function() {
            var op = retry.operation(options),
                args = Array.prototype.slice.call(arguments),
                callback = args.pop();

            args.push(function(err) {
                if (op.retry(err)) {
                    return;
                }
                if (err) {
                    arguments[0] = op.mainError();
                }
                callback.apply(this, arguments);
            });

            op.attempt(function() {
                origin.apply(obj, args);
            });
        };
    });    
}