From b7df6735bcb2d447cae7fcfba2d12accf3dd2fcc Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Sat, 03 Feb 2024 17:27:21 +0800
Subject: [PATCH] 消防检查优化

---
 src/main/java/org/springblade/modules/place/service/impl/PlaceCheckServiceImpl.java |  108 +++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 106 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/springblade/modules/place/service/impl/PlaceCheckServiceImpl.java b/src/main/java/org/springblade/modules/place/service/impl/PlaceCheckServiceImpl.java
index c33fdf7..625a132 100644
--- a/src/main/java/org/springblade/modules/place/service/impl/PlaceCheckServiceImpl.java
+++ b/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);
+	}
 }

--
Gitblit v1.9.3