From bef55f395a3a89827cd58ca4729405a38c77356b Mon Sep 17 00:00:00 2001
From: "zhaokaiyao@91isoft.com" <6922339zky>
Date: Wed, 20 Oct 2021 14:52:07 +0800
Subject: [PATCH] =?UTF-8?q?TODO=E6=A8=A1=E5=9D=97=E5=8D=95=E5=85=83?=
 =?UTF-8?q?=E6=A0=91=E7=BB=93=E6=9E=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/SarModelTreeController.java    |  2 +-
 .../impl/SarModuleTreeServiceImpl.java        | 56 +++++++++++++++----
 2 files changed, 45 insertions(+), 13 deletions(-)

diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/controller/SarModelTreeController.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/controller/SarModelTreeController.java
index dcbc369e..53ca743f 100644
--- a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/controller/SarModelTreeController.java
+++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/controller/SarModelTreeController.java
@@ -42,7 +42,7 @@ public class SarModelTreeController extends BaseController<SarModelTree> {
     @GetMapping("/list")
     public ResponseMessage<List<SarModelTree>> getAll(){
         List<SarModelTree> tsResources = iSarModelTreeService.getAll(null);
-//        List<SarModelTree> tsResources = iSarModuleTreeService.getAll(null);
+//        TODO List<SarModelTree> tsResources = iSarModuleTreeService.getAll(null);
         return Result.success(tsResources);
     }
 
diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/service/impl/SarModuleTreeServiceImpl.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/service/impl/SarModuleTreeServiceImpl.java
index 7a44d284..2fc5df32 100644
--- a/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/service/impl/SarModuleTreeServiceImpl.java
+++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/sarModelTree/service/impl/SarModuleTreeServiceImpl.java
@@ -17,6 +17,7 @@ import org.springframework.stereotype.Service;
 
 import java.util.LinkedList;
 import java.util.List;
+import java.util.Map;
 import java.util.stream.Collectors;
 
 @Service
@@ -25,27 +26,38 @@ public class SarModuleTreeServiceImpl extends ServiceImpl<SarModuleTreeDao, SarM
     @Autowired()
     private SarModuleTreeServiceImpl  sarModuleTreeService;
 
-
+    /**
+     * @param sarModelTree
+     * @return 所有的根节点
+     */
     @Override
     public List<SarModelTree> getAll(SarModelTree sarModelTree) {
         QueryWrapper<SarModuleTree> treeQueryWrapper = new QueryWrapper<>();
         treeQueryWrapper.groupBy("FTVSTYPE1");
         List<SarModuleTree> moduleTreeList = this.baseMapper.selectList(treeQueryWrapper);
-
-        //把SarModuleTree类型的list集合转换为SarModelTree类型的list集合
-        List<SarModelTree> sarModelTreeList = moduleTreeList.stream().map(SarModuleTree -> {
-            SarModelTree modelTree = new SarModelTree();
-            modelTree.setId(SarModuleTree.getGuid());
-            modelTree.setName(SarModuleTree.getFtvstype1());
-//            modelTree.setModel("TS"+SarModuleTree.get);
-            return modelTree;
-        }).collect(Collectors.toList());
-
-        return sarModelTreeList;
+        //返回转换元素后的数组
+        return listTransform(moduleTreeList);
     }
 
+    /**
+     * 根据根节点查询其所有的子节点
+     * @param parent
+     * @return
+     */
     @Override
     public List<SarModelTree> recursionGetChildren(SarModelTree parent) {
+
+        QueryWrapper<SarModuleTree> treeQueryWrapper = new QueryWrapper<>();
+        treeQueryWrapper.eq("GUID",parent.getId());
+        List<SarModuleTree> moduleTreeList = this.baseMapper.selectList(treeQueryWrapper);
+
+
+        Map<String, List<SarModuleTree>> collect = moduleTreeList.stream().collect(Collectors.groupingBy(SarModuleTree::getFtvstype1));
+        collect.forEach((k, v)->{
+            SarModelTree modelTree = new SarModelTree();
+            //TODO 生成树型结构
+//            modelTree
+        });
         return null;
     }
 
@@ -93,5 +105,25 @@ public class SarModuleTreeServiceImpl extends ServiceImpl<SarModuleTreeDao, SarM
         return head;
     }
 
+    /**
+     *  把SarModuleTree类型的list集合转换为SarModelTree类型的list集合,这样就不用修改前端代码( •̀ ω •́ )✧
+     *  --->对修改关闭,对扩展开放
+     * @param moduleTreeList
+     * @return
+     */
+    public List<SarModelTree> listTransform(List<SarModuleTree> moduleTreeList){
+
+       return  moduleTreeList.stream().map(SarModuleTree -> {
+            SarModelTree modelTree = new SarModelTree();
+            modelTree.setId(SarModuleTree.getGuid());
+            modelTree.setName(SarModuleTree.getFtvstype1());
+            modelTree.setModel("TS"
+                    +SarModuleTree.getFtvstype2().substring(SarModuleTree.getFtvstype2().length()-2)
+                    +SarModuleTree.getSeriesid()
+                    +SarModuleTree.getReserved()
+                    +SarModuleTree.getStarted()); //"TS"+SF_ID+SE_ID+SU_ID 拼接模块编号ID
+            return modelTree;
+        }).collect(Collectors.toList());
+    }
 
 }