Use iterative approach to prevent stackoverflows

serverless-heaven · Mar 22, 2018 · 810e27a · 810e27a
1 parent 68d649b
commit 810e27a
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/lib/utils.js b/lib/utils.js
@@ -34,15 +34,16 @@ function searchAndProcessCache(moduleName, processor) {
   let mod_src = require.resolve(moduleName);
   const visitedModules = [];
   if (mod_src && ((mod_src = require.cache[mod_src]) !== undefined)) {
-    (function traverse(mod) {
+    const modStack = [mod_src];
+
+    while (!_.isEmpty(modStack)) {
+      const mod = modStack.pop();
       if (!_.includes(visitedModules, mod)) {
         visitedModules.push(mod);
-        _.forEach(mod.children, function (child) {
-          traverse(child);
-        });
+        Array.prototype.push.apply(modStack, mod.children);
         processor(mod);
       }
-    }(mod_src));
+    }
   }
   return BbPromise.resolve();
 }