diff --git a/huzhou/src/main/java/com/easy/admin/modules/huzhou/service/impl/HuzhouContractinfoServiceImpl.java b/huzhou/src/main/java/com/easy/admin/modules/huzhou/service/impl/HuzhouContractinfoServiceImpl.java index 0d619a0..6114d63 100644 --- a/huzhou/src/main/java/com/easy/admin/modules/huzhou/service/impl/HuzhouContractinfoServiceImpl.java +++ b/huzhou/src/main/java/com/easy/admin/modules/huzhou/service/impl/HuzhouContractinfoServiceImpl.java @@ -281,12 +281,20 @@ public class HuzhouContractinfoServiceImpl extends ServiceImpl getContractinfoByprojectIdAndTaskName(String projectid, String taskName) { - LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.eq(HuzhouContractinfo::getProjectId, projectid); - queryWrapper.eq(HuzhouContractinfo::getTaskName, taskName); - queryWrapper.eq(HuzhouContractinfo::getFlag,"实际"); - List list = list(queryWrapper); - return list; + if (taskName.equals("合同签订")){ + LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); + queryWrapper.eq(HuzhouContractinfo::getProjectId, projectid); + queryWrapper.eq(HuzhouContractinfo::getFlag,"计划"); + List list1 = list(queryWrapper); + return list1; + }else { + LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); + queryWrapper.eq(HuzhouContractinfo::getProjectId, projectid); + queryWrapper.eq(HuzhouContractinfo::getTaskName, taskName); + queryWrapper.eq(HuzhouContractinfo::getFlag,"实际"); + List list2 = list(queryWrapper); + return list2; + } } @Override @@ -365,28 +373,38 @@ public class HuzhouContractinfoServiceImpl extends ServiceImpl queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.eq(HuzhouContractinfo::getProjectId, projectId); - queryWrapper.eq(HuzhouContractinfo::getTaskName, taskName); - List rs = list(queryWrapper); - if (rs.size() == 1) { - // 返回合同信息(计划) + if (taskName.equals("合同签订")){ ContractInfoVo contractInfoVo = new ContractInfoVo(); List planList = this.getContractinfoByprojectId(projectId); contractInfoVo.setHasInfo(true); contractInfoVo.setPlanList(planList); return contractInfoVo; } - if (rs.size()>1){ - // 返回合同信息(计划 + 实际) - ContractInfoVo contractInfoVo = new ContractInfoVo(); - List planList = this.getContractinfoByprojectId(projectId); - List realList = this.getRealContractinfoByprojectId(projectId); - contractInfoVo.setHasInfo(true); - contractInfoVo.setPlanList(planList); - contractInfoVo.setRealList(realList); - return contractInfoVo; + else { + LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); + queryWrapper.eq(HuzhouContractinfo::getProjectId, projectId); + queryWrapper.eq(HuzhouContractinfo::getTaskName, taskName); + List rs = list(queryWrapper); + if (rs.size() == 1) { + // 返回合同信息(计划) + ContractInfoVo contractInfoVo = new ContractInfoVo(); + List planList = this.getContractinfoByprojectId(projectId); + contractInfoVo.setHasInfo(true); + contractInfoVo.setPlanList(planList); + return contractInfoVo; + } + if (rs.size()>1){ + // 返回合同信息(计划 + 实际) + ContractInfoVo contractInfoVo = new ContractInfoVo(); + List planList = this.getContractinfoByprojectId(projectId); + List realList = this.getRealContractinfoByprojectId(projectId); + contractInfoVo.setHasInfo(true); + contractInfoVo.setPlanList(planList); + contractInfoVo.setRealList(realList); + return contractInfoVo; + } } + ContractInfoVo contractInfoVo = new ContractInfoVo(); contractInfoVo.setHasInfo(false); return contractInfoVo;