fix(table): allow selectedRowKeys include the keys which not exist in…

… data
Tencent · Jul 25, 2023 · 0cf784a · 0cf784a
1 parent 2f76f1b
commit 0cf784a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/table/_example/tree-select.jsx b/src/table/_example/tree-select.jsx
@@ -78,10 +78,12 @@ const columns = [
   // { colKey: 'email', title: '邮箱地址' },
 ];
 
+const defaultSelectedRowKeys = [];
+
 export default function TableSingleSort() {
   const [data, setData] = useState([...initData]);
-  const [selectedRowKeys, setSelectedRowKeys] = useState([]);
-  const [checkStrictly, setCheckStrictly] = useState(true);
+  const [selectedRowKeys, setSelectedRowKeys] = useState(defaultSelectedRowKeys);
+  const [checkStrictly, setCheckStrictly] = useState(false);
   const [expandedRowKeys, setExpandedRowKeys] = useState([]);
   const treeTableRef = useRef(null);
 
@@ -92,7 +94,7 @@ export default function TableSingleSort() {
 
   useEffect(
     () => {
-      setSelectedRowKeys([]);
+      setSelectedRowKeys(defaultSelectedRowKeys);
       setData(cloneDeep(data));
     },
     // eslint-disable-next-line react-hooks/exhaustive-deps

diff --git a/src/table/hooks/useTreeSelect.tsx b/src/table/hooks/useTreeSelect.tsx
@@ -133,6 +133,7 @@ export default function useTreeSelect(props: TdEnhancedTableProps, treeDataMap:
     for (let i = 0, len = tSelectedRowKeys.length; i < len; i++) {
       const rowValue = tSelectedRowKeys[i];
       const state = treeDataMap.get(rowValue);
+      if (!state) continue;
       const children = get(state.row, rowDataKeys.childrenKey);
       // 根据选中的叶子结点计算父节点半选状态
       if (!children || !children.length) {