断面分析功能实现

dcb

2025-07-09 e53252b99e7b49b435b7a6ee3eab21ae1bd7a055

 src/main/java/com/se/nsl/controller/SimuController.java

@@ -1,19 +1,26 @@
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.vo.R;
import com.se.nsl.domain.vo.SimuVo;
import com.se.nsl.service.SimuService;
import com.se.nsl.domain.po.SimuData;
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 io.swagger.v3.oas.annotations.Parameter;
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.*;

import javax.annotation.Resource;
import java.util.List;

@Api(tags = "2－推演模拟")
@Api(tags = "03－推演模拟")
@Slf4j
@RestController
@RequestMapping("/simu")
@@ -21,6 +28,15 @@
public class SimuController extends BaseController {
    @Resource
    SimuService simuService;

    @Resource
    ResolveService resolveService;

    @Resource
    RealTimeSimulationAsyncService rtsas;

    @Resource
    CrossSectionAnalysisService scas;

    /**
     * 分页查询推演模拟
@@ -71,10 +87,63 @@
     * @return 新增成功的记录数
     */
    @ApiOperation(value = "insert")
    @PostMapping("/insert")
    public R<Object> insert(Simu simu) {
    @PostMapping(value = "/insert", produces = "application/json; charset=UTF-8")
    public R<Object> insert(@RequestBody Simu simu) {
        try {
            return success(simuService.insert(simu));
            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字符串不是WKT格式");

            Geometry geom = Geometry.CreateFromWkt(simu.getGeom());
            if (!(geom.GetGeometryType() == ogr.wkbMultiPolygon || geom.GetGeometryType() == ogr.wkbPolygon))
                return fail("geom对象不是多边形");

            int rows = simuService.insert(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);
                }
            }
            JSONObject json = new JSONObject();
            json.put("id", simu.getId());
            return success(json);
        } 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 notFound("方案找不到");
            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("方案正在运行或已完成");
            if (StringUtils.isEmpty(simu.getGeom())) return fail("方案的图形为空");

            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);
        }
@@ -90,9 +159,116 @@
    @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字符串不是WKT格式");

            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);
        }
    }

    @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(@Parameter(description = "方案id，示例：50") Integer id,
                                  @Parameter(description = "时间戳，示例：1751552400000") Long time,
                                  @Parameter(description = "起点坐标，示例：116.59049537485063,40.564178548127686") String startPoint,
                                  @Parameter(description = "终点坐标，示例：116.5901406492509,40.56499045715429") String endPoint) {
        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("找不到对应的服务");
        }
        if (startPoint == null) {
            return clientError("起点不能为空");
        }
        if (endPoint == null) {
            return clientError("终点不能为空");
        }
        String[] sp = startPoint.split(",");
        double[] s = new double[] {Double.parseDouble(sp[0]), Double.parseDouble(sp[1])};
        String[] ep = endPoint.split(",");
        double[] e = new double[] {Double.parseDouble(ep[0]), Double.parseDouble(ep[1])};
        try {
            if (time == null) {
                List<CrossSectionAnalysisResult> result = scas.crossSectionAnalysis(serviceName, s, e);
                return success(result, result.size());
            } else {
                CrossSectionAnalysisResult res = scas.crossSectionAnalysis(serviceName, s, e, time);
                return success(res, 1);
            }
        } catch (IllegalArgumentException ex) {
            return fail(ex.getMessage(), null);
        }
    }

    @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, "正在停止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());
    }
}