linwe
2024-05-29 c10d6358b9f014375a13821465bc978d0c0da22e
src/main/java/org/springblade/modules/task/service/impl/TaskPlaceSelfCheckServiceImpl.java
@@ -48,6 +48,7 @@
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.stream.Collectors;
/**
@@ -137,30 +138,53 @@
   @Override
   @Transactional(rollbackFor = Exception.class)
   public Boolean savePlace(TaskPlaceSelfCheckVO taskPlaceSelfCheck)  {
   public Boolean savePlace(TaskPlaceSelfCheckVO taskPlaceSelfCheck) {
      taskPlaceSelfCheck.setCreateUser(AuthUtil.getUserId());
      // 1.保存任务表
      ITaskService bean2 = SpringUtils.getBean(ITaskService.class);
      Long aLong = bean2.saveTask(CommonConstant.NUMBER_ONE, DictConstant.FIRE_SELF_CHECK_NOTICE, 1,
         "", AuthUtil.getUserId(), taskPlaceSelfCheck.getHouseCode(), CommonConstant.NUMBER_SEVEN, 1);
      if (aLong <= 0) {
      ITaskService iTaskService = SpringUtils.getBean(ITaskService.class);
      Long restults;
      AtomicBoolean isState = new AtomicBoolean(false);
      List<TaskPlaceRecordVO> taskPlaceRecordList = taskPlaceSelfCheck.getTaskPlaceRecordVOList();
      for (TaskPlaceRecordVO taskPlaceRecordVO : taskPlaceRecordList) {
         // 判断类型
         if (StringUtils.isNotBlank(taskPlaceSelfCheck.getType()) && taskPlaceSelfCheck.getType().equals("2")) {
            taskPlaceRecordVO.setType("3");
         } else {
            taskPlaceRecordVO.setType("1");
         }
         // 判断是否有隐患
         if (taskPlaceRecordVO.getState().equals(0)) {
            isState.set(true);
         }
      }
      if (StringUtils.isNotBlank(taskPlaceSelfCheck.getType()) && taskPlaceSelfCheck.getType().equals("2")) {
         // 不存在隐患。状态直接为已审核
         if (!isState.get()) {
            taskPlaceSelfCheck.setStatus(2);
            taskPlaceSelfCheck.setRectificationFlag(3);
         }
         restults = iTaskService.saveTask(CommonConstant.NUMBER_ONE, DictConstant.FIRE_SCHOOL_CHECK_NOTICE, 1,
            "", AuthUtil.getUserId(), taskPlaceSelfCheck.getHouseCode(), CommonConstant.NUMBER_FOUR, taskPlaceSelfCheck.getStatus());
      } else {
         restults = iTaskService.saveTask(CommonConstant.NUMBER_ONE, DictConstant.FIRE_SELF_CHECK_NOTICE, 1,
            "", AuthUtil.getUserId(), taskPlaceSelfCheck.getHouseCode(), CommonConstant.NUMBER_SEVEN, 1);
      }
      if (restults <= 0) {
         return false;
      }
      taskPlaceSelfCheck.setTaskId(aLong);
      taskPlaceSelfCheck.setTaskId(restults);
      // 2.保存任务详情
      boolean save = save(taskPlaceSelfCheck);
      if (save) {
         // 3.保存题目记录
         List<TaskPlaceRecordVO> taskPlaceRecordList = taskPlaceSelfCheck.getTaskPlaceRecordVOList();
         ITaskPlaceRecordService bean = SpringUtil.getBean(ITaskPlaceRecordService.class);
         taskPlaceRecordList.stream().forEach(item -> {
            item.setTaskPlaceSelfCheckId(taskPlaceSelfCheck.getId());
            item.setCreateUser(AuthUtil.getUserId());
            item.setType("1");
         });
         List<TaskPlaceRecordEntity> collect = taskPlaceRecordList.stream().filter(item -> item.getState().equals(0)).collect(Collectors.toList());
         ITaskPlaceRecordService taskPlaceRecordService = SpringUtil.getBean(ITaskPlaceRecordService.class);
         List<TaskPlaceRecordEntity> collect = taskPlaceRecordList.stream().collect(Collectors.toList());
         if (collect.size() > 0) {
            bean.saveBatch(collect);
            taskPlaceRecordService.saveBatch(collect);
         }
         updateById(taskPlaceSelfCheck);
         return save;
@@ -184,10 +208,10 @@
         List<TaskPlaceRecordVO> taskPlaceRecordList = taskPlaceSelfCheck.getTaskPlaceRecordVOList();
         if (taskPlaceRecordList != null && taskPlaceRecordList.size() > 0) {
            ITaskPlaceRecordService bean = SpringUtil.getBean(ITaskPlaceRecordService.class);
            List<TaskPlaceRecordEntity> copy = BeanUtil.copy(taskPlaceRecordList, TaskPlaceRecordEntity.class);
            boolean b = bean.saveOrUpdateBatch(copy);
            if (b) {
               return b;
            List<TaskPlaceRecordEntity> placeRecordEntityList = BeanUtil.copy(taskPlaceRecordList, TaskPlaceRecordEntity.class);
            boolean updateBatch = bean.saveOrUpdateBatch(placeRecordEntityList);
            if (updateBatch) {
               return updateBatch;
            }
            throw new CustomException("保存失败!");
         }
@@ -245,17 +269,17 @@
   }
   @Override
   public Integer getCount(String neiCode, int i) {
   public Integer getCount(String neiCode, int status, int type) {
      // 获取请求头中的角色别名
      String roleName = SpringUtils.getRequestParam("roleName");
      if (AuthUtils.isMj(roleName)) {
         return baseMapper.getCount(neiCode, i, AuthUtil.getUserId());
         return baseMapper.getCount(neiCode, status, AuthUtil.getUserId(), type);
      }
      return 0;
   }
   @Override
   public TaskPlaceSelfCheckEntity getDetailByTaskId(Long taskId) {
      return baseMapper.getDetailByTaskId( taskId);
      return baseMapper.getDetailByTaskId(taskId);
   }
}