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 | 124 +++++++++++++++++++++++++++++++++++++---
1 files changed, 113 insertions(+), 11 deletions(-)
diff --git a/src/main/java/com/se/simu/controller/WaterController.java b/src/main/java/com/se/simu/controller/WaterController.java
index 50b6b1f..32a55a3 100644
--- a/src/main/java/com/se/simu/controller/WaterController.java
+++ b/src/main/java/com/se/simu/controller/WaterController.java
@@ -1,15 +1,21 @@
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;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
+import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+import java.io.File;
/**
* 鍐呮稘鎺у埗鍣�
@@ -25,6 +31,12 @@
@Resource
WaterService waterService;
+ private final static int MIN_SIZE = 10;
+
+ private final static int MAX_SIZE = 80000;
+
+ private final static long Y2000 = 949334400000L;
+
@ApiOperation(value = "鑾峰彇褰撳墠鏃堕棿")
@GetMapping("/getTime")
public Object getTime() {
@@ -33,25 +45,115 @@
@ApiOperation(value = "鑾峰彇鍏冩暟鎹俊鎭�")
@GetMapping("/{serviceName}/layer.json")
- public Object getLayer(@PathVariable String serviceName) {
- return serviceName;
+ public void getLayer(@PathVariable String serviceName, HttpServletResponse res) {
+ try {
+ if (!validate(serviceName, res)) {
+ 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());
+ }
}
@ApiOperation(value = "鑾峰彇鍦板舰楂樺害鍥�")
- @GetMapping("/{serviceName}/terrain?width={width}&height={height}")
- public Object getTerraMap(@PathVariable String serviceName, @PathVariable Integer width, @PathVariable Integer height) {
- return serviceName + "," + width + "," + height;
+ @GetMapping("/{serviceName}/terrain")
+ public void getTerraMap(@PathVariable String serviceName, Integer width, Integer height, HttpServletResponse res) {
+ try {
+ if (!validate(serviceName, width, height, res)) {
+ 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());
+ }
}
@ApiOperation(value = "鑾峰彇姘撮潰楂樺害鍥�")
- @GetMapping("/{serviceName}/waterMap?width={width}&height={height}&time={timestamp}")
- public Object getWaterMap(@PathVariable String serviceName, @PathVariable Integer width, @PathVariable Integer height, @PathVariable Long timestamp) {
- return serviceName + "," + width + "," + height + "," + timestamp;
+ @GetMapping("/{serviceName}/waterMap")
+ public void getWaterMap(@PathVariable String serviceName, Integer width, Integer height, Long timestamp, HttpServletResponse res) {
+ try {
+ if (!validate(serviceName, width, height, timestamp, res)) {
+ 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());
+ }
}
@ApiOperation(value = "鑾峰彇姘存祦鍚戞祦閫熷浘")
- @GetMapping("/{serviceName}/flowMap?width={width}&height={height}&time={timestamp}")
- public Object getFlowMap(@PathVariable String serviceName, @PathVariable Integer width, @PathVariable Integer height, @PathVariable Long timestamp) {
- return serviceName + "," + width + "," + height + "," + timestamp;
+ @GetMapping("/{serviceName}/flowMap")
+ public void getFlowMap(@PathVariable String serviceName, Integer width, Integer height, Long timestamp, HttpServletResponse res) {
+ try {
+ if (!validate(serviceName, width, height, timestamp, res)) {
+ 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());
+ }
+ }
+
+ private boolean validate(String serviceName, HttpServletResponse res) {
+ if (StringHelper.isEmpty(serviceName)) {
+ return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "鏈嶅姟鍚嶄笉鑳戒负绌�");
+ }
+
+ return true;
+ }
+
+ private boolean validate(String serviceName, Integer width, Integer height, HttpServletResponse res) {
+ 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, "鏈嶅姟鍚嶄笉鑳戒负绌�");
+ }
+ if (null == width || width < MIN_SIZE || width > MAX_SIZE) {
+ return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "鍥惧儚瀹藉害涓嶈兘涓虹┖涓斿彇鍊艰寖鍥翠负" + MIN_SIZE + "~" + MAX_SIZE + "涔嬮棿");
+ }
+ 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骞�");
+ }
+
+ 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