From 28813c0fc424bb9ed16441eaf8e6a8280e969cc2 Mon Sep 17 00:00:00 2001
From: shizhongming <zhongming4762@hotmail.com>
Date: Tue, 5 Mar 2024 09:51:18 +0800
Subject: [PATCH] =?UTF-8?q?fix(=E7=B3=BB=E7=BB=9F=E6=A8=A1=E5=9D=97):=20?=
 =?UTF-8?q?=E8=8F=9C=E5=8D=95=E4=BF=9D=E5=AD=98=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/Tree/src/BasicTree.vue | 11 ++++++++++-
 src/components/Tree/src/types/tree.ts |  1 +
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/components/Tree/src/BasicTree.vue b/src/components/Tree/src/BasicTree.vue
index db178d366..27a950920 100644
--- a/src/components/Tree/src/BasicTree.vue
+++ b/src/components/Tree/src/BasicTree.vue
@@ -43,6 +43,9 @@
     setup(props, { attrs, slots, emit, expose }) {
       const [bem] = createBEM('tree');
 
+      const antInstanceRef = ref();
+      const getAntInstance = () => unref(antInstanceRef);
+
       const state = reactive<TreeState>({
         checkStrictly: props.checkStrictly,
         expandedKeys: props.expandedKeys || [],
@@ -337,6 +340,7 @@
         updateNodeByKey,
         getSelectedNode,
         checkAll,
+        getAntInstance,
         expandAll,
         filterByLevel: (level: number) => {
           state.expandedKeys = filterByLevel(level);
@@ -459,7 +463,12 @@
               tip="加载中..."
             >
               <ScrollContainer style={scrollStyle} v-show={!unref(getNotFound)}>
-                <Tree {...unref(getBindValues)} showIcon={false} treeData={treeData.value}>
+                <Tree
+                  {...unref(getBindValues)}
+                  showIcon={false}
+                  ref={antInstanceRef}
+                  treeData={treeData.value}
+                >
                   {extendSlots(slots, ['title'])}
                 </Tree>
               </ScrollContainer>
diff --git a/src/components/Tree/src/types/tree.ts b/src/components/Tree/src/types/tree.ts
index 604adb3e6..541b336a5 100644
--- a/src/components/Tree/src/types/tree.ts
+++ b/src/components/Tree/src/types/tree.ts
@@ -192,4 +192,5 @@ export interface TreeActionType {
     treeList?: TreeItem[],
     selectNode?: TreeItem | null,
   ) => TreeItem | null;
+  getAntInstance: () => any;
 }