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.*;
|
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 = "03－推演模拟")
|
@Slf4j
|
@RestController
|
@RequestMapping("/simu")
|
@SuppressWarnings("ALL")
|
public class SimuController extends BaseController {
|
@Resource
|
SimuService simuService;
|
|
@Resource
|
ResolveService resolveService;
|
|
@Resource
|
RealTimeSimulationAsyncService rtsas;
|
|
@Resource
|
CrossSectionAnalysisService scas;
|
|
/**
|
* 分页查询推演模拟
|
*
|
* @param pageNum 页码
|
* @param pageSize 每页数量
|
* @return 分页后的推演模拟
|
*/
|
@ApiOperation(value = "selectPage")
|
@GetMapping("/selectPage")
|
public R<Object> selectPage(SimuVo vo, Integer pageNum, Integer pageSize) {
|
try {
|
if (null == pageNum || pageNum < 1) pageNum = 1;
|
if (null == pageSize || pageSize < 1) pageSize = 10;
|
if (pageSize > 1000) pageSize = 1000;
|
|
IPage<Simu> paged = simuService.selectPage(vo, pageNum, pageSize);
|
if (null == paged) {
|
return success(null, 0);
|
}
|
|
return success(paged.getRecords(), paged.getTotal());
|
} catch (Exception ex) {
|
return fail(ex, null);
|
}
|
}
|
|
/**
|
* 根据ID批量删除推演模拟
|
*
|
* @param ids 要删除的推演模拟ID列表
|
* @return 删除成功的记录数
|
*/
|
@ApiOperation(value = "deleteByIds")
|
@DeleteMapping("/deleteByIds")
|
public R<Object> deleteByIds(@RequestParam List<Integer> ids) {
|
try {
|
return success(simuService.deleteByIds(ids));
|
} catch (Exception ex) {
|
return fail(ex, null);
|
}
|
}
|
|
/**
|
* 新增推演模拟
|
*
|
* @param simu 推演模拟对象
|
* @return 新增成功的记录数
|
*/
|
@ApiOperation(value = "insert")
|
@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为空");
|
|
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);
|
}
|
}
|
|
/**
|
* 修改推演模拟
|
*
|
* @param simu 推演模拟对象
|
* @return 修改成功的记录数
|
*/
|
@ApiOperation(value = "updateById")
|
@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());
|
}
|
}
|
