划分树问题

blade-service-api/blade-manager-api/src/main/java/org/springblade/manager/vo/WbsTreeContractTreeVOS.java

@@ -37,6 +37,9 @@ public class WbsTreeContractTreeVOS {
     @ApiModelProperty("是否是工序 true为工序")
     private Boolean leaf;
 
+    @ApiModelProperty("是否有下级")
+    private Boolean exsitChild;
+
     @ApiModelProperty("节点类型")
     private Integer deptCategory;
 

blade-service/blade-manager/src/main/java/org/springblade/manager/feign/WbsTreeContractClientImpl.java

@@ -58,7 +58,12 @@ public class WbsTreeContractClientImpl implements WbsTreeContractClient {
 
         //不知道为什么通过远程调用时实体字段不是想要的，所以中间用字段相同且不继承其它实体的类转换一次
         List<WbsTreeContractTreeVOS> vos = JSONArray.parseArray(JSONObject.toJSONString(vo), WbsTreeContractTreeVOS.class);
-        vos.forEach(voData -> voData.setLeaf(new Integer("6").equals(voData.getDeptCategory())));
+        vos.forEach(voData -> {
+            voData.setLeaf(new Integer("6").equals(voData.getDeptCategory()));
+            //检查是否有下级
+            List<WbsTreeContractTreeVO> childs = this.contractInfoService.queryContractWbsTreeByContractIdAndType(contractId, wbsType, voData.getId());
+            voData.setExsitChild(childs == null || childs.size() == 0);
+        });
         return vos;
     }