修改站点树接口返回值结构,增加层级

common-lib/src/main/java/com/mortals/xhx/common/pdu/site/SiteTreeSelectVO.java

@@ -50,6 +50,8 @@ public class SiteTreeSelectVO implements Serializable {
      * 图标
      */
     private String icon;
+    /** 层级 */
+    private Integer level;
 
     /**
      * 子节点

sst-manager/src/main/java/com/mortals/xhx/module/sst/web/SstBasicController.java

@@ -203,8 +203,9 @@ public class SstBasicController extends BaseCRUDJsonBodyMappingController<SstBas
                 rootNode.setIsLeaf(false);
                 rootNode.setLatitude("30.595081");
                 rootNode.setLongitude("104.062983");
-                rootNode.setSiteCount(getChildrenCount(resultList)-1);
+                rootNode.setSiteCount(getChildrenCount(resultList,2)-1);
                 rootNode.setChildren(resultList);
+                rootNode.setLevel(1);
                 List<SiteTreeSelectVO> dataList = new ArrayList<>();
                 dataList.add(rootNode);
                 model.put("data", dataList);
@@ -225,14 +226,15 @@ public class SstBasicController extends BaseCRUDJsonBodyMappingController<SstBas
         return ret;
     }
 
-    private static Integer getChildrenCount(List<SiteTreeSelectVO> children){
+    private static Integer getChildrenCount(List<SiteTreeSelectVO> children,int initLevel){
         if(CollectionUtils.isEmpty(children)) {
             return 1;
         }else {
             int size = 1;
             for (SiteTreeSelectVO vo:children){
-                int c = getChildrenCount(vo.getChildren());
+                int c = getChildrenCount(vo.getChildren(),initLevel+1);
                 vo.setSiteCount(c);
+                vo.setLevel(initLevel);
                 size += c;
             }
             return size;