perf(tree): inherit props status while changing the data

"fix #1140"
Tencent · Jul 25, 2022 · 9ef9525 · 9ef9525
1 parent 51055e5
commit 9ef9525
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/tree/useStore.ts b/src/tree/useStore.ts
@@ -153,8 +153,14 @@ export function useStore(props: TdTreeProps, refresh: () => void): TreeStore {
 
   useUpdateEffect(() => {
     if (data && Array.isArray(data)) {
+      const expanded = store.getExpanded();
+      const checked = store.getChecked();
+      const actived = store.getActived();
       store.removeAll();
       store.append(data);
+      store.setChecked(checked);
+      store.setActived(actived);
+      store.setExpanded(expanded);
     }
   }, [data, store]);