|
@@ -504,7 +504,7 @@ public class InformationWriteQueryController extends BladeController {
|
|
|
this.setRootNodeName(parentId, childList);
|
|
|
if(childList != null && childList.size() == 1){
|
|
|
//需要向下展开
|
|
|
- this.foreachQueryChildNode(childList, childList.get(0).getContractIdRelation(), 1);
|
|
|
+ this.foreachQueryChildNode(childList, contractId, childList.get(0).getContractIdRelation());
|
|
|
}
|
|
|
if(StringUtils.isEmpty(contractIdRelation) && "0".equals(parentId)){
|
|
|
//给个顶点
|
|
@@ -526,7 +526,7 @@ public class InformationWriteQueryController extends BladeController {
|
|
|
result = this.wbsTreeContractClient.queryContractWbsTreeByContractIdAndType(contractId, 1, parentId);
|
|
|
//判断当前节点下是不是只有一个子节点
|
|
|
if(result != null && result.size() == 1){
|
|
|
- this.foreachQueryChildNode(result, contractId, contractInfo.getContractType());
|
|
|
+ this.foreachQueryChildNode(result, contractId);
|
|
|
}
|
|
|
}
|
|
|
}
|
|
@@ -559,6 +559,24 @@ public class InformationWriteQueryController extends BladeController {
|
|
|
return R.data(result);
|
|
|
}
|
|
|
|
|
|
+ /**
|
|
|
+ * 监理查询下级节点
|
|
|
+ */
|
|
|
+ private void foreachQueryChildNode(List<WbsTreeContractTreeVOS> result, String contractId, String contractIdRelation){
|
|
|
+ result.forEach(vos -> {
|
|
|
+ //查询子节点
|
|
|
+ List<WbsTreeContractTreeVOS> childList = this.wbsTreeContractClient.lazyTree(Long.parseLong(vos.getPrimaryKeyId()), contractId, contractIdRelation, 2);
|
|
|
+ if(childList != null && childList.size() == 1){
|
|
|
+ this.foreachQueryChildNode(childList, contractId, contractIdRelation);
|
|
|
+ }
|
|
|
+ //判断当前节点是否被标记为首件
|
|
|
+ TreeContractFirst first = this.treeContractFirstService.getOne(Wrappers.<TreeContractFirst>lambdaQuery().eq(TreeContractFirst::getIsDeleted, 0).eq(TreeContractFirst::getWbsNodeId, vos.getId()));
|
|
|
+ vos.setIsFirst(first != null);
|
|
|
+
|
|
|
+ vos.setChildren(childList);
|
|
|
+ });
|
|
|
+ }
|
|
|
+
|
|
|
/**
|
|
|
* 循环处理节点颜色
|
|
|
*/
|
|
@@ -593,18 +611,15 @@ public class InformationWriteQueryController extends BladeController {
|
|
|
* 如果子节点只有一个,则进一步查询该子节点的下级节点
|
|
|
* @param result 子节点集合
|
|
|
*/
|
|
|
- private void foreachQueryChildNode(List<WbsTreeContractTreeVOS> result, String contractId, Integer initialContractType){
|
|
|
+ private void foreachQueryChildNode(List<WbsTreeContractTreeVOS> result, String contractId){
|
|
|
result.forEach(vos -> {
|
|
|
- String id = new Integer("2").equals(initialContractType) ? vos.getPrimaryKeyId() : vos.getId();
|
|
|
- String primaryKeyId = new Integer("2").equals(initialContractType) ? vos.getId() : vos.getPrimaryKeyId();
|
|
|
-
|
|
|
- List<WbsTreeContractTreeVOS> child = this.wbsTreeContractClient.queryContractWbsTreeByContractIdAndType(contractId, 1, id);
|
|
|
+ List<WbsTreeContractTreeVOS> child = this.wbsTreeContractClient.queryContractWbsTreeByContractIdAndType(contractId, 1, vos.getId());
|
|
|
if(child != null && child.size() == 1){
|
|
|
//如果子节点还是只有一个,则进一步向下查询
|
|
|
- this.foreachQueryChildNode(child, contractId, initialContractType);
|
|
|
+ this.foreachQueryChildNode(child, contractId);
|
|
|
}
|
|
|
//判断当前节点是否被标记为首件
|
|
|
- TreeContractFirst first = this.treeContractFirstService.getOne(Wrappers.<TreeContractFirst>lambdaQuery().eq(TreeContractFirst::getIsDeleted, 0).eq(TreeContractFirst::getWbsNodeId, primaryKeyId));
|
|
|
+ TreeContractFirst first = this.treeContractFirstService.getOne(Wrappers.<TreeContractFirst>lambdaQuery().eq(TreeContractFirst::getIsDeleted, 0).eq(TreeContractFirst::getWbsNodeId, vos.getPrimaryKeyId()));
|
|
|
vos.setIsFirst(first != null);
|
|
|
|
|
|
vos.setChildren(child);
|