From f31f0991c0d2036e563b886f57de4cf45d3c72cb Mon Sep 17 00:00:00 2001 From: dcb <xgybdcb@163.com> Date: 星期二, 01 七月 2025 14:51:59 +0800 Subject: [PATCH] 实时模拟异步功能实现 --- src/main/java/com/se/nsl/controller/SimuController.java | 124 +++++++++++++++++++++++++++++++++++++---- 1 files changed, 111 insertions(+), 13 deletions(-) diff --git a/src/main/java/com/se/nsl/controller/SimuController.java b/src/main/java/com/se/nsl/controller/SimuController.java index 0028a67..f289da9 100644 --- a/src/main/java/com/se/nsl/controller/SimuController.java +++ b/src/main/java/com/se/nsl/controller/SimuController.java @@ -1,13 +1,13 @@ package com.se.nsl.controller; import com.alibaba.fastjson.JSON; +import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.metadata.IPage; import com.se.nsl.domain.po.Simu; 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 com.se.nsl.domain.vo.*; +import com.se.nsl.service.*; +import com.se.nsl.utils.SimulateType; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import lombok.extern.slf4j.Slf4j; @@ -30,6 +30,12 @@ @Resource ResolveService resolveService; + + @Resource + RealTimeSimulationAsyncService rtsas; + + @Resource + CrossSectionAnalysisService scas; /** * 鍒嗛〉鏌ヨ鎺ㄦ紨妯℃嫙 @@ -94,12 +100,18 @@ return fail("geom瀵硅薄涓嶆槸澶氳竟褰�"); int rows = simuService.insert(simu); - if (rows > 0 && (null == data.getRainfalls() || data.getRainfalls().size() < 2)) { - resolveService.createRainfall(simu); - simuService.updateById(simu); + System.out.println(String.format("id:%s", simu.getId())); + Integer type = data.getType(); + SimulateType simulateType = SimulateType.of(type); + if (simulateType == SimulateType.HISTORY) { + if (rows > 0 && (null == data.getRainfalls() || data.getRainfalls().size() < 2)) { + resolveService.createRainfall(simu); + simuService.updateById(simu); + } } - - return success(rows); + JSONObject json = new JSONObject(); + json.put("id", simu.getId()); + return success(json); } catch (Exception ex) { return fail(ex, null); } @@ -112,7 +124,7 @@ if (null == id || id < 1) return fail("id涓虹┖"); Simu simu = simuService.selectById(id); - if (null == simu) return fail("鏂规鎵句笉鍒�"); + if (null == simu) return notFound("鏂规鎵句笉鍒�"); if (StringUtils.isEmpty(simu.getData())) return fail("鏂规鏁版嵁(JSON)涓虹┖"); SimuData data = JSON.parseObject(simu.getData(), SimuData.class); @@ -121,9 +133,16 @@ //if (simu.getStatus() != 0) return fail("鏂规姝e湪杩愯鎴栧凡瀹屾垚"); if (StringUtils.isEmpty(simu.getGeom())) return fail("鏂规鐨勫浘褰负绌�"); - int rows = resolveService.start(simu); - - return success("ok"); + Short type = simu.getType(); + SimulateType simulateType = SimulateType.of(type); + if (simulateType == SimulateType.HISTORY) { + int rows = resolveService.start(simu); + return success("ok"); + } else if (simulateType == SimulateType.REAL_TIME) { + rtsas.startSimulation(simu); + return success(null, "瀹炴椂妯℃嫙浠诲姟宸叉彁浜�"); + } + return fail("妯℃嫙绫诲瀷鏆備笉鏀寔"); } catch (Exception ex) { return fail(ex, null); } @@ -156,4 +175,83 @@ return fail(ex, null); } } + + @ApiOperation(value = "position") + @GetMapping("/position") + public R<Object> queryByPosition(double lon, double lat, Long time, String serviceName) { + if (lon > 180 || lon < -180) { + return clientError("缁忓害鑼冨洿搴旇鍦�-180鍒�180"); + } + if (lat > 90 || lat < -90) { + return clientError("绾害鑼冨洿搴旇鍦�-90鍒�90"); + } + if (serviceName == null || serviceName.trim().isEmpty()) { + return clientError("鏈嶅姟鍚嶄笉鑳戒负绌�"); + } + if (time != null) { + SimuResult result = simuService.queryByPosition(lon, lat, time, serviceName); + if (result == null) { + return notFound("鏈煡鎵惧埌鐩稿叧鏁版嵁"); + } + return success(result, 1); + } else { + List<SimuResult> simuResults = simuService.queryByPosition(lon, lat, serviceName); + if (simuResults.isEmpty()) { + return notFound("鏈煡鎵惧埌鐩稿叧鏁版嵁"); + } + return success(simuResults, simuResults.size()); + } + } + + @ApiOperation(value = "crossSection") + @GetMapping("/crossSection") + public R<Object> crossSection(String serviceName, double[] startPoint, double[] endPoint) { + if (serviceName == null) { + return clientError("鏈嶅姟鍚嶄笉鑳戒负绌�"); + } + if (startPoint == null) { + return clientError("璧风偣涓嶈兘涓虹┖"); + } + if (startPoint.length < 2) { + return clientError("璧风偣鑷冲皯鍖呭惈x,y涓や釜鍊�"); + } + if (endPoint == null) { + return clientError("缁堢偣涓嶈兘涓虹┖"); + } + if (endPoint.length < 2) { + return clientError("缁堢偣鑷冲皯鍖呭惈x,y涓や釜鍊�"); + } + List<CrossSectionAnalysisResult> result = scas.crossSectionAnalysis(serviceName, startPoint, endPoint); + return success(result); + } + + @ApiOperation(value = "stop") + @GetMapping("/stop") + public R<Object> stop(Integer id) { + if (id == null) { + return fail("id涓嶈兘涓虹┖"); + } + try { + rtsas.stopSimulation(id); + } catch (IllegalArgumentException e) { + return notFound(e.getMessage()); + } + return success(null, "姝e湪鍋滄id涓�" + id + "鐨勬ā鎷熶换鍔�"); + } + + @ApiOperation(value = "results") + @GetMapping("/results") + public R<Object> querySimulationResult(Integer id) { + if (null == id || id < 1) return clientError("id涓嶈兘涓虹┖"); + Simu simu = simuService.selectById(id); + if (simu == null) { + return clientError("鎵句笉鍒板搴旂殑鏈嶅姟"); + } + String serviceName = simu.getServiceName(); + if (serviceName == null) { + return fail("鎵句笉鍒板搴旂殑鏈嶅姟"); + } + List<String> results = resolveService.simulationResults(serviceName); + return success(results, results.size()); + } } -- Gitblit v1.9.3