wangyifei
2024-10-31 794f620632fca78834677d6b0890bfa0fe78eb02
src/main/java/com/se/simu/controller/WaterController.java
@@ -1,6 +1,5 @@
package com.se.simu.controller;
import com.se.simu.helper.StringHelper;
import com.se.simu.helper.WebHelper;
import com.se.simu.service.WaterService;
import io.swagger.annotations.Api;
@@ -16,7 +15,7 @@
import javax.servlet.http.HttpServletResponse;
/**
 * 内涝控制器
 * 内涝管理
 *
 * @author WWW
 * @date   2024-07-16
@@ -31,15 +30,9 @@
    private final static int MIN_SIZE = 10;
    private final static int MAX_SIZE = 10000000;
    private final static int MAX_SIZE = 80000;
    private final static long Y2000 = 949334400000L;
    @ApiOperation(value = "获取当前时间")
    @GetMapping("/getTime")
    public Object getTime() {
        return System.currentTimeMillis();
    }
    @ApiOperation(value = "获取元数据信息")
    @GetMapping("/{serviceName}/layer.json")
@@ -49,7 +42,9 @@
                return;
            }
            //
            byte[] bytes = waterService.getLayerJson(serviceName);
            WebHelper.writeBytes(bytes, res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
@@ -64,7 +59,9 @@
                return;
            }
            //
            String file = waterService.getTerraMap(serviceName, width, height);
            WebHelper.writePng(file, res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
@@ -79,7 +76,9 @@
                return;
            }
            //
            String file = waterService.getWaterMap(serviceName, width, height, timestamp);
            WebHelper.writePng(file, res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
@@ -94,16 +93,51 @@
                return;
            }
            //
            String file = waterService.getFlowMap(serviceName, width, height, timestamp);
            WebHelper.writePng(file, res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
        }
    }
    @ApiOperation(value = "获取降水曲线文件曲线图")
    @GetMapping("/{serviceName}/rainfall.json")
    public void getRainfall(@PathVariable String serviceName, HttpServletResponse res) {
        try {
            if (!validate(serviceName, res)) {
                return;
            }
            byte[] bytes = waterService.getRainfall(serviceName);
            WebHelper.writeBytes(bytes, res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
        }
    }
    @ApiOperation(value = "根据坐标查询积水深度")
    @GetMapping("/{serviceName}/getWaterHeight")
    public Double getWaterHeight(@PathVariable String serviceName, Double x, Double y, Long timestamp, HttpServletResponse res) {
        try {
            if (!validate(serviceName, res)) {
                return null;
            }
            // 根据服务名+时间戳+坐标,查询对应的积水深度
            return waterService.getWaterHeight(serviceName, x, y, timestamp);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return null;
        }
    }
    private boolean validate(String serviceName, HttpServletResponse res) {
        if (StringHelper.isEmpty(serviceName)) {
            return WebHelper.writeStr2Page(res, HttpStatus.BAD_REQUEST, "服务名不能为空");
        if (WebHelper.isEmpty(serviceName)) {
            return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "服务名不能为空");
        }
        return true;
@@ -113,18 +147,21 @@
        return validate(serviceName, width, height, Y2000, res);
    }
    /**
     * 验证
     */
    private boolean validate(String serviceName, Integer width, Integer height, Long timestamp, HttpServletResponse res) {
        if (StringHelper.isEmpty(serviceName)) {
            return WebHelper.writeStr2Page(res, HttpStatus.BAD_REQUEST, "服务名不能为空");
        if (WebHelper.isEmpty(serviceName)) {
            return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "服务名不能为空");
        }
        if (null == width || width < MIN_SIZE || width > MAX_SIZE) {
            return WebHelper.writeStr2Page(res, HttpStatus.BAD_REQUEST, "图像宽度不能为空且取值范围为" + MIN_SIZE + "~" + MAX_SIZE + "之间");
            return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "图像宽度不能为空且取值范围为" + MIN_SIZE + "~" + MAX_SIZE + "之间");
        }
        if (null == height || height < MIN_SIZE || height > MAX_SIZE) {
            return WebHelper.writeStr2Page(res, HttpStatus.BAD_REQUEST, "图像高度不能为空且取值范围为" + MIN_SIZE + "~" + MAX_SIZE + "之间");
            return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "图像高度不能为空且取值范围为" + MIN_SIZE + "~" + MAX_SIZE + "之间");
        }
        if (null == timestamp || timestamp < Y2000) {
            return WebHelper.writeStr2Page(res, HttpStatus.BAD_REQUEST, "时间不能为空且大于2000年");
        if (null == timestamp || timestamp < 0) {
            return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "时间不能为空且大于0");
        }
        return true;