linwei
2024-01-31 b3b566ebdfed4005aaa513da3d5d2fd3924903cc
src/main/java/org/springblade/modules/place/service/impl/PlaceCheckServiceImpl.java
@@ -17,6 +17,11 @@
package org.springblade.modules.place.service.impl;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.core.secure.utils.AuthUtil;
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.vo.PlaceCheckVO;
import org.springblade.modules.place.mapper.PlaceCheckMapper;
@@ -24,6 +29,10 @@
import org.springblade.core.mp.base.BaseServiceImpl;
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.stream.Collectors;
/**
 * 场所检查表 服务实现类
@@ -36,8 +45,59 @@
   @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());
      boolean save = save(placeCheck);
      if (save) {
         IPatrolRecordService bean = SpringUtil.getBean(IPatrolRecordService.class);
         List<PatrolRecord> patrolRecordVOList = placeCheck.getPatrolRecordVOList();
         patrolRecordVOList.stream().forEach(item -> {
            item.setPlaceCheckId(placeCheck.getId());
            item.setCreateUser(AuthUtil.getUserId());
         });
         boolean b = bean.saveBatch(placeCheck.getPatrolRecordVOList());
         if (b) {
            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);
   }
}