fix: tree-select filter error

ocxers · Oct 29, 2021 · 7c86e9e · 7c86e9e
1 parent 95b785a
commit 7c86e9e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/components/TreeSelect/tree-select.tsx b/components/TreeSelect/tree-select.tsx
@@ -158,19 +158,19 @@ const TreeSelect: ForwardRefRenderFunction<
   }, [inputValue]);
 
   const searchKeys = useMemo<string[]>(() => {
-    const newKeys = new Set(hitKeys);
+    let newKeys: string[] = [];
     if (inputValue) {
       for (const key in key2nodeProps) {
         const item = key2nodeProps[key];
 
-        if (newKeys.has(key)) {
-          item.pathParentKeys.forEach((_key) => newKeys.add(_key));
-        } else if (newKeys.has(item.parentKey)) {
-          newKeys.add(key);
+        const pathKeys = [...item.pathParentKeys, key];
+
+        if (pathKeys.some((_key) => hitKeys && hitKeys.has(_key))) {
+          newKeys = newKeys.concat(pathKeys);
         }
       }
     }
-    return Array.from(newKeys);
+    return Array.from(new Set(newKeys));
   }, [inputValue, key2nodeProps, hitKeys]);
 
   useEffect(() => {