diff --git a/xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/contractor/service/impl/SubContractorServiceImpl.java b/xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/contractor/service/impl/SubContractorServiceImpl.java index 3cc821d2..d87df117 100644 --- a/xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/contractor/service/impl/SubContractorServiceImpl.java +++ b/xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/contractor/service/impl/SubContractorServiceImpl.java @@ -14,7 +14,6 @@ import org.dromara.common.core.utils.ObjectUtils; import org.dromara.common.core.utils.StringUtils; import org.dromara.common.mybatis.core.page.PageQuery; import org.dromara.common.mybatis.core.page.TableDataInfo; -import org.dromara.common.satoken.utils.LoginHelper; import org.dromara.contractor.domain.SubConstructionUser; import org.dromara.contractor.domain.SubContractor; import org.dromara.contractor.domain.dto.contractor.SubContractorCreateReq; @@ -118,16 +117,19 @@ public class SubContractorServiceImpl extends ServiceImpl 0) { - throw new ServiceException("分包单位名称重复", HttpStatus.BAD_REQUEST); - } Long projectId = req.getProjectId(); BusProject project = projectService.getById(projectId); if (project == null) { throw new ServiceException("项目不存在", HttpStatus.NOT_FOUND); } + String name = req.getName(); + Long count = this.lambdaQuery() + .eq(SubContractor::getName, name) + .eq(SubContractor::getProjectId, projectId) + .count(); + if (count > 0) { + throw new ServiceException("分包单位名称重复", HttpStatus.BAD_REQUEST); + } // 操作数据库 boolean save = this.save(contractor); if (!save) { @@ -160,7 +162,10 @@ public class SubContractorServiceImpl extends ServiceImpl 0) { throw new ServiceException("分包单位名称重复", HttpStatus.BAD_REQUEST); }