fix: treeTable can not expand properly after folding all nodes (#1729)

* fix: Table fold all * chore: remove console --------- Co-authored-by: sheepluo <[email protected]>
Tencent · Feb 25, 2024 · 0be283e · 0be283e
1 parent a8f1fd7
commit 0be283e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/js/table/tree-store.ts b/js/table/tree-store.ts
@@ -577,15 +577,17 @@ class TableTreeStore<T extends TableRowData = TableRowData> {
    */
   foldAll(dataSource: T[], keys: KeysType) {
     const newData: T[] = [];
+    let index = 0;
     for (let i = 0, len = dataSource.length; i < len; i++) {
       const item = dataSource[i];
       const rowValue = get(item, keys.rowKey);
       const state = this.treeDataMap.get(rowValue);
-      state.rowIndex = state.level === 0 ? i : -1;
+      state.rowIndex = state.level === 0 ? index : -1;
       state.expanded = false;
       state.expandChildrenLength = 0;
       if (state.level === 0) {
         newData.push(item);
+        index += 1;
       }
       const children = get(item, keys.childrenKey);
       if (children?.length) {