From aced1be6e87598fe93477ebcd419840fb561c6b5 Mon Sep 17 00:00:00 2001
From: bendtherules <abhasbhattacharya2@gmail.com>
Date: Wed, 11 Sep 2019 15:13:21 +0530
Subject: [PATCH] Rewrite traverse.traverse using traverseGenerator

---
 src/traverse.ts | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/src/traverse.ts b/src/traverse.ts
index 849d971..be64885 100644
--- a/src/traverse.ts
+++ b/src/traverse.ts
@@ -125,13 +125,9 @@ function* traverseGenerator(
  *
  */
 function traverse(node: FiberNode, fn: (node: FiberNode) => any) {
-  fn.call(null, node);
-
-  if (node.child !== null) {
-    traverse(node.child, fn);
-  }
-  if (node.sibling !== null) {
-    traverse(node.sibling, fn);
+  const nodeIterator = traverseGenerator(node);
+  for (const tmpNode of nodeIterator) {
+    fn.call(null, tmpNode);
   }
 }