添加参数验证

13693261870

2024-07-16 577701a313e21448467558b0a507bb7196415674

 src/main/java/com/se/simu/controller/WaterController.java

@@ -1,15 +1,19 @@
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;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;

/**
 * 内涝控制器
@@ -25,6 +29,12 @@
    @Resource
    WaterService waterService;

    private final static int MIN_SIZE = 10;

    private final static int MAX_SIZE = 10000000;

    private final static long Y2000 = 949334400000L;

    @ApiOperation(value = "获取当前时间")
    @GetMapping("/getTime")
    public Object getTime() {
@@ -33,25 +43,90 @@

    @ApiOperation(value = "获取元数据信息")
    @GetMapping("/{serviceName}/layer.json")
    public Object getLayer(@PathVariable String serviceName) {
        return serviceName;
    public void getLayer(@PathVariable String serviceName, HttpServletResponse res) {
        try {
            if (!validate(serviceName, res)) {
                return;
            }

            //
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
        }
    }

    @ApiOperation(value = "获取地形高度图")
    @GetMapping("/{serviceName}/terrain?width={width}&height={height}")
    public Object getTerraMap(@PathVariable String serviceName, @PathVariable Integer width, @PathVariable Integer height) {
        return serviceName + "," + width + "," + height;
    @GetMapping("/{serviceName}/terrain")
    public void getTerraMap(@PathVariable String serviceName, Integer width, Integer height, HttpServletResponse res) {
        try {
            if (!validate(serviceName, width, height, res)) {
                return;
            }

            //
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
        }
    }

    @ApiOperation(value = "获取水面高度图")
    @GetMapping("/{serviceName}/waterMap?width={width}&height={height}&time={timestamp}")
    public Object getWaterMap(@PathVariable String serviceName, @PathVariable Integer width, @PathVariable Integer height, @PathVariable Long timestamp) {
        return serviceName + "," + width + "," + height + "," + timestamp;
    @GetMapping("/{serviceName}/waterMap")
    public void getWaterMap(@PathVariable String serviceName, Integer width, Integer height, Long timestamp, HttpServletResponse res) {
        try {
            if (!validate(serviceName, width, height, timestamp, res)) {
                return;
            }

            //
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
        }
    }

    @ApiOperation(value = "获取水流向流速图")
    @GetMapping("/{serviceName}/flowMap?width={width}&height={height}&time={timestamp}")
    public Object getFlowMap(@PathVariable String serviceName, @PathVariable Integer width, @PathVariable Integer height, @PathVariable Long timestamp) {
        return serviceName + "," + width + "," + height + "," + timestamp;
    @GetMapping("/{serviceName}/flowMap")
    public void getFlowMap(@PathVariable String serviceName, Integer width, Integer height, Long timestamp, HttpServletResponse res) {
        try {
            if (!validate(serviceName, width, height, timestamp, res)) {
                return;
            }

            //
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
        }
    }

    private boolean validate(String serviceName, HttpServletResponse res) {
        if (StringHelper.isEmpty(serviceName)) {
            return WebHelper.writeStr2Page(res, HttpStatus.BAD_REQUEST, "服务名不能为空");
        }

        return true;
    }

    private boolean validate(String serviceName, Integer width, Integer height, HttpServletResponse res) {
        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 (null == width || width < MIN_SIZE || width > MAX_SIZE) {
            return WebHelper.writeStr2Page(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 + "之间");
        }
        if (null == timestamp || timestamp < Y2000) {
            return WebHelper.writeStr2Page(res, HttpStatus.BAD_REQUEST, "时间不能为空且大于2000年");
        }

        return true;
    }
}