From a6fe405947004d6571806edabd8f14357e144cfa Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 14 十月 2024 17:31:31 +0800
Subject: [PATCH] 1
---
src/main/java/com/se/simu/controller/WaterController.java | 113 ++++++++++++++++++++++++++++++++++++++++++++++++--------
1 files changed, 96 insertions(+), 17 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..6d25649 100644
--- a/src/main/java/com/se/simu/controller/WaterController.java
+++ b/src/main/java/com/se/simu/controller/WaterController.java
@@ -1,18 +1,21 @@
package com.se.simu.controller;
+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;
/**
- * 鍐呮稘鎺у埗鍣�
+ * 鍐呮稘绠＄悊
*
* @author WWW
* @date 2024-07-16
@@ -25,33 +28,109 @@
@Resource
WaterService waterService;
- @ApiOperation(value = "鑾峰彇褰撳墠鏃堕棿")
- @GetMapping("/getTime")
- public Object getTime() {
- return System.currentTimeMillis();
- }
+ private final static int MIN_SIZE = 10;
+
+ private final static int MAX_SIZE = 80000;
+
+ private final static long Y2000 = 949334400000L;
@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;
+ }
+
+ byte[] bytes = waterService.getLayerJson(serviceName);
+
+ WebHelper.writeBytes(bytes, res);
+ } 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);
+
+ WebHelper.writePng(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, timestamp);
+
+ WebHelper.writePng(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);
+
+ WebHelper.writePng(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 (WebHelper.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 (WebHelper.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 < 0) {
+ return WebHelper.writeJson2Page(res, HttpStatus.BAD_REQUEST, "鏃堕棿涓嶈兘涓虹┖涓斿ぇ浜�0");
+ }
+
+ return true;
}
}
--
Gitblit v1.9.3