From dd50418315218979f8d596f34f185ae4a28adc4a Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 10 四月 2023 12:39:04 +0800
Subject: [PATCH] 1

---
 src/main/java/com/terra/coal/controller/MainController.java |   66 +++++++++++++++++++++++++++++++++
 1 files changed, 66 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/terra/coal/controller/MainController.java b/src/main/java/com/terra/coal/controller/MainController.java
index e58fffa..016be14 100644
--- a/src/main/java/com/terra/coal/controller/MainController.java
+++ b/src/main/java/com/terra/coal/controller/MainController.java
@@ -1,14 +1,24 @@
 package com.terra.coal.controller;
 
+import com.terra.coal.entity.CountEntity;
+import com.terra.coal.entity.ResponseMsg;
+import com.terra.coal.entity.StaticData;
+import com.terra.coal.helper.StringHelper;
 import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiImplicitParam;
+import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.servlet.ModelAndView;
+import com.terra.coal.service.MainService;
 
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import java.io.File;
+import java.util.List;
 
 /**
  * CoalController
@@ -18,6 +28,9 @@
 @RestController
 @RequestMapping("/main")
 public class MainController extends BaseController {
+    @Autowired
+    MainService mainService;
+
     @ApiOperation(value = "璺宠浆棣栭〉")
     @GetMapping({"/", "/toIndex"})
     public ModelAndView toIndex(ModelAndView mv, HttpServletRequest req) {
@@ -34,4 +47,57 @@
 
         return mv;
     }
+
+    @ApiOperation(value = "鍔犺浇54鏁版嵁(鍏ュ簱)")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "txtPath", value = "鍧椾綋鏁版嵁鏂囦欢璺緞", dataType = "String", paramType = "query", example = "D:\\鍧椾綋鏁版嵁.txt")
+    })
+    @GetMapping(value = "/load54Data")
+    public ResponseMsg<Integer> load54Data(String txtPath) {
+        try {
+            if (StringHelper.isEmpty(txtPath)) {
+                return fail("璇烽�夋嫨鈥樺潡浣撴暟鎹�.txt鈥�", null);
+            }
+
+            File f = new File(txtPath);
+            if (!f.exists() || f.isDirectory()) {
+                return fail("鏂囦欢涓嶅瓨鍦�", null);
+            }
+
+            Integer rows = mainService.load54Data(f);
+
+            return success("鎴愬姛", rows);
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+            return fail(ex.getMessage(), null);
+        }
+    }
+
+    @ApiOperation(value = "鏍规嵁鑼冨洿缁熻")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "wkt", value = "WKT鏂囨湰", dataType = "String", paramType = "query", example = "POLYGON ((113.5807 45.4897,113.5898 45.4893,113.5857 45.4837,113.5807 45.4897))"),
+            @ApiImplicitParam(name = "z", value = "楂樺害", dataType = "double", paramType = "query", example = "640"),
+            @ApiImplicitParam(name = "deep", value = "娣卞害", dataType = "double", paramType = "query", example = "50")
+    })
+    @GetMapping(value = "/countByRange")
+    public ResponseMsg<Object> countByRange(String wkt, double z, double deep) {
+        try {
+            if (StringHelper.isEmpty(wkt)) {
+                return fail("璇疯緭鍏ユ煡璇㈣寖鍥�(WKT瀛楃涓�)", null);
+            }
+            if (z < StaticData.MINUS_ONE_THOUSAND || z > StaticData.NINE_THOUSAND) {
+                return fail("楂樺害鍊艰秴鍑鸿寖鍥�(-1000~9000)", null);
+            }
+            if (deep < 1) {
+                return fail("寮�鎸栨繁搴﹁繃灏�", null);
+            }
+
+            List<CountEntity> rs = mainService.countByRange("'" + wkt + "'", z, z + deep);
+
+            return success("鎴愬姛", rs);
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+            return fail(ex.getMessage(), null);
+        }
+    }
 }

--
Gitblit v1.9.3