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/controller/GeomInfoController.java |   61 ++++++++++++++++++++++++++++++
 1 files changed, 61 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/yw/controller/GeomInfoController.java b/src/main/java/org/springblade/modules/yw/controller/GeomInfoController.java
index 0fa7761..3f509d6 100644
--- a/src/main/java/org/springblade/modules/yw/controller/GeomInfoController.java
+++ b/src/main/java/org/springblade/modules/yw/controller/GeomInfoController.java
@@ -6,6 +6,7 @@
 import io.swagger.annotations.ApiOperation;
 import io.swagger.annotations.ApiParam;
 import lombok.AllArgsConstructor;
+import org.apache.logging.log4j.util.Strings;
 import org.springblade.common.constant.FileConstant;
 import org.springblade.common.utils.ShapeFileUtil;
 import org.springblade.common.utils.ZipUtil;
@@ -14,7 +15,10 @@
 import org.springblade.core.tool.api.R;
 import org.springblade.core.tool.utils.Func;
 import org.springblade.modules.yw.entity.GeomInfoEntity;
+import org.springblade.modules.yw.service.IEmergencySpaceService;
+import org.springblade.modules.yw.service.IFirmInfoService;
 import org.springblade.modules.yw.service.IGeomInfoService;
+import org.springblade.modules.yw.vo.EmergencySpaceVO;
 import org.springblade.modules.yw.vo.GeomInfoVO;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
@@ -45,6 +49,8 @@
 	private final IGeomInfoService geomInfoService;
 
 	private final FileConstant fileConstant;
+
+	private final IEmergencySpaceService emergencySpaceService;
 
 	/**
 	 * 空间信息表 详情
@@ -131,6 +137,18 @@
 	}
 
 	/**
+	 * 导入 应急空间点位 shp zip 包文件解析空间信息
+	 * @param file
+	 * @return
+	 */
+	@PostMapping("/importSpaceShpZip")
+	@ApiOperationSupport(order = 8)
+	@ApiOperation(value = "导入应急空间点位 shp zip 包文件解析空间信息", notes = "传入文件")
+	public R importSpaceShpZip(MultipartFile file) {
+		return geomInfoService.importSpaceShpZip(file);
+	}
+
+	/**
 	 * 导出 空间信息shp zip 包
 	 * @param ids
 	 * @return
@@ -168,6 +186,49 @@
 
 	/**
 	 * 根据传入的点线面数据-导出空间信息shp zip 包
+	 * @param response
+	 * @return
+	 */
+	@GetMapping("/exporEmergencySpaceShpZip")
+	@ApiOperationSupport(order = 9)
+	@ApiOperation(value = "根据传入的点线面数据-导出空间信息shp zip 包", notes = "传入空间数据")
+	public void exporEmergencySpaceShpZip(HttpServletResponse response) throws IOException {
+		Query query = new Query().setCurrent(1).setSize(100);
+		IPage<EmergencySpaceVO> emergencySpaceVOIPage
+			= emergencySpaceService.selectEmergencySpacePage(Condition.getPage(query), null);
+		// 查询应急空间数据
+		List<EmergencySpaceVO> emergencySpaceVOList = emergencySpaceVOIPage.getRecords();
+		List<Map<String,Object>> propertyList = new ArrayList<>();
+		// 要导出的文件夹路径
+		String sourceFolderPath = fileConstant.getZipTempSavePath();
+		for (EmergencySpaceVO emergencySpaceVO : emergencySpaceVOList) {
+			// 先生成文件
+			Map<String,Object> map = new HashMap<>();
+			map.put("ID", emergencySpaceVO.getId());
+			map.put("name", emergencySpaceVO.getName());
+			if (!Strings.isBlank(emergencySpaceVO.getLng())){
+				map.put("geometry","POINT("+emergencySpaceVO.getLng() +" " + emergencySpaceVO.getLat() + ")");
+			}
+			map.put("firmId",emergencySpaceVO.getFirmId());
+			map.put("firmName",emergencySpaceVO.getFirmName());
+			propertyList.add(map);
+		}
+		// 导出shp 文件
+		ShapeFileUtil.exportShp(propertyList, "yikj", "Point",sourceFolderPath);
+		// ZIP文件名
+		String zipFileName = System.currentTimeMillis() + "_exported_shp.zip";
+		response.setContentType("application/zip");
+		response.setHeader("Content-Disposition", "attachment; filename=\"" + zipFileName + "\"");
+		try (
+			BufferedOutputStream bos = new BufferedOutputStream(response.getOutputStream());
+			ZipOutputStream zipOut = new ZipOutputStream(bos);
+		) {
+			ZipUtil.zipFolder(sourceFolderPath, zipOut);
+		}
+	}
+
+	/**
+	 * 根据传入的点线面数据-导出空间信息shp zip 包
 	 * @param geomInfoEntity
 	 * @return
 	 */

--
Gitblit v1.9.3