diff --git a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/obj/web/ObjectController.java b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/obj/web/ObjectController.java index 643859c..22383d9 100644 --- a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/obj/web/ObjectController.java +++ b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/obj/web/ObjectController.java @@ -7,18 +7,6 @@ import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport; -import org.springblade.core.mp.support.Condition; -import org.springblade.hzinfo_inspect.obj.entity.ObjectDangerEntity; -import org.springblade.hzinfo_inspect.obj.entity.ObjectEntity; -import org.springblade.hzinfo_inspect.obj.entity.ObjectTemplateEntity; -import org.springblade.hzinfo_inspect.obj.entity.OtherObjectEntity; -import org.springblade.hzinfo_inspect.obj.services.*; -import org.springblade.hzinfo_inspect.obj.utils.CodeUtils; -import org.springblade.hzinfo_inspect.obj.vo.*; -import org.springblade.hzinfo_inspect.plan.PlanContants; -import org.springblade.hzinfo_inspect.plan.entity.PlanObjectTemplateEntity; -import org.springblade.hzinfo_inspect.plan.service.IPlanObjectTemplateService; -import org.springblade.hzinfo_inspect.task.TaskContants; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import io.swagger.annotations.ApiParam; @@ -26,12 +14,24 @@ import lombok.AllArgsConstructor; import org.apache.commons.io.FileUtils; import org.springblade.core.boot.ctrl.BladeController; import org.springblade.core.log.exception.ServiceException; +import org.springblade.core.mp.support.Condition; import org.springblade.core.mp.support.Query; import org.springblade.core.secure.BladeUser; import org.springblade.core.tool.api.R; -import org.springblade.core.tool.constant.BladeConstant; import org.springblade.core.tool.utils.CollectionUtil; import org.springblade.core.tool.utils.Func; +import org.springblade.hzinfo_inspect.obj.entity.ObjectDangerEntity; +import org.springblade.hzinfo_inspect.obj.entity.ObjectEntity; +import org.springblade.hzinfo_inspect.obj.entity.ObjectTemplateEntity; +import org.springblade.hzinfo_inspect.obj.entity.OtherObjectEntity; +import org.springblade.hzinfo_inspect.obj.services.*; +import org.springblade.hzinfo_inspect.obj.utils.CodeUtils; +import org.springblade.hzinfo_inspect.obj.vo.ObjectBatchVo; +import org.springblade.hzinfo_inspect.obj.vo.ObjectExcelInputData; +import org.springblade.hzinfo_inspect.obj.vo.ObjectListQueryVO; +import org.springblade.hzinfo_inspect.obj.vo.OtherObjectListQueryVO; +import org.springblade.hzinfo_inspect.plan.PlanContants; +import org.springblade.hzinfo_inspect.plan.service.IPlanObjectTemplateService; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; @@ -40,10 +40,9 @@ import javax.validation.Valid; import java.io.File; import java.io.IOException; import java.io.OutputStream; -import java.util.ArrayList; import java.util.Arrays; +import java.util.Collections; import java.util.List; -import java.util.stream.Collectors; /** @@ -164,11 +163,8 @@ public class ObjectController extends BladeController { @ApiOperation(value = "逻辑删除", notes = "传入ids") public R remove(@ApiParam(value = "主键集合", required = true) @RequestParam String ids) { List delIdList = Func.toLongList(ids); - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().in(PlanObjectTemplateEntity::getObjectId, delIdList).eq(PlanObjectTemplateEntity::getIsDeleted, 0); - int count = planObjectTemplateService.count(qw); - if (count > 0) { - return R.fail("该巡检对象已关联模板,请先取消关联"); + if (planObjectTemplateService.hasRelatedPlan(delIdList)) { + return R.fail("该巡检对象存在使用中任务,请删除任务,或等待任务完成"); } for (Long aLong : delIdList) { objectTemplateService.remove(new QueryWrapper() {{ @@ -183,11 +179,8 @@ public class ObjectController extends BladeController { @ApiOperation(value = "启用/停用", notes = "传入id,以及目标状态0-启用,1-停用") public R disableOrEnable(@RequestParam Long id, @RequestParam Integer status) { if (status == 1) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(PlanObjectTemplateEntity::getObjectId, id).eq(PlanObjectTemplateEntity::getIsDeleted, 0); - int count = planObjectTemplateService.count(qw); - if (count > 0) { - return R.fail("该巡检对象已关联模板,请先取消关联"); + if (planObjectTemplateService.hasRelatedPlan(Collections.singletonList(id))) { + return R.fail("该巡检对象存在使用中任务,请删除任务,或等待任务完成"); } } UpdateWrapper ew = new UpdateWrapper<>(); diff --git a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/mapper/PlanObjectTemplateMapper.java b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/mapper/PlanObjectTemplateMapper.java index ce35895..ef3ac9a 100644 --- a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/mapper/PlanObjectTemplateMapper.java +++ b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/mapper/PlanObjectTemplateMapper.java @@ -2,14 +2,18 @@ package org.springblade.hzinfo_inspect.plan.mapper; import com.baomidou.mybatisplus.core.mapper.BaseMapper; import org.apache.ibatis.annotations.Mapper; +import org.apache.ibatis.annotations.Param; import org.springblade.hzinfo_inspect.plan.entity.PlanObjectTemplateEntity; +import java.util.List; + /** - * Mapper 接口 + * Mapper 接口 * * @author Chill */ @Mapper public interface PlanObjectTemplateMapper extends BaseMapper { + long hasRelatedPlan(@Param("objectId") List objectId, @Param("dateStr") String dateStr); } diff --git a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/mapper/PlanObjectTemplateMapper.xml b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/mapper/PlanObjectTemplateMapper.xml new file mode 100644 index 0000000..238202b --- /dev/null +++ b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/mapper/PlanObjectTemplateMapper.xml @@ -0,0 +1,22 @@ + + + + + + \ No newline at end of file diff --git a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/service/IPlanObjectTemplateService.java b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/service/IPlanObjectTemplateService.java index 3a6568b..cbb2d6f 100644 --- a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/service/IPlanObjectTemplateService.java +++ b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/service/IPlanObjectTemplateService.java @@ -1,14 +1,15 @@ package org.springblade.hzinfo_inspect.plan.service; import org.springblade.core.mp.base.BaseService; - import org.springblade.hzinfo_inspect.plan.entity.PlanObjectTemplateEntity; +import java.util.List; + /** - * 巡检计划对象模板服务类 + * 巡检计划对象模板服务类 * * @author Chill */ public interface IPlanObjectTemplateService extends BaseService { - + boolean hasRelatedPlan(List objectId); } diff --git a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/service/impl/PlanObjectTemplateServiceImpl.java b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/service/impl/PlanObjectTemplateServiceImpl.java index 9dd48ce..78bd1e8 100644 --- a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/service/impl/PlanObjectTemplateServiceImpl.java +++ b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/plan/service/impl/PlanObjectTemplateServiceImpl.java @@ -6,13 +6,23 @@ import org.springblade.hzinfo_inspect.plan.mapper.PlanObjectTemplateMapper; import org.springblade.hzinfo_inspect.plan.service.IPlanObjectTemplateService; import org.springframework.stereotype.Service; +import java.time.LocalDateTime; +import java.time.format.DateTimeFormatter; +import java.util.List; + /** - * 巡检计划对象模板服务实现类 + * 巡检计划对象模板服务实现类 * * @author Chill */ @Service public class PlanObjectTemplateServiceImpl extends BaseServiceImpl implements IPlanObjectTemplateService { + @Override + public boolean hasRelatedPlan(List objectId) { + LocalDateTime now = LocalDateTime.now(); + String dateStr = now.format(DateTimeFormatter.ofPattern("yyyy-MM-dd")); + return this.baseMapper.hasRelatedPlan(objectId, dateStr) > 0; + } } diff --git a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/task/job/TaskJob.java b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/task/job/TaskJob.java index c2ae9d4..3bc8168 100644 --- a/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/task/job/TaskJob.java +++ b/HZInfo-RIS-PXHD/hzinfo-ris-pxhd-service/hzinfo-inspect-service/src/main/java/org/springblade/hzinfo_inspect/task/job/TaskJob.java @@ -48,14 +48,11 @@ public class TaskJob { private ITaskService taskService; @Autowired private TaskRemindManager remindManager; -// @Autowired -// private IRobotTaskService robotTaskService; /** * 派发任务 */ @Scheduled(cron = "30 0 0 * * ?") -// @XxlJob("taskJobDistribute") @Transactional public ReturnT taskDistribute() throws Exception { // log.info("=====================派发任务===================");