消防检查优化

linwei

2024-02-03 b7df6735bcb2d447cae7fcfba2d12accf3dd2fcc

 src/main/java/org/springblade/modules/place/service/impl/PlaceCheckServiceImpl.java

@@ -17,13 +17,33 @@
package org.springblade.modules.place.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.xxl.job.core.util.FileUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springblade.common.constant.CommonConstant;
import org.springblade.common.constant.DictConstant;
import org.springblade.common.utils.SpringUtils;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.core.tool.utils.SpringUtil;
import org.springblade.modules.patrol.entity.PatrolRecord;
import org.springblade.modules.patrol.service.IPatrolRecordService;
import org.springblade.modules.place.dto.PlaceCheckDTO;
import org.springblade.modules.place.entity.PlaceCheckEntity;
import org.springblade.modules.place.service.IPlaceService;
import org.springblade.modules.place.vo.PlaceCheckVO;
import org.springblade.modules.place.mapper.PlaceCheckMapper;
import org.springblade.modules.place.service.IPlaceCheckService;
import org.springblade.core.mp.base.BaseServiceImpl;
import org.springblade.modules.task.service.ITaskService;
import org.springblade.modules.taskPlaceRectification.entity.TaskPlaceRectificationEntity;
import org.springblade.modules.taskPlaceRectification.service.ITaskPlaceRectificationService;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import org.springframework.transaction.annotation.Transactional;

import java.util.List;
import java.util.concurrent.atomic.AtomicReference;
import java.util.stream.Collectors;

/**
 * 场所检查表 服务实现类
@@ -34,10 +54,94 @@
@Service
public class PlaceCheckServiceImpl extends ServiceImpl<PlaceCheckMapper, PlaceCheckEntity> implements IPlaceCheckService {

   private static Logger logger = LoggerFactory.getLogger(PlaceCheckServiceImpl.class);

   @Override
   public IPage<PlaceCheckVO> selectPlaceCheckPage(IPage<PlaceCheckVO> page, PlaceCheckVO placeCheck) {
      return page.setRecords(baseMapper.selectPlaceCheckPage(page, placeCheck));
      List<PlaceCheckVO> placeCheckVOS = baseMapper.selectPlaceCheckPage(page, placeCheck);
      for (PlaceCheckVO placeCheckVO : placeCheckVOS) {
         int number = 0;
         for (PatrolRecord patrolRecord : placeCheckVO.getPatrolRecordVOList()) {
            if (patrolRecord.getState().equals(0)) {
               number++;
            }
         }
         placeCheckVO.setNumber(number);
      }
      return page.setRecords(placeCheckVOS);
   }

   @Override
   @Transactional(rollbackFor = Exception.class)
   public Boolean savePlace(PlaceCheckVO placeCheck) throws Exception {
      placeCheck.setCreateUser(AuthUtil.getUserId());
      List<PatrolRecord> patrolRecordVOList = placeCheck.getPatrolRecordVOList();
      AtomicReference<Integer> number = new AtomicReference<>(0);
      patrolRecordVOList.stream().forEach(item -> {
         if (item.getState().equals(0)) {
            number.getAndSet(number.get() + 1);
         }
      });
      placeCheck.setHiddenDangerNumber(number.get());
      boolean save = save(placeCheck);
      if (save) {
         IPatrolRecordService bean = SpringUtil.getBean(IPatrolRecordService.class);
         patrolRecordVOList.stream().forEach(item -> {
            item.setPlaceCheckId(placeCheck.getId());
            item.setCreateUser(AuthUtil.getUserId());
         });
         List<PatrolRecord> collect = patrolRecordVOList.stream().filter(item -> item.getState().equals(0)).collect(Collectors.toList());
         boolean b = bean.saveBatch(collect);
         updateById(placeCheck);
         if (b) {
            // 隐患问题大于0 则创建任务
            try {
               Integer integer = number.get();
               if (integer > 0) {
                  // 保存任务表
                  ITaskService bean2 = SpringUtils.getBean(ITaskService.class);
                  Long aLong = bean2.saveTask(CommonConstant.NUMBER_FOUR, DictConstant.FIRE_RECTIFICATION_NOTICE, 1,
                     "", AuthUtil.getUserId(), placeCheck.getHouseCode(), CommonConstant.NUMBER_EIGHT,4);
                  if (aLong > 0) {
                     // 保存任务详情表
                     ITaskPlaceRectificationService bean1 = SpringUtil.getBean(ITaskPlaceRectificationService.class);
                     TaskPlaceRectificationEntity copy = BeanUtil.copy(placeCheck, TaskPlaceRectificationEntity.class);
                     copy.setTaskId(aLong);
                     copy.setPlaceCheckId(placeCheck.getId());
                     copy.setId(null);
                     copy.setStatus(4);
                     bean1.save(copy);
                  }
               }
            } catch (Exception e) {
               logger.error("任务保存失败！", e);
            }
            return b;
         }
         throw new Exception("保存失败！");
      }
      return false;
   }

   /**
    * 查询场所检查表
    *
    * @param id 场所检查表ID
    * @return 场所检查表
    */
   @Override
   public PlaceCheckVO selectPlaceCheckById(Long id) {
      return this.baseMapper.selectPlaceCheckById(id);
   }

   /**
    * 查询场所检查表列表
    *
    * @param placeCheckDTO 场所检查表
    * @return 场所检查表集合
    */
   @Override
   public List<PlaceCheckDTO> selectPlaceCheckList(PlaceCheckDTO placeCheckDTO) {
      return this.baseMapper.selectPlaceCheckList(placeCheckDTO);
   }
}