perf(check-tree-select): 优化大数据量下树选择器性能

XiaoMi · Sep 12, 2024 · f096bb6 · f096bb6
1 parent 6188997
commit f096bb6
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/.changeset/smart-sheep-hide.md b/.changeset/smart-sheep-hide.md
@@ -0,0 +1,5 @@
+---
+"@hi-ui/hiui": patch
+---
+
+perf(check-tree-select): 优化大数据量下树选择器性能
diff --git a/.changeset/stale-cougars-lay.md b/.changeset/stale-cougars-lay.md
@@ -0,0 +1,5 @@
+---
+"@hi-ui/check-tree-select": patch
+---
+
+perf: 优化大数据量下树选择器性能
diff --git a/packages/ui/check-tree-select/src/utils/index.ts b/packages/ui/check-tree-select/src/utils/index.ts
@@ -1,5 +1,5 @@
 import React from 'react'
-import { fFindNodeById, findNestedChildren, getNodeAncestors } from '@hi-ui/tree-utils'
+import { findNestedChildren, getNodeAncestors } from '@hi-ui/tree-utils'
 import { isArrayNonEmpty } from '@hi-ui/type-assertion'
 
 /**
@@ -16,11 +16,12 @@ export const processCheckedIds = (
   allowCheck: (node: any) => boolean
 ) => {
   const keySet = new Set(checkedIds)
+  const flattedDataMap = new Map(flattenData.map((item: any) => [item.id, item]))
 
   switch (type) {
     case 'CHILD':
       return checkedIds.filter((id) => {
-        const node = fFindNodeById(flattenData, id)
+        const node = flattedDataMap.get(id) as any
 
         if (node) {
           const { children } = node
@@ -38,7 +39,8 @@ export const processCheckedIds = (
 
     case 'PARENT':
       return checkedIds.filter((id) => {
-        const node = fFindNodeById(flattenData, id) as any
+        const node = flattedDataMap.get(id) as any
+
         if (node) {
           // 向上递归遍历是否被勾选
           const ancestors = getNodeAncestors(node)