From a1272d78844574b86645149a0d8ca257c2b8fe91 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Thu, 26 Dec 2024 09:43:14 +0800
Subject: [PATCH] 风险源统计等级空的设置为其他
---
src/main/java/org/springblade/modules/yw/service/impl/GeomInfoServiceImpl.java | 113 +++++++++++++++++++++++++++++++++++++++++++++++++++-----
1 files changed, 103 insertions(+), 10 deletions(-)
diff --git a/src/main/java/org/springblade/modules/yw/service/impl/GeomInfoServiceImpl.java b/src/main/java/org/springblade/modules/yw/service/impl/GeomInfoServiceImpl.java
index 8b77059..f3a28c9 100644
--- a/src/main/java/org/springblade/modules/yw/service/impl/GeomInfoServiceImpl.java
+++ b/src/main/java/org/springblade/modules/yw/service/impl/GeomInfoServiceImpl.java
@@ -1,27 +1,24 @@
package org.springblade.modules.yw.service.impl;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import org.geotools.geojson.geom.GeometryJSON;
-import org.geotools.geometry.jts.WKBReader;
-import org.locationtech.jts.geom.Geometry;
-import org.locationtech.jts.io.ParseException;
+import org.apache.logging.log4j.util.Strings;
import org.springblade.common.utils.FileUtil;
import org.springblade.common.utils.ShapeFileUtil;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.BeanUtil;
+import org.springblade.modules.yw.entity.FirmInfo;
import org.springblade.modules.yw.entity.GeomInfoEntity;
import org.springblade.modules.yw.mapper.GeomInfoMapper;
+import org.springblade.modules.yw.service.IFirmInfoService;
import org.springblade.modules.yw.service.IGeomInfoService;
import org.springblade.modules.yw.vo.GeomInfoVO;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
-
import java.io.File;
-import java.io.IOException;
-import java.util.List;
-import java.util.Map;
-import java.util.Objects;
+import java.util.*;
/**
* 空间信息表 服务实现类
@@ -31,6 +28,9 @@
*/
@Service
public class GeomInfoServiceImpl extends ServiceImpl<GeomInfoMapper, GeomInfoEntity> implements IGeomInfoService {
+
+ @Autowired
+ private IFirmInfoService firmInfoService;
/**
* 自定义分页
@@ -67,11 +67,104 @@
geomInfoEntity.setGeom("'" + geometry +"'");
// 保存
baseMapper.saveGeoInfo(geomInfoEntity);
+ // 返回
+ return R.data(200,geometry,"操作成功！");
}
} catch (Exception e) {
e.printStackTrace();
}
// 返回
- return R.data(200,"操作成功","操作成功");
+ return R.data(200,"操作失败","操作失败！读取数据错误！");
+ }
+
+ /**
+ * 导出 空间信息shp zip 包
+ * @param ids
+ * @param response
+ * @return
+ */
+ @Override
+ public List<GeomInfoVO> getGeomInfoList(String ids) {
+ // 查询空间数据
+ return baseMapper.getGeomInfoList(ids);
+ }
+
+
+ /**
+ * 导入 shp zip 包文件解析空间信息
+ * @param multipartFile
+ * @return
+ */
+ @Override
+ public R importSpaceShpZip(MultipartFile multipartFile) {
+ // 判断文件是否为 zip 文件
+ if (!FileUtil.getFileExtension(multipartFile).equals("zip")){
+ return R.data(400,"文件格式不对，必须是 zip 文件","文件格式不对，必须是 zip 文件");
+ }
+ // 转 file
+ File file = FileUtil.toFile(multipartFile);
+ // 获取文件信息
+ try {
+ List<Map<String, Object>> list = ShapeFileUtil.shpToGeoJson(file);
+ // 写入数据库,暂时考虑只有一组数据的情况
+ if (list.size()>0){
+ for (Map<String, Object> map : list) {
+ GeomInfoEntity geomInfoEntity = Objects.requireNonNull(BeanUtil.copy(map, GeomInfoEntity.class));
+ String geometry = map.get("geometry").toString();
+ geomInfoEntity.setGeom("'" + geometry +"'");
+ String firmName = null;
+ if (null!=map.get("firmName")){
+ firmName = map.get("firmName").toString();
+ }
+ // 设置所属单位
+ setFirmId(geomInfoEntity,firmName);
+ // 判断是否已经存在
+ Long id = isSave(geomInfoEntity);
+ if (id!=null){
+ // 覆盖更新
+// updateById();
+ continue;
+ }
+ // 保存
+ baseMapper.saveGeoInfo(geomInfoEntity);
+ }
+ // 返回
+ return R.data(200,list,"操作成功！");
+ }
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ // 返回
+ return R.data(200,"操作失败","操作失败！读取数据错误！");
+ }
+
+ /**
+ * 设置所属企业id
+ * @param geomInfoEntity
+ * @param firmName
+ */
+ private void setFirmId(GeomInfoEntity geomInfoEntity,String firmName) {
+ if (!Strings.isBlank(firmName)) {
+ QueryWrapper<FirmInfo> wrapper = new QueryWrapper<>();
+ wrapper.like("name", '%' +firmName + '%');
+ FirmInfo one = firmInfoService.getOne(wrapper);
+ if (null!=one)
+ geomInfoEntity.setFirmId(one.getId());
+ }
+ }
+
+ /**
+ * 判断是否已经保存
+ * @param geomInfoEntity
+ * @return
+ */
+ private Long isSave(GeomInfoEntity geomInfoEntity) {
+ QueryWrapper<GeomInfoEntity> wrapper = new QueryWrapper<>();
+ wrapper.eq("name",geomInfoEntity);
+ GeomInfoEntity one = getOne(wrapper);
+ if (null!=one){
+ return one.getId();
+ }
+ return null;
}
}
--
Gitblit v1.9.3