From 893998916eab354d4002fbcdcb0a242a36a49684 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期四, 18 七月 2024 11:28:14 +0800 Subject: [PATCH] 3 --- src/main/java/com/se/simu/controller/WaterController.java | 34 +++++++++------------------------- 1 files changed, 9 insertions(+), 25 deletions(-) diff --git a/src/main/java/com/se/simu/controller/WaterController.java b/src/main/java/com/se/simu/controller/WaterController.java index 21959c2..ae3a6c2 100644 --- a/src/main/java/com/se/simu/controller/WaterController.java +++ b/src/main/java/com/se/simu/controller/WaterController.java @@ -1,6 +1,5 @@ package com.se.simu.controller; -import com.se.simu.domain.Layer; import com.se.simu.helper.WebHelper; import com.se.simu.service.WaterService; import lombok.extern.slf4j.Slf4j; @@ -12,7 +11,6 @@ import javax.annotation.Resource; import javax.servlet.http.HttpServletResponse; -import java.io.File; /** * 鍐呮稘鎺у埗鍣� @@ -51,9 +49,9 @@ return; } - Layer 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 +69,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 +87,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 +107,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()); @@ -142,23 +143,6 @@ if (null == timestamp || timestamp < Y2000) { return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "鏃堕棿涓嶈兘涓虹┖涓斿ぇ浜�2000骞�"); } - - 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; } -- Gitblit v1.9.3