improved waterfall performance with ensureAsync

2 files changed, 6 insertions, 4 deletions

diff --git a/lib/async.js b/lib/async.js
index 314432e..4486ba0 100644
--- a/lib/async.js
+++ b/lib/async.js
@@ -651,9 +651,7 @@
                     else {
                         args.push(callback);
                     }
-                    async.setImmediate(function () {
-                        iterator.apply(null, args);
-                    });
+                    ensureAsync(iterator).apply(null, args);
                 }
             };
         }
diff --git a/perf/suites.js b/perf/suites.js
index 28b5a32..b4ea76b 100644
--- a/perf/suites.js
+++ b/perf/suites.js
@@ -148,7 +148,11 @@ module.exports = [
             return cb(null, 1);
           }
         ].concat(_.range(count).map(function (i) {
-          return function (arg, cb) { cb(null, i); };
+          return function (arg, cb) {
+            setImmediate(function () {
+              cb(null, i);
+            });
+          };
         }));
     },
     fn: function (async, done) {