diff --git a/components/vc-tree-select/hooks/useTreeData.ts b/components/vc-tree-select/hooks/useTreeData.ts index d69c34fe4..a274f3173 100644 --- a/components/vc-tree-select/hooks/useTreeData.ts +++ b/components/vc-tree-select/hooks/useTreeData.ts @@ -64,7 +64,7 @@ function formatTreeData( function dig(dataNodes: DataNode[]) { return (dataNodes || []).map(node => { - const { key, disableCheckbox, disabled, checkable, selectable } = node; + const { key, disableCheckbox, disabled, checkable, selectable, isLeaf } = node; const value = node[fieldValue]; const mergedValue = fieldValue in node ? value : key; @@ -77,6 +77,7 @@ function formatTreeData( title: getLabelProp(node), node, selectable, + isLeaf, dataRef: node, checkable, }; diff --git a/components/vc-tree-select/interface.ts b/components/vc-tree-select/interface.ts index d14244a4b..c39d58239 100644 --- a/components/vc-tree-select/interface.ts +++ b/components/vc-tree-select/interface.ts @@ -37,6 +37,7 @@ export interface InternalDataEntity { disableCheckbox: boolean; disabled: boolean; selectable: boolean; + isLeaf: boolean; children?: InternalDataEntity[]; /** Origin DataNode */