Only set state.keys when necessary

state.keys is set when (a) a non-primitive value is initially traversed, (b) a primitive value is replaced with an object or vice-versa.
ljharb · Jun 18, 2012 · ee66cd1 · ee66cd1
1 parent 3857dca
commit ee66cd1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -162,7 +162,9 @@ function walk (root, cb, immutable) {
 
         function updateState() {
             if (typeof state.node === 'object' && state.node !== null) {
-                state.keys = Object_keys(state.node);
+                if (!state.keys || state.node_ !== state.node) {
+                    state.keys = Object_keys(state.node)
+                }
 
                 state.isLeaf = state.keys.length == 0;
 
@@ -175,6 +177,7 @@ function walk (root, cb, immutable) {
             }
             else {
                 state.isLeaf = true;
+                state.keys = null;
             }
 
             state.notLeaf = !state.isLeaf;