dcb
2025-06-24 043d1bd4860bbee636b3935e94a71246091cbf3f
src/main/java/com/se/nsl/controller/SimuController.java
@@ -5,11 +5,11 @@
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.SimuResult;
import com.se.nsl.domain.vo.SimuVo;
import com.se.nsl.domain.vo.*;
import com.se.nsl.service.RealTimeSimulationService;
import com.se.nsl.service.ResolveService;
import com.se.nsl.service.SimuService;
import com.se.nsl.utils.SimulateType;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
@@ -32,6 +32,9 @@
    @Resource
    ResolveService resolveService;
    @Resource
    RealTimeSimulationService rts;
    /**
     * 分页查询推演模拟
@@ -97,10 +100,13 @@
            int rows = simuService.insert(simu);
            System.out.println(String.format("id:%s", simu.getId()));
            if (rows > 0 && (null == data.getRainfalls() || data.getRainfalls().size() < 2)) {
                resolveService.createRainfall(simu);
                simuService.updateById(simu);
            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());
@@ -126,9 +132,16 @@
            //if (simu.getStatus() != 0) return fail("方案正在运行或已完成");
            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) {
                String s = rts.realTimeSimulate(simu);
                return success(s);
            }
            return fail("模拟类型暂不支持");
        } catch (Exception ex) {
            return fail(ex, null);
        }
@@ -171,16 +184,21 @@
        if (lat > 90 || lat < -90) {
            return clientError("纬度范围应该在-90到90");
        }
        if (time == null) {
            return clientError("时间戳不能为空");
        }
        if (serviceName == null || serviceName.trim().isEmpty()) {
            return clientError("服务名不能为空");
        }
        SimuResult result = simuService.queryByPosition(lon, lat, time, serviceName);
        if (result == null) {
            return notFound("未查找到相关数据");
        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());
        }
        return success(result);
    }
}