From a6c82c251a6a1871bcf9a7e9890455f40d84e0a3 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期二, 16 七月 2024 15:55:48 +0800 Subject: [PATCH] 输出png图片 --- src/main/java/com/se/simu/controller/WaterController.java | 47 +++++++++++++++++++++++++++++++++++++---------- 1 files changed, 37 insertions(+), 10 deletions(-) diff --git a/src/main/java/com/se/simu/controller/WaterController.java b/src/main/java/com/se/simu/controller/WaterController.java index cd9e4bc..32a55a3 100644 --- a/src/main/java/com/se/simu/controller/WaterController.java +++ b/src/main/java/com/se/simu/controller/WaterController.java @@ -1,5 +1,6 @@ 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; @@ -14,6 +15,7 @@ import javax.annotation.Resource; import javax.servlet.http.HttpServletResponse; +import java.io.File; /** * 鍐呮稘鎺у埗鍣� @@ -31,7 +33,7 @@ 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; @@ -49,7 +51,9 @@ return; } - // + LayerVo layer = waterService.getLayer(serviceName); + + WebHelper.writeJson2Page(res, HttpStatus.OK, layer); } catch (Exception ex) { log.error(ex.getMessage(), ex); WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage()); @@ -64,7 +68,8 @@ return; } - // + String file = waterService.getTerraMap(serviceName, width, height); + writeFile(file, res); } catch (Exception ex) { log.error(ex.getMessage(), ex); WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage()); @@ -79,7 +84,8 @@ return; } - // + String file = waterService.getWaterMap(serviceName, width, height); + writeFile(file, res); } catch (Exception ex) { log.error(ex.getMessage(), ex); WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage()); @@ -94,7 +100,8 @@ return; } - // + String file = waterService.getFlowMap(serviceName, width, height, timestamp); + writeFile(file, res); } catch (Exception ex) { log.error(ex.getMessage(), ex); WebHelper.writeStr2Page(res, HttpStatus.INTERNAL_SERVER_ERROR, ex.getMessage()); @@ -103,7 +110,7 @@ private boolean validate(String serviceName, HttpServletResponse res) { if (StringHelper.isEmpty(serviceName)) { - return WebHelper.writeStr2Page(res, HttpStatus.BAD_REQUEST, "鏈嶅姟鍚嶄笉鑳戒负绌�"); + return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "鏈嶅姟鍚嶄笉鑳戒负绌�"); } return true; @@ -113,20 +120,40 @@ 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, "鏈嶅姟鍚嶄笉鑳戒负绌�"); + 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骞�"); + return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "鏃堕棿涓嶈兘涓虹┖涓斿ぇ浜�2000骞�"); } 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