fix(comp:table): table selection emptyed after data change and select (…

…#1939)
IDuxFE · Jun 12, 2024 · 373b434 · 373b434
1 parent 403b2ca
commit 373b434
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/packages/components/utils/src/useTreeCheckState.ts b/packages/components/utils/src/useTreeCheckState.ts
@@ -70,15 +70,16 @@ export function useTreeCheckState<V extends TreeTypeData<V, C>, C extends keyof
 
     const mergedData = mergeTree(data ?? [], cachedData ?? [], childrenKey.value, getKey.value)
     const mergedDataMap = new Map(dataMap)
+    const _cachedDataMap = new Map(cachedDataMap)
 
     mergedDataMap.forEach((item, key) => {
-      if (cachedDataMap.has(key)) {
-        const cachedItem = cachedDataMap.get(key)!
+      if (_cachedDataMap.has(key)) {
+        const cachedItem = _cachedDataMap.get(key)!
         mergedDataMap.set(key, mergeTree([item], [cachedItem], childrenKey.value, getKey.value)[0])
-        cachedDataMap.delete(key)
+        _cachedDataMap.delete(key)
       }
     })
-    cachedDataMap.forEach((item, key) => {
+    _cachedDataMap.forEach((item, key) => {
       mergedDataMap.set(key, item)
     })