From e6d257017aa294188957e190c8f6dbd079640cfc Mon Sep 17 00:00:00 2001
From: HOU Ce <594965698@qq.com>
Date: Fri, 28 Sep 2018 23:31:24 +0800
Subject: [PATCH] OPT: check element fisrt (#2429)

run ` if (!element)` at first to avoid rest code running.

If` if (!element)` equals true, then stop executing the rest other wise we run `element.forEach(item => walkTree(item, context, visitor, newContext));`  all the time at first
---
 src/getDataFromTree.ts | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/getDataFromTree.ts b/src/getDataFromTree.ts
index a0467c11f8..8fea08ad18 100755
--- a/src/getDataFromTree.ts
+++ b/src/getDataFromTree.ts
@@ -57,12 +57,12 @@ export function walkTree(
   ) => boolean | void,
   newContext: Map<any, any> = new Map(),
 ) {
-  if (Array.isArray(element)) {
-    element.forEach(item => walkTree(item, context, visitor, newContext));
+  if (!element) {
     return;
   }
-
-  if (!element) {
+                       
+  if (Array.isArray(element)) {
+    element.forEach(item => walkTree(item, context, visitor, newContext));
     return;
   }