From 67066e2fb6fab2ab2cb1e037be35935cd85a6f4a Mon Sep 17 00:00:00 2001
From: tangjinzhou <415800467@qq.com>
Date: Sat, 20 Nov 2021 21:45:28 +0800
Subject: [PATCH] fix: treeSelect isLeaf not work #4883 close #4883

---
 components/vc-tree-select/hooks/useTreeData.ts | 3 ++-
 components/vc-tree-select/interface.ts         | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

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 */