From d0e4c868677d0443308a9cc3542a8549822240fe Mon Sep 17 00:00:00 2001
From: zrj <646384940@qq.com>
Date: Tue, 29 Oct 2024 14:54:36 +0800
Subject: [PATCH] 新增应急空间信息导入
---
src/main/java/org/springblade/modules/yw/service/impl/EmergencySpaceServiceImpl.java | 89 ++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 89 insertions(+), 0 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..2f9a643 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,22 @@
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.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.Objects;
/**
* 应急空间表 服务实现类
@@ -17,9 +27,88 @@
@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));
}
+ /**
+ * 导入应急空间信息
+ * @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.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);
+ 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());
+ }
+ }
}
--
Gitblit v1.9.3