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