From 428851fa44382ddd420d88dc79ee944162e5f93a Mon Sep 17 00:00:00 2001
From: xingjinshuang <xingjs@qq.com>
Date: 星期四, 26 十二月 2024 17:18:52 +0800
Subject: [PATCH] @xingjs@20241226@添加查询管线工具类,修改解决查询站点接口token存在无法获取的情况,解决后可正常获取token,并查询出需要的站点;解析shp文件获取站点范围;新增把降水CSV文件内容解析保存到数据库中;计算CSV文件中的降水时间和降水总量

---
 src/main/java/com/se/simu/controller/SimuController.java |  176 ++++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 162 insertions(+), 14 deletions(-)

diff --git a/src/main/java/com/se/simu/controller/SimuController.java b/src/main/java/com/se/simu/controller/SimuController.java
index 274004c..50ff53a 100644
--- a/src/main/java/com/se/simu/controller/SimuController.java
+++ b/src/main/java/com/se/simu/controller/SimuController.java
@@ -1,34 +1,33 @@
 package com.se.simu.controller;
 
-import cn.hutool.json.JSONArray;
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.se.simu.config.PropertiesConfig;
 import com.se.simu.domain.dto.GeDb;
 import com.se.simu.domain.dto.GeLayer;
 import com.se.simu.domain.po.DataPo;
 import com.se.simu.domain.po.SimuPo;
+import com.se.simu.domain.vo.CreateFilesSimuVo;
 import com.se.simu.domain.vo.CreateSimuVo;
 import com.se.simu.domain.vo.R;
 import com.se.simu.domain.vo.SimuVo;
-import com.se.simu.service.GedbService;
-import com.se.simu.service.ResultService;
-import com.se.simu.service.SimuService;
-import com.se.simu.service.UwService;
+import com.se.simu.service.*;
+import com.se.simu.utils.CsvToSQLiteUtils;
+import com.se.simu.utils.ShpToolUtils;
 import io.swagger.annotations.*;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
+import java.io.IOException;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Date;
 import java.util.List;
 
-/**
- * 浠跨湡绠$悊
- *
- * @author WWW
- * @date   2024-09-18
- */
 @Api(tags = "浠跨湡绠$悊")
 @Slf4j
 @RestController
@@ -43,6 +42,16 @@
 
     @Resource
     SimuService simuService;
+
+    @Resource
+    SimuFilesService simuFilesService;
+
+    @Resource
+    PropertiesConfig config;
+
+    @Value("${simu-app.filePath}")
+    private String uploadedFolder;
+
 
     @Resource
     ResultService resultService;
@@ -148,9 +157,6 @@
         }
     }
 
-    /**
-     * 浣跨敤绌洪棿鑼冨洿鏌ヨ 绠$偣,绠$嚎,寤虹瓚鐗� 3涓浘灞傦紝涓旀瘡涓浘灞傜殑鏁伴噺澶т簬0锛岃繑鍥� true銆�
-     */
     @ApiOperation(value = "闄嶆按鑼冨洿鏍¢獙")
     @GetMapping("/rangeVerify")
     public R<Object> rangeVerify(@RequestParam @ApiParam("鏈�灏廥") double minx,
@@ -174,4 +180,146 @@
             return fail(ex, false);
         }
     }
