From 1f0549559468a1501ac572f8b26b3daedd2b44e1 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 15 二月 2023 14:15:18 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/show/ComprehensiveController.java |  121 +++++++++++++++++++++++++++++++++++++++-
 1 files changed, 118 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/show/ComprehensiveController.java b/src/main/java/com/lf/server/controller/show/ComprehensiveController.java
index 8843aa0..33e4624 100644
--- a/src/main/java/com/lf/server/controller/show/ComprehensiveController.java
+++ b/src/main/java/com/lf/server/controller/show/ComprehensiveController.java
@@ -2,16 +2,33 @@
 
 import com.lf.server.annotation.SysLog;
 import com.lf.server.controller.all.BaseController;
+import com.lf.server.entity.all.HttpStatus;
 import com.lf.server.entity.all.ResponseMsg;
+import com.lf.server.entity.all.StaticData;
+import com.lf.server.entity.ctrl.MarkJsonEntity;
+import com.lf.server.entity.data.DownloadEntity;
+import com.lf.server.entity.data.MetaFileEntity;
+import com.lf.server.entity.sys.UserEntity;
+import com.lf.server.helper.PathHelper;
+import com.lf.server.helper.StringHelper;
+import com.lf.server.helper.WebHelper;
+import com.lf.server.mapper.show.MarkMapper;
 import com.lf.server.service.all.BaseQueryService;
+import com.lf.server.service.all.BaseUploadService;
+import com.lf.server.service.data.DownloadService;
+import com.lf.server.service.show.MarkService;
+import com.lf.server.service.sys.DownlogService;
+import com.lf.server.service.sys.TokenService;
 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.bind.annotation.*;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.util.List;
 
 /**
  * 缁煎悎灞曠ず
@@ -23,6 +40,21 @@
 public class ComprehensiveController extends BaseController {
     @Autowired
     BaseQueryService baseQueryService;
+
+    @Autowired
+    MarkService markService;
+
+    @Autowired
+    TokenService tokenService;
+
+    @Autowired
+    BaseUploadService baseUploadService;
+
+    @Autowired
+    DownlogService downlogService;
+
+    @Autowired
+    DownloadService downloadService;
 
     @SysLog()
     @ApiOperation(value = "鏌ヨ璺綉")
@@ -42,4 +74,87 @@
             return fail(ex.getMessage(), null);
         }
     }
+
+    @SysLog()
+    @ApiOperation(value = "涓婁紶ShapeFile鏂囦欢")
+    @ResponseBody
+    @PostMapping(value = "/uploadShp", produces = "application/json; charset=UTF-8")
+    public ResponseMsg<Object> uploadShp(HttpServletRequest req, HttpServletResponse res) {
+        try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                return fail("鐢ㄦ埛鏈櫥褰�", null);
+            }
+
+            List<MetaFileEntity> meList = baseUploadService.uploadData(null, null, req, res);
+            if (meList == null || meList.size() < StaticData.FOUR) {
+                return fail("娌℃湁鎵惧埌宸蹭笂浼犵殑鏁版嵁鎴栦笉瀹屾暣", null);
+            }
+
+            List<MarkJsonEntity> list = markService.readShpForMarks(meList);
+
+            return success(list);
+        } catch (Exception ex) {
+            return fail(ex.getMessage(), null);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "涓嬭浇ShapeFile鏂囦欢")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "list", value = "鏍囩粯JSON瀹炰綋绫婚泦鍚�", dataType = "MarkJsonEntity", paramType = "body")
+    })
+    @ResponseBody
+    @PostMapping(value = "/downloadShp", produces = "application/json; charset=UTF-8")
+    public ResponseMsg<String> downloadShp(@RequestBody List<MarkJsonEntity> list, HttpServletRequest req, HttpServletResponse res) {
+        try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                return fail("鐢ㄦ埛鏈櫥褰�", null);
+            }
+            if (list == null || list.isEmpty()) {
+                return fail("鎵句笉鍒版爣缁樻暟鎹�", null);
+            }
+
+            String guid = markService.downloadShp(ue, list);
+            if (StringHelper.isNull(guid)) {
+                return fail("鐢熸垚ShapeFile鏂囦欢澶辫触", null);
+            }
+
+            return success(guid);
+        } catch (Exception ex) {
+            return fail(ex.getMessage(), null);
+        }
+    }
+
+    @SysLog()
+    @ApiOperation(value = "涓嬭浇鏂囦欢")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "guid", value = "鏂囦欢GUID", dataType = "String", paramType = "query")
+    })
+    @RequestMapping(value = "/downloadFile", method = RequestMethod.GET)
+    public void downloadFile(String guid, HttpServletRequest req, HttpServletResponse res) {
+        try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, "鐢ㄦ埛鏈櫥褰�"));
+            }
+
+            DownloadEntity de = downloadService.selectByGuid(guid);
+            if (de == null) {
+                WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�"));
+            }
+
+            downlogService.updateInfos(ue, de, req);
+
+            String filePath = markService.getDownloadFilePath(de);
+            WebHelper.download(filePath, de.getName(), res);
+        } catch (Exception ex) {
+            try {
+                WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, ex.getMessage()));
+            } catch (Exception e) {
+                log.error(e.getMessage(), e);
+            }
+        }
+    }
 }

--
Gitblit v1.9.3