添加基础类

13693261870

2024-09-12 b5326f8d497a6f6e97a487cb9c565fdae1dc4790

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

@@ -1,7 +1,7 @@
package com.se.simu.controller;

import com.se.simu.domain.LayerVo;
import com.se.simu.helper.WebHelper;
import com.se.simu.service.SedbService;
import com.se.simu.service.WaterService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
@@ -12,7 +12,6 @@

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.File;

/**
 * 内涝控制器
@@ -24,6 +23,9 @@
@RestController
@RequestMapping("/waterlogging")
public class WaterController {
    @Resource
    SedbService sedbService;

    @Resource
    WaterService waterService;

@@ -51,9 +53,9 @@
                return;
            }

            LayerVo layer = waterService.getLayer(serviceName);
            byte[] bytes = waterService.getLayerJson(serviceName);

            WebHelper.writeJson2Page(res, HttpStatus.OK, layer);
            WebHelper.writeBytes(bytes, res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
@@ -71,7 +73,8 @@
            }

            String file = waterService.getTerraMap(serviceName, width, height);
            writeFile(file, res);

            WebHelper.writePng(file, res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
@@ -88,8 +91,9 @@
                return;
            }

            String file = waterService.getWaterMap(serviceName, width, height);
            writeFile(file, res);
            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());
@@ -107,7 +111,8 @@
            }

            String file = waterService.getFlowMap(serviceName, width, height, timestamp);
            writeFile(file, res);

            WebHelper.writePng(file, res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage());
@@ -139,27 +144,15 @@
        if (null == height || height < MIN_SIZE || height > MAX_SIZE) {
            return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "图像高度不能为空且取值范围为" + MIN_SIZE + "~" + MAX_SIZE + "之间");
        }
        if (null == timestamp || timestamp < Y2000) {
            return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "时间不能为空且大于2000年");
        if (null == timestamp || timestamp < 0) {
            return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "时间不能为空且大于0");
        }

        return true;
    }

    /**
     * 写文件
     */
    private boolean writeFile(String path, HttpServletResponse res) throws Exception {
        if (WebHelper.isEmpty(path)) {
            return WebHelper.writeJson2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, "生成PNG失败");
        }
        File file = new File(path);
        if (!file.exists() || file.isDirectory()) {
            return WebHelper.writeJson2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, "PNG文件找不到");
        }

        WebHelper.download(path, file.getName(), res);

        return true;
    @GetMapping("/test")
    public Object test() {
        return sedbService.getToken();
    }
}