+
+    /**
+     * 鏂囦欢鍒涘缓妯℃嫙
+     * <p>
+     * "鐘舵�侊細
+     * 0-鍒涘缓浠跨湡浠诲姟锛�
+     * 1-杩炴帴GEDB搴擄紝
+     * 2-涓嬭浇绌洪棿鏁版嵁锛�
+     * 3-涓嬭浇楂樼▼鏁版嵁锛�
+     * 4-鐢熸垚闄嶉洦鏂囦欢锛�
+     * 5-鐢熸垚閰嶇疆鏂囦欢锛�
+     * 6-妯℃嫙鍐呮稘浠跨湡锛�
+     * 7-澶勭悊姘翠綅鏂囦欢锛�
+     * 8-澶勭悊鎺掓按鏂囦欢锛�
+     * 9-澶勭悊浠跨湡缁撴灉锛�
+     * 10-瀹屾垚锛�-10-鍑洪敊
+     *
+     * @param vo VO
+     * @return {@link R}<{@link Object}>
+     */
+    @ApiOperation(value = "鏂囦欢鍒涘缓")
+    @PostMapping(value = "/file_create", produces = "application/json; charset=UTF-8")
+    public R<Object> fileCreate(@RequestBody @ApiParam("鍒涘缓浠跨湡瑙嗗浘绫�") CreateFilesSimuVo vo) throws IOException {
+        // 鑾峰彇涓婁紶鏂囦欢璺緞
+        String targetDir = uploadedFolder;
+        log.info("涓婁紶鏂囦欢璺緞锛歿}", targetDir);
+        // 鑾峰彇鍚勪釜鏂囦欢鐨勫湴鍧�
+        String floodFile = vo.getFloodFile();
+        log.info("鑼冨洿鏂囦欢鍦板潃锛歿}", floodFile);
+        try {
+            // 鍒ゆ柇鍦板潃涓嶄负绌�
+            if (StringUtils.isEmpty(floodFile)) {
+                return fail("鑼冨洿鏂囦欢鍦板潃涓嶈兘涓虹┖", false);
+            } else {
+                // 鑾峰彇鏂囦欢鐨勫悗缂�鍚�
+                String fileName = floodFile.substring(floodFile.lastIndexOf("."));
+                // 鍒ゆ柇鍚庣紑鍚嶆槸鍚︿负.shp
+                if (!fileName.equalsIgnoreCase(".shp")) {
+                    return fail("鑼冨洿鏂囦欢鏍煎紡涓嶆纭�", false);
+                } else {
+                    // 1 璇诲彇shp鏂囦欢锛岃幏鍙栬寖鍥村��
+                    JSONObject jsonObject = ShpToolUtils.readShp(floodFile);
+                    // 2 鑾峰彇jsonObject涓殑鑼冨洿鍊�
+                    vo.setMinx(jsonObject.getDouble("minX"));
+                    vo.setMaxx(jsonObject.getDouble("maxX"));
+                    vo.setMiny(jsonObject.getDouble("minY"));
+                    vo.setMaxy(jsonObject.getDouble("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);
+            }
+        }
+
+        String stationFile = vo.getStationFile();
+        log.info("绔欑偣鏂囦欢shp鍦板潃锛歿}", stationFile);
+        try {
+            // 鍒ゆ柇鍦板潃涓嶄负绌�
+            if (StringUtils.isEmpty(stationFile)) {
+                return fail("绔欑偣鏂囦欢shp鍦板潃涓嶈兘涓虹┖", false);
+            }
+            // 1 璇诲彇shp鏂囦欢锛岃幏鍙栫珯鐐瑰潗鏍囧��
+            JSONArray jsonArray = ShpToolUtils.readShpGetLocal(stationFile);
+            System.out.println("jsonArray = " + jsonArray);
+        } catch (Exception e) {
+            log.error("瑙f瀽绔欑偣鏂囦欢shp澶辫触");
+        }
+
+        String stationRainFile = vo.getStationRainFile();
+        log.info("绔欑偣闆ㄩ噺CSV鏂囦欢鍦板潃锛歿}", floodFile);
+        try {
+            // 鍒ゆ柇鍦板潃涓嶄负绌�
+            if (StringUtils.isEmpty(stationRainFile)) {
+                return fail("绔欑偣闆ㄩ噺CSV鏂囦欢鍦板潃涓嶈兘涓虹┖", false);
+            }
+            // 鍒涘缓琛ㄥ悕 鏃堕棿鎴�
+            String tableName = "station_rain_" + System.currentTimeMillis();
+            // 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);
+
+        } catch (Exception e) {
+            log.error("瑙f瀽绔欑偣闆ㄩ噺CSV鏂囦欢澶辫触");
+        }
+        // TODO: 2024/12/24 鏍规嵁杩欎簺鏂囦欢鐨勫湴鍧�锛岃幏鍙栨枃浠跺唴瀹癸紝鍒涘缓浠跨湡瑙嗗浘
+        try {
+            if (null == vo.getDuration() || vo.getDuration() < 1 || vo.getDuration() > 10080) {
+                return fail("浠跨湡鏃堕暱涓嶈兘涓虹┖锛屼笖鍙栧�煎湪1~10080涔嬮棿");
+            }
+            if (null == vo.getPid() || vo.getPid() < 0) {
+                vo.setPid(0);
+            }
+            if (null == vo.getNum() || vo.getNum() < 1) {
+                vo.setNum(simuService.getMaxId() + 1);
+            }
+            if (null == vo.getStartTime()) {
+                vo.setStartTime(new Date());
+            }
+            // 闃叉睕浣滀笟
+            if (vo.getPid() > 0) {
+                SimuPo pp = simuService.getSimuByPid(vo.getPid());
+                if (null == pp) {
+                    return fail("pid涓嶅瓨鍦�");
+                }
+                if (null == vo.getFloodStart() || vo.getFloodStart() < 1 || vo.getFloodStart() > vo.getDuration() * 60) {
+                    return fail("闃叉睕寮�濮嬫椂闂翠笉鑳戒负绌猴紝涓斿彇鍊煎湪1~" + (vo.getDuration() * 60) + "涔嬮棿");
+                }
+                if (null == vo.getFloodEnd() || vo.getFloodEnd() < vo.getFloodStart() || vo.getFloodEnd() > vo.getDuration() * 60) {
+                    return fail("闃叉睕缁撴潫鏃堕棿涓嶈兘涓虹┖锛屼笖鍙栧�煎湪" + vo.getFloodStart() + "~" + (vo.getDuration() * 60) + "涔嬮棿");
+                }
+                if (null == vo.getFloodHeight() || vo.getFloodHeight() < 1 || vo.getFloodHeight() > 2000) {
+                    return fail("闃叉睕楂樺害涓嶈兘涓虹┖锛屼笖鍙栧�煎湪1~2000涔嬮棿");
+                }
+                if (!FLOOD_TYPE.contains(vo.getFloodType())) {
+                    return fail("闃叉睕绫诲瀷涓嶈兘涓虹┖锛屼笖鍙兘鏄矙琚嬪拰闃叉按鏉�");
+                }
+                if (null == vo.getFloodMinx() || null == vo.getFloodMiny() || null == vo.getFloodMaxx() || null == vo.getFloodMaxy()) {
+                    return fail("闃叉睕鑼冨洿涓嶈兘涓虹┖", false);
+                }
+            }
+            // 寮�濮嬫ā鎷熻绠�
+            boolean flag = simuFilesService.createByfiles(vo);
+            return success(flag, flag ? "鎴愬姛" : "澶辫触");
+        } catch (Exception ex) {
+            return fail(ex, null);
+        }
+    }
+
+
 }

--
Gitblit v1.9.3