fix: 创建 treeStore时传入 keys

Tencent · Apr 8, 2022 · 677f4e2 · 677f4e2
1 parent db6b79c
commit 677f4e2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/cascader/Cascader.tsx b/src/cascader/Cascader.tsx
@@ -93,15 +93,21 @@ const Cascader: React.FC<CascaderProps> = (props) => {
   /**
    * build tree
    */
-  const { disabled, options = [], keys, checkStrictly = false, lazy = true, load, valueMode = 'onlyLeaf' } = props;
+  const { disabled, options = [], keys = {}, checkStrictly = false, lazy = true, load, valueMode = 'onlyLeaf' } = props;
 
   useEffect(() => {
     if (!options.length) return;
     if (!treeStore) {
       const createStore = (onLoad: () => void) => {
+        const treePropsKeys = {
+          ...keys,
+          children: typeof keys.children === 'string' ? keys.children : 'children',
+        };
+
         const treeProps = {
           onLoad,
           options,
+          keys: treePropsKeys,
         };
         const store = new TreeStore(treeProps);
         store.append(options);
@@ -119,7 +125,7 @@ const Cascader: React.FC<CascaderProps> = (props) => {
       treeStoreExpendEffect(treeStore, value, []);
       treeNodesEffect(inputVal, treeStore, setTreeNodes);
     }
-  }, [inputVal, options, value, treeStore]);
+  }, [inputVal, options, value, treeStore, keys]);
 
   useEffect(() => {
     if (!treeStore) return;