From 22f80ba79735ab9897530b79e34889d2082cedbc Mon Sep 17 00:00:00 2001
From: wuww <252740454@qq.com>
Date: 星期四, 08 五月 2025 16:25:40 +0800
Subject: [PATCH] 开发根据参数生成降雨文件接口

---
 src/main/java/com/se/nsl/controller/SimuController.java |   55 +++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 53 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/se/nsl/controller/SimuController.java b/src/main/java/com/se/nsl/controller/SimuController.java
index ff0abec..0028a67 100644
--- a/src/main/java/com/se/nsl/controller/SimuController.java
+++ b/src/main/java/com/se/nsl/controller/SimuController.java
@@ -6,10 +6,13 @@
 import com.se.nsl.domain.po.SimuData;
 import com.se.nsl.domain.vo.R;
 import com.se.nsl.domain.vo.SimuVo;
+import com.se.nsl.service.ResolveService;
 import com.se.nsl.service.SimuService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
+import org.gdal.ogr.Geometry;
+import org.gdal.ogr.ogr;
 import org.springframework.util.StringUtils;
 import org.springframework.web.bind.annotation.*;
 
@@ -24,6 +27,9 @@
 public class SimuController extends BaseController {
     @Resource
     SimuService simuService;
+
+    @Resource
+    ResolveService resolveService;
 
     /**
      * 鍒嗛〉鏌ヨ鎺ㄦ紨妯℃嫙
@@ -77,14 +83,47 @@
     @PostMapping(value = "/insert", produces = "application/json; charset=UTF-8")
     public R<Object> insert(@RequestBody Simu simu) {
         try {
-            if (StringUtils.isEmpty(simu.getData())) return fail("data is null");
+            if (StringUtils.isEmpty(simu.getData())) return fail("data涓虹┖");
 
             SimuData data = JSON.parseObject(simu.getData(), SimuData.class);
-            if (null == data) return fail("data is null");
+            if (null == data) return fail("data鏁版嵁鏍煎紡(JSON)涓嶆纭�");
+            if (StringUtils.isEmpty(simu.getGeom())) return fail("geom瀛楃涓蹭笉鏄疻KT鏍煎紡");
+
+            Geometry geom = Geometry.CreateFromWkt(simu.getGeom());
+            if (!(geom.GetGeometryType() == ogr.wkbMultiPolygon || geom.GetGeometryType() == ogr.wkbPolygon))
+                return fail("geom瀵硅薄涓嶆槸澶氳竟褰�");
 
             int rows = simuService.insert(simu);
+            if (rows > 0 && (null == data.getRainfalls() || data.getRainfalls().size() < 2)) {
+                resolveService.createRainfall(simu);
+                simuService.updateById(simu);
+            }
 
             return success(rows);
+        } catch (Exception ex) {
+            return fail(ex, null);
+        }
+    }
+
+    @ApiOperation(value = "start")
+    @GetMapping(value = "/start", produces = "application/json; charset=UTF-8")
+    public R<Object> start(Integer id) {
+        try {
+            if (null == id || id < 1) return fail("id涓虹┖");
+
+            Simu simu = simuService.selectById(id);
+            if (null == simu) return fail("鏂规鎵句笉鍒�");
+            if (StringUtils.isEmpty(simu.getData())) return fail("鏂规鏁版嵁(JSON)涓虹┖");
+
+            SimuData data = JSON.parseObject(simu.getData(), SimuData.class);
+            if (null == data) return fail("鏂规鏁版嵁鏍煎紡(JSON)涓嶆纭�");
+
+            //if (simu.getStatus() != 0) return fail("鏂规姝e湪杩愯鎴栧凡瀹屾垚");
+            if (StringUtils.isEmpty(simu.getGeom())) return fail("鏂规鐨勫浘褰负绌�");
+
+            int rows = resolveService.start(simu);
+
+            return success("ok");
         } catch (Exception ex) {
             return fail(ex, null);
         }
@@ -100,6 +139,18 @@
     @PutMapping(value = "/updateById", produces = "application/json; charset=UTF-8")
     public R<Object> updateById(@RequestBody Simu simu) {
         try {
+            if (StringUtils.isEmpty(simu.getData())) return fail("data涓虹┖");
+
+            SimuData data = JSON.parseObject(simu.getData(), SimuData.class);
+            if (null == data) return fail("data鏁版嵁鏍煎紡(JSON)涓嶆纭�");
+            if (StringUtils.isEmpty(simu.getGeom())) return fail("geom瀛楃涓蹭笉鏄疻KT鏍煎紡");
+
+            Geometry geom = Geometry.CreateFromWkt(simu.getGeom());
+            if (!(geom.GetGeometryType() == ogr.wkbMultiPolygon || geom.GetGeometryType() == ogr.wkbPolygon))
+                return fail("geom瀵硅薄涓嶆槸澶氳竟褰�");
+
+            if (null == data.getRainfalls() || data.getRainfalls().size() < 2) resolveService.createRainfall(simu);
+
             return success(simuService.updateById(simu));
         } catch (Exception ex) {
             return fail(ex, null);

--
Gitblit v1.9.3