From a9a2d0f1d8d271985869dd2b26ac29217191ab51 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Tue, 10 Dec 2024 11:51:58 +0800
Subject: [PATCH] 应急空间新增字典，查询修改
---
src/main/java/org/springblade/modules/yw/service/impl/EmergencySpaceServiceImpl.java | 127 ++++++++++++++++++++++++++++++++++++++++++
1 files changed, 126 insertions(+), 1 deletions(-)
diff --git a/src/main/java/org/springblade/modules/yw/service/impl/EmergencySpaceServiceImpl.java b/src/main/java/org/springblade/modules/yw/service/impl/EmergencySpaceServiceImpl.java
index e4e79c1..2b5a360 100644
--- a/src/main/java/org/springblade/modules/yw/service/impl/EmergencySpaceServiceImpl.java
+++ b/src/main/java/org/springblade/modules/yw/service/impl/EmergencySpaceServiceImpl.java
@@ -1,12 +1,25 @@
package org.springblade.modules.yw.service.impl;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.apache.logging.log4j.util.Strings;
+import org.springblade.common.cache.DictBizCache;
+import org.springblade.common.cache.SysCache;
+import org.springblade.common.utils.PositionUtil;
+import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.modules.yw.entity.EmergencySpaceEntity;
+import org.springblade.modules.yw.entity.FirmInfo;
+import org.springblade.modules.yw.excel.EmergencySpaceExcel;
+import org.springblade.modules.yw.service.IFirmInfoService;
import org.springblade.modules.yw.vo.EmergencySpaceVO;
import org.springblade.modules.yw.mapper.EmergencySpaceMapper;
import org.springblade.modules.yw.service.IEmergencySpaceService;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
/**
* 应急空间表 服务实现类
@@ -17,9 +30,121 @@
@Service
public class EmergencySpaceServiceImpl extends ServiceImpl<EmergencySpaceMapper, EmergencySpaceEntity> implements IEmergencySpaceService {
+ @Autowired
+ private IFirmInfoService firmInfoService;
+
+ /**
+ * 自定义分页查询
+ * @param page
+ * @param emergencySpace
+ * @return
+ */
@Override
public IPage<EmergencySpaceVO> selectEmergencySpacePage(IPage<EmergencySpaceVO> page, EmergencySpaceVO emergencySpace) {
- return page.setRecords(baseMapper.selectEmergencySpacePage(page, emergencySpace));
+ List<EmergencySpaceVO> emergencySpaceVOS = baseMapper.selectEmergencySpacePage(page, emergencySpace);
+ for (EmergencySpaceVO emergencySpaceVO : emergencySpaceVOS) {
+ // 设置应急空间主要用途
+ emergencySpaceVO.setMainFuncName(DictBizCache.getValues("emergency_space_use",emergencySpaceVO.getMainFunc()));
+ }
+ return page.setRecords(emergencySpaceVOS);
}
+ /**
+ * 导入应急空间信息
+ * @param data
+ * @param isCovered
+ * @return
+ */
+ @Override
+ public String importEmergencySpace(List<EmergencySpaceExcel> data, boolean isCovered) {
+ for (EmergencySpaceExcel emergencySpaceExcel : data) {
+ // 数据拷贝
+ EmergencySpaceEntity emergencySpaceEntity = Objects.requireNonNull(BeanUtil.copy(emergencySpaceExcel, EmergencySpaceEntity.class));
+ // 类型转换
+ if (!Strings.isBlank(emergencySpaceExcel.getType())){
+ emergencySpaceEntity.setType(Integer.parseInt(emergencySpaceExcel.getType()));
+ }
+ // 防控级别转换
+ if (!Strings.isBlank(emergencySpaceExcel.getPreLevel())){
+ emergencySpaceEntity.setPreLevel(Integer.parseInt(emergencySpaceExcel.getPreLevel()));
+ }
+ // 经纬度转换
+// if (!Strings.isBlank(emergencySpaceExcel.getLng())) {
+// emergencySpaceEntity.setLng(PositionUtil.tranformPos(emergencySpaceExcel.getLng()).toString());
+// }
+// if (!Strings.isBlank(emergencySpaceExcel.getLat())) {
+// emergencySpaceEntity.setLat(PositionUtil.tranformPos(emergencySpaceExcel.getLat()).toString());
+// }
+ // 设置企业id
+ setFirm(emergencySpaceExcel,emergencySpaceEntity);
+ // 是否已入库
+ Long id = isSave(emergencySpaceEntity);
+ if (null!=id){
+ if (isCovered){
+ // 覆盖更新
+ emergencySpaceEntity.setId(id);
+ updateById(emergencySpaceEntity);
+ continue;
+ }
+ }
+ // 保存
+ save(emergencySpaceEntity);
+ }
+ return null;
+ }
+
+ /**
+ * 判断是否保存入库
+ * @param emergencySpaceEntity
+ * @return
+ */
+ private Long isSave(EmergencySpaceEntity emergencySpaceEntity) {
+ QueryWrapper<EmergencySpaceEntity> wrapper = new QueryWrapper<>();
+ wrapper.eq("name",emergencySpaceEntity.getName())
+ .eq("is_deleted",0);
+ if (null!=emergencySpaceEntity.getFirmId()){
+ wrapper.eq("firm_id",emergencySpaceEntity.getFirmId());
+ }
+ EmergencySpaceEntity one = getOne(wrapper);
+ if (null!=one){
+ return one.getId();
+ }
+ return null;
+ }
+
+ /**
+ * 设置企业信息
+ * @param emergencySpaceExcel
+ * @param emergencySpaceEntity
+ */
+ private void setFirm(EmergencySpaceExcel emergencySpaceExcel, EmergencySpaceEntity emergencySpaceEntity) {
+ QueryWrapper<FirmInfo> wrapper = new QueryWrapper<>();
+ wrapper.eq("name",emergencySpaceExcel.getFirmName())
+ .eq("is_deleted",0);
+ FirmInfo firmInfo = firmInfoService.getOne(wrapper);
+ if (null!=firmInfo){
+ emergencySpaceEntity.setFirmId(firmInfo.getId());
+ }
+ }
+
+ /**
+ * 获取应急空间统计数据
+ * @param emergencySpace
+ * @return
+ */
+ @Override
+ public List<Map<String,Object>> getStatisticData(EmergencySpaceVO emergencySpace) {
+ // 按应急空间类型进行统计
+ List<Map<String,Object>> list = baseMapper.getStatisticData(emergencySpace);
+ return list;
+ }
+
+ /**
+ * 获取应急池容量
+ * @return
+ */
+ @Override
+ public String getEmePool() {
+ return baseMapper.getEmePool();
+ }
}
--
Gitblit v1.9.3