From 419f519430317848ae202c32a0f2e41a62c03d5c Mon Sep 17 00:00:00 2001
From: "zhaokaiyao@91isoft.com" <6922339zky>
Date: Sat, 27 Nov 2021 14:14:18 +0800
Subject: [PATCH] =?UTF-8?q?=E5=85=B3=E8=81=94=E6=A8=A1=E5=9D=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../slrs/sarModelTree/entity/SarModelTree.java   |  7 +++++++
 .../service/impl/SarModelTreeServiceImpl.java    | 16 ++++++++++++++++
 2 files changed, 23 insertions(+)

diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/entity/SarModelTree.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/entity/SarModelTree.java
index 0b68d34a..670b4fc7 100644
--- a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/entity/SarModelTree.java
+++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/entity/SarModelTree.java
@@ -65,4 +65,11 @@ public class SarModelTree extends BaseEntity {
     @TableField(exist=false)
     private List<String> childMenuIds;
 
+    @TableField("HAS_CHILDREN")
+    private String hasChildren;
+
+    @TableField(exist = false)
+    private  boolean leaf;
+
+
 }
diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/service/impl/SarModelTreeServiceImpl.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/service/impl/SarModelTreeServiceImpl.java
index 0635dd2d..faba2999 100644
--- a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/service/impl/SarModelTreeServiceImpl.java
+++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/service/impl/SarModelTreeServiceImpl.java
@@ -33,6 +33,14 @@ public class SarModelTreeServiceImpl extends ServiceImpl<SarModelTreeDao, SarMod
 //            tree.setChildren(recursionGetChildren((tree)));
 //        }
         list.forEach(item->{
+            switch (item.getHasChildren()){
+                case "0":
+                    item.setLeaf(true);
+                    break;
+                case "1":
+                    item.setLeaf(false);
+                    break;
+            }
             item.setId(item.getName());
         });
         return list;
@@ -55,6 +63,14 @@ public class SarModelTreeServiceImpl extends ServiceImpl<SarModelTreeDao, SarMod
 //        }
 
         children.forEach(item->{
+            switch (item.getHasChildren()){
+                case "0":
+                    item.setLeaf(true);
+                    break;
+                case "1":
+                    item.setLeaf(false);
+                    break;
+            }
             item.setId(item.getName());
         });
         return children;