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 ++++++++++++++++++++++++++++----
 src/main/java/com/se/simu/config/Knife4jConfig.java       |    2 +-
 src/main/java/com/se/simu/service/WaterService.java       |   12 ++++++++++++
 pom.xml                                                   |    2 +-
 4 files changed, 42 insertions(+), 6 deletions(-)

diff --git a/pom.xml b/pom.xml
index 50b5602..70e64f7 100644
--- a/pom.xml
+++ b/pom.xml
@@ -140,7 +140,7 @@
     </profiles>
 
     <build>
-        <finalName>simu</finalName>
+        <finalName>SimuServer</finalName>
         <plugins>
             <plugin>
                 <groupId>org.springframework.boot</groupId>
diff --git a/src/main/java/com/se/simu/config/Knife4jConfig.java b/src/main/java/com/se/simu/config/Knife4jConfig.java
index a1468a4..10fc70b 100644
--- a/src/main/java/com/se/simu/config/Knife4jConfig.java
+++ b/src/main/java/com/se/simu/config/Knife4jConfig.java
@@ -63,7 +63,7 @@
                 .description("鎺ュ彛鏂囨。")
                 .contact(new Contact("WuWeiwei", "http://127.0.0.1:" + serverPort + contextPath + "/doc.html", "252740454@qq.com"))
                 .version("1.0.0")
-                .title("鍐呮稘浠跨湡娓叉煋鏈嶅姟API鎺ュ彛鏂囨。")
+                .title("鍐呮稘浠跨湡鏈嶅姟API鎺ュ彛鏂囨。")
                 .build();
     }
 }
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;
+    }
 }
diff --git a/src/main/java/com/se/simu/service/WaterService.java b/src/main/java/com/se/simu/service/WaterService.java
index 50a81ce..04e408c 100644
--- a/src/main/java/com/se/simu/service/WaterService.java
+++ b/src/main/java/com/se/simu/service/WaterService.java
@@ -39,4 +39,16 @@
 
         return layer;
     }
+
+    public String getTerraMap(String serviceName, Integer width, Integer height) {
+        return "D:/simu/test/geo.png";
+    }
+
+    public String getWaterMap(String serviceName, Integer width, Integer height) {
+        return null;
+    }
+
+    public String getFlowMap(String serviceName, Integer width, Integer height, Long timestamp) {
+        return null;
+    }
 }

--
Gitblit v1.9.3