From b1e8a099eb3117bcbd5f2e0c9a2eb0accabd8cbf Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Tue, 11 Feb 2025 08:56:46 +0800
Subject: [PATCH] 通过风险源点位分析三级防控点查询修改

---
 src/main/java/org/springblade/modules/yw/service/impl/IndParkInfoServiceImpl.java |   72 +++++++++++++++++++++++++++++++++++-
 1 files changed, 70 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/springblade/modules/yw/service/impl/IndParkInfoServiceImpl.java b/src/main/java/org/springblade/modules/yw/service/impl/IndParkInfoServiceImpl.java
index 462b88b..e9e51c9 100644
--- a/src/main/java/org/springblade/modules/yw/service/impl/IndParkInfoServiceImpl.java
+++ b/src/main/java/org/springblade/modules/yw/service/impl/IndParkInfoServiceImpl.java
@@ -6,12 +6,13 @@
 import org.springblade.core.tool.utils.BeanUtil;
 import org.springblade.modules.system.entity.Region;
 import org.springblade.modules.system.service.IRegionService;
+import org.springblade.modules.yw.entity.EmergencySpaceEntity;
 import org.springblade.modules.yw.entity.IndParkInfoEntity;
+import org.springblade.modules.yw.entity.RescueTeamEntity;
 import org.springblade.modules.yw.excel.IndParkInfoExcel;
-import org.springblade.modules.yw.service.IFirmInfoService;
+import org.springblade.modules.yw.service.*;
 import org.springblade.modules.yw.vo.IndParkInfoVO;
 import org.springblade.modules.yw.mapper.IndParkInfoMapper;
-import org.springblade.modules.yw.service.IIndParkInfoService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -32,6 +33,18 @@
 
 	@Autowired
 	private IFirmInfoService firmInfoService;
+
+	@Autowired
+	private IRescueTeamService rescueTeamService;
+
+	@Autowired
+	private IRiskSourceService riskSourceService;
+
+	@Autowired
+	private IProTarService proTarService;
+
+	@Autowired
+	private IEmergencySpaceService emergencySpaceService;
 
 	/**
 	 * 自定义分页查询
@@ -126,11 +139,66 @@
 		indParkInfoVO.setFirmNum((int) count);
 		// 设置行政区划
 		setRegionInfo(indParkInfoVO);
+		// 设置救援人员数量
+		indParkInfoVO.setRecNum(getRecNum());
+		// 设置应急池容量
+		indParkInfoVO.setEmePool(getEmePool());
+		// 设置风险企业数量
+		indParkInfoVO.setRiskFirmNum(getRiskFirmNum());
+		// 设置保护目标数量
+		indParkInfoVO.setProTarNum(getProTarNum());
+		// 设置应急空间数量
+		indParkInfoVO.setSpaceNum(getSpaceNum());
 		// 返回
 		return indParkInfoVO;
 	}
 
 	/**
+	 * 获取应急池容量
+	 * @return
+	 */
+	private String getEmePool() {
+		return emergencySpaceService.getEmePool();
+	}
+
+	/**
+	 * 查询应急空间数量
+	 * @return
+	 */
+	private Integer getSpaceNum() {
+		QueryWrapper<EmergencySpaceEntity> wrapper = new QueryWrapper<>();
+		wrapper.eq("is_deleted",0);
+		return (int)emergencySpaceService.count(wrapper);
+	}
+
+	/**
+	 * 查询保护目标数量
+	 * @return
+	 */
+	private Integer getProTarNum() {
+		return (int)proTarService.count();
+	}
+
+	/**
+	 * 查询风险企业数量
+	 * @return
+	 */
+	private Integer getRiskFirmNum() {
+		return (int)riskSourceService.count();
+	}
+
+	/**
+	 * 获取救援人员数量
+	 * @return
+	 */
+	private Integer getRecNum() {
+		QueryWrapper<RescueTeamEntity> wrapper = new QueryWrapper<>();
+		wrapper.eq("type",1);
+		long count = rescueTeamService.count(wrapper);
+		return (int)count;
+	}
+
+	/**
 	 * 设置行政区划
 	 * @param indParkInfoVO
 	 */

--
Gitblit v1.9.3