From 94493e6798ce4be955d1ff1f4d918463423ba9b5 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期五, 13 九月 2024 17:05:49 +0800 Subject: [PATCH] 1 --- src/main/java/com/se/simu/controller/WaterController.java | 45 ++++++++++++++------------------------------- 1 files changed, 14 insertions(+), 31 deletions(-) diff --git a/src/main/java/com/se/simu/controller/WaterController.java b/src/main/java/com/se/simu/controller/WaterController.java index 32a55a3..f391f72 100644 --- a/src/main/java/com/se/simu/controller/WaterController.java +++ b/src/main/java/com/se/simu/controller/WaterController.java @@ -1,7 +1,5 @@ package com.se.simu.controller; -import com.se.simu.domain.LayerVo; -import com.se.simu.helper.StringHelper; import com.se.simu.helper.WebHelper; import com.se.simu.service.WaterService; import io.swagger.annotations.Api; @@ -15,10 +13,9 @@ import javax.annotation.Resource; import javax.servlet.http.HttpServletResponse; -import java.io.File; /** - * 鍐呮稘鎺у埗鍣� + * 鍐呮稘绠$悊 * * @author WWW * @date 2024-07-16 @@ -51,9 +48,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()); @@ -69,7 +66,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()); @@ -84,8 +82,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()); @@ -101,7 +100,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()); @@ -109,7 +109,7 @@ } private boolean validate(String serviceName, HttpServletResponse res) { - if (StringHelper.isEmpty(serviceName)) { + if (WebHelper.isEmpty(serviceName)) { return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "鏈嶅姟鍚嶄笉鑳戒负绌�"); } @@ -124,7 +124,7 @@ * 楠岃瘉 */ private boolean validate(String serviceName, Integer width, Integer height, Long timestamp, HttpServletResponse res) { - if (StringHelper.isEmpty(serviceName)) { + if (WebHelper.isEmpty(serviceName)) { return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "鏈嶅姟鍚嶄笉鑳戒负绌�"); } if (null == width || width < MIN_SIZE || width > MAX_SIZE) { @@ -133,26 +133,9 @@ 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 (StringHelper.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; } -- Gitblit v1.9.3