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 |   32 ++++++++++++++++++++++++++++----
 1 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/se/simu/controller/WaterController.java b/src/main/java/com/se/simu/controller/WaterController.java
index 1e50750..32a55a3 100644
--- a/src/main/java/com/se/simu/controller/WaterController.java
+++ b/src/main/java/com/se/simu/controller/WaterController.java
@@ -15,6 +15,7 @@
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletResponse;
+import java.io.File;
 
 /**
  * 鍐呮稘鎺у埗鍣�
@@ -32,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;
 
@@ -67,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());
@@ -82,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());
@@ -97,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());
@@ -116,6 +120,9 @@
         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.writeJson2Page(res, HttpStatus.BAD_REQUEST, "鏈嶅姟鍚嶄笉鑳戒负绌�");
@@ -132,4 +139,21 @@
 
         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