From acb0e0de3ba95f9f38996d049bc73231e563d3a1 Mon Sep 17 00:00:00 2001 From: 张洋洋 <10611411+yang-yang-z@user.noreply.gitee.com> Date: 星期四, 20 二月 2025 15:33:10 +0800 Subject: [PATCH] [add]服务发布 --- src/main/java/com/se/simu/controller/SimuController.java | 152 +++++++++++++++++++++++++++++++++----------------- 1 files changed, 99 insertions(+), 53 deletions(-) diff --git a/src/main/java/com/se/simu/controller/SimuController.java b/src/main/java/com/se/simu/controller/SimuController.java index 9374580..c9b8452 100644 --- a/src/main/java/com/se/simu/controller/SimuController.java +++ b/src/main/java/com/se/simu/controller/SimuController.java @@ -11,9 +11,11 @@ import com.se.simu.domain.po.DataPo; import com.se.simu.domain.po.SimuPo; import com.se.simu.domain.vo.*; +import com.se.simu.enums.RadioEnums; import com.se.simu.service.*; import com.se.simu.utils.*; import io.swagger.annotations.*; +import lombok.SneakyThrows; import lombok.extern.slf4j.Slf4j; import org.apache.commons.codec.binary.Base64; import org.apache.commons.lang3.StringUtils; @@ -24,11 +26,16 @@ import javax.annotation.Resource; import java.io.*; +import java.math.BigDecimal; import java.net.URL; import java.net.URLEncoder; import java.nio.charset.StandardCharsets; import java.text.SimpleDateFormat; +import java.time.LocalDateTime; +import java.time.format.DateTimeFormatter; import java.util.*; +import java.util.concurrent.ExecutorService; +import java.util.concurrent.Executors; import java.util.stream.Collectors; @Api(tags = "浠跨湡绠$悊") @@ -231,6 +238,9 @@ if (StringUtils.isEmpty(floodFile)) { return fail("鑼冨洿鏂囦欢鍦板潃涓嶈兘涓虹┖", false); } else { + // todo: 瑙f瀽鑼冨洿鏂囦欢 鏍规嵁涓嶅悓鐨勭殑鏍煎紡鏂囦欢杩涜涓嶅悓鐨勮В鏋愶紝鑾峰彇闇�瑕佽绠楃殑鑼冨洿鍊� + // 涓婁紶鏍煎紡锛�.shp/.tiff/.img/.geojson + // 閲嶆柊缁檝o璧嬪�� // 鑾峰彇鏂囦欢鐨勫悗缂�鍚� String fileName = floodFile.substring(floodFile.lastIndexOf(".")); // 鍒ゆ柇鍚庣紑鍚嶆槸鍚︿负.shp @@ -252,54 +262,79 @@ rangeVerify(minX, maxX, minY, maxY); } } - // todo: 瑙f瀽鑼冨洿鏂囦欢 鏍规嵁涓嶅悓鐨勭殑鏍煎紡鏂囦欢杩涜涓嶅悓鐨勮В鏋愶紝鑾峰彇闇�瑕佽绠楃殑鑼冨洿鍊� - // 涓婁紶鏍煎紡锛�.shp/.tiff/.img/.geojson - // 閲嶆柊缁檝o璧嬪�� } catch (Exception e) { log.error("瑙f瀽鑼冨洿鏂囦欢澶辫触"); if (null == vo.getMinx() || null == vo.getMiny() || null == vo.getMaxx() || null == vo.getMaxy()) { return fail("瑙f瀽鑼冨洿鏂囦欢澶辫触锛侀�夋嫨鑼冨洿涓嶈兘涓虹┖锛岃閲嶆柊閫夋嫨鏂囦欢锛�", false); } } - JSONArray jsonArray = new JSONArray(); String stationFile = vo.getStationFile(); log.info("绔欑偣鏂囦欢shp鍦板潃锛歿}", stationFile); try { - // 鍒ゆ柇鍦板潃涓嶄负绌� - if (StringUtils.isEmpty(stationFile)) { - return fail("绔欑偣鏂囦欢shp鍦板潃涓嶈兘涓虹┖", false); + if (RadioEnums.ACCESS.getCode().equals(vo.getRadio()) && StringUtils.isNotEmpty(stationFile)) { + // 1 璇诲彇shp鏂囦欢锛岃幏鍙栫珯鐐瑰潗鏍囧�� + ShpToolUtils.readShpGetLocal(stationFile); } - // 1 璇诲彇shp鏂囦欢锛岃幏鍙栫珯鐐瑰潗鏍囧�� - jsonArray = ShpToolUtils.readShpGetLocal(stationFile); - System.out.println("jsonArray = " + jsonArray); } catch (Exception e) { log.error("瑙f瀽绔欑偣鏂囦欢shp澶辫触"); } - - String stationRainFile = vo.getStationRainFile(); - log.info("绔欑偣闆ㄩ噺CSV鏂囦欢鍦板潃锛歿}", floodFile); - JSONArray array = new JSONArray(); // 鍒涘缓琛ㄥ悕 鏃堕棿鎴� String tableName = "station_rain_" + System.currentTimeMillis(); try { - // 鍒ゆ柇鍦板潃涓嶄负绌� - if (StringUtils.isEmpty(stationRainFile)) { - return fail("绔欑偣闆ㄩ噺CSV鏂囦欢鍦板潃涓嶈兘涓虹┖", false); + //鎺ュ叆澶勭悊 + if (RadioEnums.ACCESS.getCode().equals(vo.getRadio())) { + String stationRainFile = vo.getStationRainFile(); + log.info("绔欑偣闆ㄩ噺CSV鏂囦欢鍦板潃锛歿}", floodFile); + // 鍒ゆ柇鍦板潃涓嶄负绌� + if (StringUtils.isEmpty(stationRainFile)) { + return fail("绔欑偣闆ㄩ噺CSV鏂囦欢鍦板潃涓嶈兘涓虹┖", false); + } + // 1 璇诲彇CSV 鏂囦欢 + CsvToSQLiteUtils.readCsvSaveLocal(stationRainFile, tableName); + // 鑾峰彇浠跨湡鏃堕棿 duration + Integer duration = CsvToSQLiteUtils.getDuration(tableName); + log.info("浠跨湡鏃堕棿 duration = {}", duration); + vo.setDuration(duration); + // 鑾峰彇闄嶉洦鎬婚噺 total + Double total = CsvToSQLiteUtils.getTotal(tableName); + log.info("闄嶉洦鎬婚噺 total = {}", total); + vo.setTotal(total); + } else if (RadioEnums.SIMULATE.getCode().equals(vo.getRadio())) { + if (StringUtils.isEmpty(vo.getStation())) { + return fail("妯℃嫙璁$畻绔欑偣涓嶈兘涓虹┖锛�", false); + } + if (vo.getTotal() == null) { + return fail("妯℃嫙璁$畻闄嶉洦閲忎笉鑳戒负绌猴紒", false); + } + if (vo.getDuration() == null || vo.getDuration() <= 0) { + return fail("妯℃嫙璁$畻璁$畻鏃堕暱涓嶈兘涓虹┖锛�", false); + } + if (vo.getStationLatitude() == null) { + return fail("妯℃嫙璁$畻璁$畻鍧愭爣锛�", false); + } + if (vo.getStationLongitude() == null) { + return fail("妯℃嫙璁$畻璁$畻鍧愭爣锛�", false); + } + double aveTotal = BigDecimal.valueOf(vo.getTotal()).divide(BigDecimal.valueOf(vo.getDuration()), 2, BigDecimal.ROUND_HALF_UP).doubleValue(); + JSONArray jsonArray = new JSONArray(); + LocalDateTime now = LocalDateTime.now(); + DateTimeFormatter customFormatter = DateTimeFormatter.ofPattern("yyyy/MM/dd HH:mm"); + for (int i = 0; i < vo.getDuration(); i++) { + JSONObject jsonObject = new JSONObject(); + jsonObject.put("stationName", vo.getStation()); + jsonObject.put("rainfall", aveTotal); + jsonObject.put("longitude", vo.getStationLongitude()); + jsonObject.put("latitude", vo.getStationLatitude()); + LocalDateTime plusMinutes = now.plusMinutes(i); + jsonObject.put("datetime", customFormatter.format(plusMinutes)); + jsonArray.add(jsonObject); + } + CsvToSQLiteUtils.readCsvSaveLocal(jsonArray, tableName); + } else { + return fail("鏃犳晥鐨勬ā鍨嬭绠楁柟寮�", false); } - // 1 璇诲彇CSV 鏂囦欢 - array = CsvToSQLiteUtils.readCsvSaveLocal(stationRainFile, tableName); - // 鑾峰彇浠跨湡鏃堕棿 duration - Integer duration = CsvToSQLiteUtils.getDuration(tableName); - log.info("浠跨湡鏃堕棿 duration = {}", duration); - vo.setDuration(duration); - - // 鑾峰彇闄嶉洦鎬婚噺 total - Double total = CsvToSQLiteUtils.getTotal(tableName); - log.info("闄嶉洦鎬婚噺 total = {}", total); - vo.setTotal(total); - } catch (Exception e) { - log.error("瑙f瀽绔欑偣闆ㄩ噺CSV鏂囦欢澶辫触"); + return fail("闆ㄩ噺澶勭悊澶辫触锛�", false); } // TODO: 2024/12/24 鏍规嵁杩欎簺鏂囦欢鐨勫湴鍧�锛岃幏鍙栨枃浠跺唴瀹癸紝鍒涘缓浠跨湡瑙嗗浘 try { @@ -345,9 +380,24 @@ dtos.add(new GridDto(vo.getMaxy(), vo.getMinx())); dtos.add(new GridDto(vo.getMaxy(), vo.getMaxx())); dtos.add(new GridDto(vo.getMiny(), vo.getMaxx())); - createSem(tableName, dtos, vo); + String bbox = ""; + //鐢熸垚鐐瑰拰绾跨殑瀹炰綋搴撴ā鍨嬪苟淇濆瓨璺緞 + String name = vo.getSemUrl(); + String token = EntityLibraryUtils.login(); + vo.setPointName(name + "-point"); + vo.setLinkName(name + "-line"); + JSONArray min = ProjectionToGeographicUtil.get4548Point(dtos.get(0).getLon(), dtos.get(0).getLat()); + JSONArray max = ProjectionToGeographicUtil.get4548Point(dtos.get(2).getLon(), dtos.get(2).getLat()); + bbox = bbox + min.getDouble(0) + "," + min.getDouble(1) + "," + max.getDouble(0) + "," + max.getDouble(1); + JSONObject point = getModule("layerQueryPointParams.json"); + String pointUrl = assemble(vo.getPointName(), point.getString("layerid"), bbox, token); + vo.setPointName(pointUrl); + JSONObject link = getModule("layerQueryParams.json"); + String linkUrl = assemble(vo.getLinkName(), link.getString("layerid"), bbox, token); + vo.setLinkName(linkUrl); + asyncCall(tableName, dtos, vo, token); // 寮�濮嬫ā鎷熻绠� - boolean flag = simuFilesService.createByfiles(vo); + boolean flag = simuFilesService.createByfiles(vo,token); return success(flag, flag ? "鎴愬姛" : "澶辫触"); } catch (Exception ex) { return fail(ex, null); @@ -356,28 +406,39 @@ @ApiOperation(value = "鑾峰彇鐐硅矾寰�") @GetMapping("/getPointUrl") - public R<String> getPointUrl(@RequestParam("id") Integer id) throws Exception{ + public R<String> getPointUrl(@RequestParam("id") Integer id) throws Exception { SimuPo po = simuService.getSimuById(id); String token = EntityLibraryUtils.login(); if (po != null) { - return success(po.getPointUrl().replace("{token}",token)); + return success(po.getPointUrl().replace("{token}", token)); } return success(null); } @ApiOperation(value = "鑾峰彇绾胯矾寰�") @GetMapping("/getLineUrl") - public R<String> getLineUrl(@RequestParam("id") Integer id) throws Exception{ + public R<String> getLineUrl(@RequestParam("id") Integer id) throws Exception { SimuPo po = simuService.getSimuById(id); String token = EntityLibraryUtils.login(); if (po != null) { - return success(po.getLinkUrl().replace("{token}",token)); + return success(po.getLinkUrl().replace("{token}", token)); } return success(null); } - public R<Boolean> createSem(String tableName, List<GridDto> dtos, CreateFilesSimuVo vo) throws Exception { - String name = vo.getSemUrl(); + private void asyncCall(String tableName, List<GridDto> dtos, CreateFilesSimuVo vo, String token) { + ExecutorService executor = Executors.newSingleThreadExecutor(); + executor.execute(new Runnable() { + @Override + @SneakyThrows + public void run() { + createSem(tableName, dtos, vo, token); + } + }); + executor.shutdown(); + } + + public void createSem(String tableName, List<GridDto> dtos, CreateFilesSimuVo vo, String token) throws Exception { vo.setSemUrl(inPath + "\\" + vo.getSemUrl()); boolean ends = vo.getSemUrl().endsWith("\\"); if (!ends) { @@ -387,7 +448,6 @@ if (!file.exists()) { file.mkdirs(); } - String token = EntityLibraryUtils.login(); //闄嶉洦鏂囦欢鐢熸垚 saveZarr(tableName, vo.getSemUrl()); //绠$偣鐢熸垚 @@ -402,20 +462,6 @@ landuseToSem(vo.getSemUrl()); //鍦板舰鐢熸垚 terrainToSem(dtos, vo.getSemUrl()); - String bbox = ""; - //鐢熸垚鐐瑰拰绾跨殑瀹炰綋搴撴ā鍨嬪苟淇濆瓨璺緞 - vo.setPointName(name + "-point"); - vo.setLinkName(name + "-line"); - JSONArray min = ProjectionToGeographicUtil.get4548Point(dtos.get(0).getLon(), dtos.get(0).getLat()); - JSONArray max = ProjectionToGeographicUtil.get4548Point(dtos.get(2).getLon(), dtos.get(2).getLat()); - bbox = bbox + min.getDouble(0) + "," + min.getDouble(1) + "," + max.getDouble(0) + "," + max.getDouble(1); - JSONObject point = getModule("layerQueryPointParams.json"); - String pointUrl = assemble(vo.getPointName(), point.getString("layerid"), bbox, token); - vo.setPointName(pointUrl); - JSONObject link = getModule("layerQueryParams.json"); - String linkUrl = assemble(vo.getLinkName(), link.getString("layerid"), bbox, token); - vo.setLinkName(linkUrl); - return success(true); } -- Gitblit v1.9.3