From d2e269f772dee57eaea202d0696107e84252af97 Mon Sep 17 00:00:00 2001
From: pengpeng <30883395+webvs2@users.noreply.github.com>
Date: Sun, 26 Jan 2025 15:21:56 +0800
Subject: [PATCH] fix(components): [table] the toggleTreeExpansion failed to
 trigger load function (#18728)

* feat(components): [table] fix lazy loading

* Modify the function usage
---
 packages/components/table/src/store/tree.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/components/table/src/store/tree.ts b/packages/components/table/src/store/tree.ts
index 4c6b899fd9..f2558199ba 100644
--- a/packages/components/table/src/store/tree.ts
+++ b/packages/components/table/src/store/tree.ts
@@ -161,7 +161,9 @@ function useTree<T>(watcherData: WatcherPropsData<T>) {
     expandRowKeys.value = value
     updateTreeData()
   }
-
+  const isUseLazy = (data): boolean => {
+    return lazy.value && data && 'loaded' in data && !data.loaded
+  }
   const toggleTreeExpansion = (row: T, expanded?: boolean) => {
     instance.store.assertRowKey()
 
@@ -175,6 +177,7 @@ function useTree<T>(watcherData: WatcherPropsData<T>) {
       if (oldExpanded !== expanded) {
         instance.emit('expand-change', row, expanded)
       }
+      isUseLazy(data) && loadData(row, id, data)
       instance.store.updateTableScrollY()
     }
   }
@@ -184,7 +187,7 @@ function useTree<T>(watcherData: WatcherPropsData<T>) {
     const rowKey = watcherData.rowKey.value
     const id = getRowIdentity(row, rowKey)
     const data = treeData.value[id]
-    if (lazy.value && data && 'loaded' in data && !data.loaded) {
+    if (isUseLazy(data)) {
       loadData(row, id, data)
     } else {
       toggleTreeExpansion(row, undefined)