From 2db1aa88e8ab53096a936163d686b90d8e056a99 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Wed, 21 Aug 2024 23:18:33 +0800
Subject: [PATCH] 国土对接返回信息加密

---
 src/main/java/com/dji/sample/territory/controller/TbDkjbxxController.java |   84 +++++++++++++++++++++++++----------------
 1 files changed, 51 insertions(+), 33 deletions(-)

diff --git a/src/main/java/com/dji/sample/territory/controller/TbDkjbxxController.java b/src/main/java/com/dji/sample/territory/controller/TbDkjbxxController.java
index 30f6619..262ff0a 100644
--- a/src/main/java/com/dji/sample/territory/controller/TbDkjbxxController.java
+++ b/src/main/java/com/dji/sample/territory/controller/TbDkjbxxController.java
@@ -72,21 +72,39 @@
      * @return
      */
     @PostMapping("/upload")
-    public ResponseResult uploadFile(@RequestParam("file") MultipartFile file){
-        String workspaceId="4a574db8-4ad3-48f7-9f16-3edbcd8056e1";
+    public ResponseResult uploadFile(@RequestParam("file") MultipartFile file,
+                                     @RequestParam String taskName,
+                                     @RequestParam String taskId) {
+        String workspaceId = "4a574db8-4ad3-48f7-9f16-3edbcd8056e1";
+        if (file == null) {
+            return ResponseResult.error("上传文件为空");
+        }
         tbDkjbxxService.uploadFile(file);
         List<TbDkjbxxEntity> list = tbDkjbxxService.list();
-        shpToDataSourceService.savaInMysql(list, workspaceId);
+        if (list.size() == 0) {
+            return ResponseResult.error("上传文件数据为空");
+        }
+        try {
+            shpToDataSourceService.savaInMysql(list, workspaceId, taskId, taskName);
+        } catch (Exception e) {
+            return ResponseResult.error("上传db文件的内容格式不匹配");
+        }
         return ResponseResult.success("上传成功");
     }
 
+
     @PostMapping("/uploadUrl")
-    public ResponseResult uploadUrl( @RequestBody UploadUrlParam param){
-        tbDkjbxxService.uploadUrl(param.getDbUrl());
-        String workspaceId="4a574db8-4ad3-48f7-9f16-3edbcd8056e1";
-        List<TbDkjbxxEntity> list = tbDkjbxxService.list();
-        shpToDataSourceService.savaInMysql(list, workspaceId);
-        return ResponseResult.success("上传成功");
+    public ResponseResult uploadUrl(@RequestBody UploadUrlParam param) {
+        try {
+            tbDkjbxxService.uploadUrl(param.getDbUrl(), param.getTaskName());
+            String workspaceId = "4a574db8-4ad3-48f7-9f16-3edbcd8056e1";
+            List<TbDkjbxxEntity> list = tbDkjbxxService.list();
+            shpToDataSourceService.savaInMysql(list, workspaceId, param.getTaskId(), param.getTaskName());
+        } catch (Exception e) {
+            return ResponseResult.error("文件下载发生错误" + e);
+        }
+
+        return ResponseResult.success(200);
     }
 
     /**
@@ -100,28 +118,28 @@
      * @return
      * @throws Exception
      */
-    @PostMapping("/uploadLot")
-    public ResponseResult getGeo(@RequestParam String workspaceId,
-                                 @RequestParam String waylineName,
-                                 @RequestParam double airportLat,
-                                 @RequestParam double airportLon,
-                                 HttpServletRequest request) throws Exception {
-        List<TbDkjbxxEntity> list = tbDkjbxxService.list();
-        shpToDataSourceService.savaInMysql(list, workspaceId);
-        MultipartFile multipartFile = tbDkjbxxService.listFile(workspaceId, waylineName, airportLat, airportLon);
-        CustomClaim customClaim = (CustomClaim) request.getAttribute(TOKEN_CLAIM);
-        String creator = customClaim.getUsername();
-        waylineFileService.importKmzFileBack(multipartFile, workspaceId, creator);
-        WaylineFileEntity entity = waylineFileService.selectByName(waylineName);
-        Map<String, String> infoMap = new HashMap<>();
-        URL url = null;
-        try {
-            url = waylineFileService.getObjectUrl(workspaceId, entity.getWaylineId());
-            infoMap.put("url", String.valueOf(MinioUrlUtils.getUrl(url)));
-            infoMap.put("waylineId", entity.getWaylineId());
-        } catch (SQLException e) {
-            throw new RuntimeException(e);
-        }
-        return ResponseResult.success(infoMap);
-    }
+//    @PostMapping("/uploadLot")
+//    public ResponseResult getGeo(@RequestParam String workspaceId,
+//                                 @RequestParam String waylineName,
+//                                 @RequestParam double airportLat,
+//                                 @RequestParam double airportLon,
+//                                 HttpServletRequest request) throws Exception {
+//        List<TbDkjbxxEntity> list = tbDkjbxxService.list();
+//        shpToDataSourceService.savaInMysql(list, workspaceId);
+//        MultipartFile multipartFile = tbDkjbxxService.listFile(workspaceId, waylineName, airportLat, airportLon);
+//        CustomClaim customClaim = (CustomClaim) request.getAttribute(TOKEN_CLAIM);
+//        String creator = customClaim.getUsername();
+//        waylineFileService.importKmzFileBack(multipartFile, workspaceId, creator);
+//        WaylineFileEntity entity = waylineFileService.selectByName(waylineName);
+//        Map<String, String> infoMap = new HashMap<>();
+//        URL url = null;
+//        try {
+//            url = waylineFileService.getObjectUrl(workspaceId, entity.getWaylineId());
+//            infoMap.put("url", String.valueOf(MinioUrlUtils.getUrl(url)));
+//            infoMap.put("waylineId", entity.getWaylineId());
+//        } catch (SQLException e) {
+//            throw new RuntimeException(e);
+//        }
+//        return ResponseResult.success(infoMap);
+//    }
 }
\ No newline at end of file

--
Gitblit v1.9.3