From 3f7f4c4237819b826694973035e6db393d3554a6 Mon Sep 17 00:00:00 2001
From: gx-mac <gaox2025f@gmail.com>
Date: Wed, 10 Mar 2021 16:42:04 +0800
Subject: [PATCH] fix: sync cacheExpandedKeys when passing loadData func

---
 src/Tree.js | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/Tree.js b/src/Tree.js
index 7ecde04..5ec9dc5 100644
--- a/src/Tree.js
+++ b/src/Tree.js
@@ -345,7 +345,8 @@ onExpand(treeNode,keyType) {
       node: treeNode,
       expanded
     });
-
+    //收起和展开时，缓存 expandedKeys
+    this.cacheExpandedKeys = new Set(expandedKeys);
     // after data loaded, need set new expandedKeys
     if (expanded && this.props.loadData) {
       return this.props.loadData(treeNode).then(() => {
@@ -356,8 +357,6 @@ onExpand(treeNode,keyType) {
         }
       });
     }
-    //收起和展开时，缓存 expandedKeys
-    this.cacheExpandedKeys = new Set(expandedKeys);
     //启用懒加载，把 Tree 结构拍平，为后续动态截取数据做准备
     if(lazyLoad) {
       let flatTreeData = this.deepTraversal(treeData);