|
@@ -84,10 +84,10 @@ public class ImageClassificationFileServiceImpl extends BaseServiceImpl<ImageCla
|
|
|
//查询下级节点信息
|
|
|
WbsTreeContract node = wbsTreeContractClient.getContractWbsTreeByPrimaryKeyId(Long.valueOf(imageClassificationFile.getWbsIdsStr()));
|
|
|
List<WbsTreeContract> wbsTreeContracts;
|
|
|
- if (node.getParentId() == 0) {
|
|
|
- wbsTreeContracts = wbsTreeContractClient.queryAllChild(imageClassificationFile.getContractId());
|
|
|
+ if (node.getPId() == 0) {
|
|
|
+ wbsTreeContracts = wbsTreeContractClient.queryAllChild(Long.parseLong(node.getContractId()));
|
|
|
} else {
|
|
|
- wbsTreeContracts = wbsTreeContractClient.queryCurrentNodeAllChild(imageClassificationFile.getContractId(), node.getId());
|
|
|
+ wbsTreeContracts = wbsTreeContractClient.queryCurrentNodeAllChild(Long.parseLong(node.getContractId()), node.getPKeyId());
|
|
|
}
|
|
|
wbsTreeContracts.add(node);
|
|
|
imageClassificationFile.setWbsIds(wbsTreeContracts.stream().map(wtc -> wtc.getPKeyId() + "").collect(Collectors.toList()));
|