|
@@ -165,14 +165,24 @@ public class EVisaTaskCheckController {
|
|
|
WbsTreeContract contract = wbsTreeContractClient.getContractWbsTreeByPrimaryKeyId(Long.valueOf(nodeId));
|
|
|
//获取节点下每个填报表
|
|
|
List<WbsTreeContract> node = wbsTreeContractClient.queryChildByParentId(contract, "", "");
|
|
|
- List<Long> ids = new ArrayList<>();
|
|
|
+ Set<Long> ids = new HashSet<>();
|
|
|
//获取每个填报表对应的wbs_tree_private的id
|
|
|
- for (WbsTreeContract treeContract : node) {
|
|
|
- WbsTreePrivate wbsTreePrivate = wbsTreePrivateClient.queryPeersNodeByProjectIdAndId(treeContract.getProjectId(), treeContract.getId());
|
|
|
- ids.add(wbsTreePrivate.getPKeyId());
|
|
|
-// if (StringUtils.isNotBlank(treeContract.getIsTypePrivatePid()+"")){
|
|
|
-// ids.add(treeContract.getIsTypePrivatePid());
|
|
|
-// }
|
|
|
+// for (WbsTreeContract treeContract : node) {
|
|
|
+// WbsTreePrivate wbsTreePrivate = wbsTreePrivateClient.queryPeersNodeByProjectIdAndId(treeContract.getProjectId(), treeContract.getId());
|
|
|
+// ids.add(wbsTreePrivate.getPKeyId());
|
|
|
+//// if (StringUtils.isNotBlank(treeContract.getIsTypePrivatePid()+"")){
|
|
|
+//// ids.add(treeContract.getIsTypePrivatePid());
|
|
|
+//// }
|
|
|
+// }
|
|
|
+ if (node != null && node.size() > 0){
|
|
|
+ List<Long> setIds = node.stream().map(l -> l.getId()).collect(Collectors.toList());
|
|
|
+ Set<Long> setPIds = node.stream().map(l -> l.getIsTypePrivatePid()).collect(Collectors.toSet());
|
|
|
+ ids.addAll(setPIds);
|
|
|
+ Set<Long> tableIds = wbsTreePrivateClient.getAllPrivateTableByIds(Long.parseLong(contract.getProjectId()), setIds);
|
|
|
+ if (tableIds != null && tableIds.size() > 0) {
|
|
|
+ ids.addAll(tableIds);
|
|
|
+ }
|
|
|
+ ids.addAll(tableIds);
|
|
|
}
|
|
|
JSONObject object = new JSONObject();
|
|
|
object.put("privatePKeyId", ids);
|
|
@@ -328,11 +338,20 @@ public class EVisaTaskCheckController {
|
|
|
WbsTreeContract contract = wbsTreeContractClient.getContractWbsTreeByPrimaryKeyId(Long.valueOf(nodeId));
|
|
|
//获取节点下每个填报表
|
|
|
List<WbsTreeContract> node = wbsTreeContractClient.queryChildByParentId(contract, "", "");
|
|
|
- List<Long> ids = new ArrayList<>();
|
|
|
+ Set<Long> ids = new HashSet<>();
|
|
|
//获取每个填报表对应的wbs_tree_private的id
|
|
|
- for (WbsTreeContract treeContract : node) {
|
|
|
- WbsTreePrivate wbsTreePrivate = wbsTreePrivateClient.queryPeersNodeByProjectIdAndId(treeContract.getProjectId(), treeContract.getId());
|
|
|
- ids.add(wbsTreePrivate.getPKeyId());
|
|
|
+// for (WbsTreeContract treeContract : node) {
|
|
|
+// WbsTreePrivate wbsTreePrivate = wbsTreePrivateClient.queryPeersNodeByProjectIdAndId(treeContract.getProjectId(), treeContract.getId());
|
|
|
+// ids.add(wbsTreePrivate.getPKeyId());
|
|
|
+// }
|
|
|
+ if (node != null && node.size() > 0){
|
|
|
+ List<Long> setIds = node.stream().map(l -> l.getId()).collect(Collectors.toList());
|
|
|
+ Set<Long> setPIds = node.stream().map(l -> l.getIsTypePrivatePid()).collect(Collectors.toSet());
|
|
|
+ ids.addAll(setPIds);
|
|
|
+ Set<Long> tableIds = wbsTreePrivateClient.getAllPrivateTableByIds(Long.parseLong(contract.getProjectId()), setIds);
|
|
|
+ if (tableIds != null && tableIds.size() > 0) {
|
|
|
+ ids.addAll(tableIds);
|
|
|
+ }
|
|
|
}
|
|
|
JSONObject object = new JSONObject();
|
|
|
object.put("privatePKeyId", ids);
|