fix(Table): fix expandedTreeNodes change effect

Tencent · Nov 14, 2024 · 44adef5 · 44adef5
1 parent 683e359
commit 44adef5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/table/hooks/useTreeData.tsx b/src/table/hooks/useTreeData.tsx
@@ -19,7 +19,7 @@ export interface UseSwapParams<T> extends SwapParams<T> {
 }
 
 export default function useTreeData(props: TdEnhancedTableProps) {
-  const { data, columns, tree, rowKey, treeExpandAndFoldIcon } = props;
+  const { data, columns, tree, rowKey, treeExpandAndFoldIcon, expandedTreeNodes } = props;
   const [store] = useState(new TableTreeStore() as InstanceType<typeof TableTreeStore>);
   const [treeNodeCol, setTreeNodeCol] = useState<PrimaryTableCol>(() => getTreeNodeColumnCol());
   const [dataSource, setDataSource] = useState<TdEnhancedTableProps['data']>(data || []);
@@ -70,7 +70,7 @@ export default function useTreeData(props: TdEnhancedTableProps) {
       }
     },
     // eslint-disable-next-line react-hooks/exhaustive-deps
-    [data],
+    [data, expandedTreeNodes],
   );
 
   useEffect(