From 75e6eea8cd3fac9cc888e2ffc9cdb126fab8429d Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Wed, 14 Aug 2024 15:45:55 +0800
Subject: [PATCH] 任务下发、更新、取消,优化图斑逻辑

---
 src/main/java/com/dji/sample/patches/controller/PatchesController.java |   32 +++++++++++++++++++++++++++-----
 1 files changed, 27 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/dji/sample/patches/controller/PatchesController.java b/src/main/java/com/dji/sample/patches/controller/PatchesController.java
index f6e26b9..2bee9dc 100644
--- a/src/main/java/com/dji/sample/patches/controller/PatchesController.java
+++ b/src/main/java/com/dji/sample/patches/controller/PatchesController.java
@@ -10,19 +10,21 @@
 import com.dji.sample.patches.model.entity.LotInfo;
 import com.dji.sample.patches.service.GetPatchesService;
 import com.dji.sample.patches.service.ShpToDataSourceService;
-import com.dji.sample.patches.utils.DistrictCodeUtils;
-import com.dji.sample.patches.utils.GeoToolsUtil;
-import com.dji.sample.patches.utils.PointPO;
-import com.dji.sample.patches.utils.TimerUtil;
+import com.dji.sample.patches.utils.*;
 import com.dji.sample.territory.service.impl.TbDkjbxxServiceImpl;
 import com.dji.sample.wayline.model.entity.WaylineFileEntity;
 import com.dji.sample.wayline.service.IWaylineFileService;
 import com.dji.sample.wayline.service.IWaylineJobService;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletRequest;
+import java.io.ByteArrayInputStream;
 import java.io.IOException;
 import java.net.URL;
 import java.sql.SQLException;
@@ -176,7 +178,7 @@
 //            timerUtil.mytask5();
 //            timerUtil.mytask6();
 //            timerUtil.mytask7();
-
+            timerUtil.GtdbOperation();
         } catch (Exception e) {
             throw new RuntimeException("db存储发送出现异常" + e);
         }
@@ -269,4 +271,24 @@
         }
         return ResponseResult.success(infoMap);
     }
+
+    @GetMapping("/getExcel")
+    public ResponseEntity<byte[]> exportExcel(@RequestParam String workspaceId) {
+        List<LotInfo> lotInfos = getPatchesService.listLotinfo(workspaceId);
+        try {
+            ByteArrayInputStream excelStream = ExcelUtil.exportLotInfoToExcel(lotInfos);
+            byte[] excelData = excelStream.readAllBytes();
+            excelStream.close();
+
+            HttpHeaders headers = new HttpHeaders();
+            headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=lot_info_data.xlsx");
+            headers.add(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_OCTET_STREAM_VALUE);
+            headers.setContentLength(excelData.length);
+
+            return new ResponseEntity<>(excelData, headers, HttpStatus.OK);
+        } catch (IOException e) {
+            e.printStackTrace();
+            return new ResponseEntity<>(("Failed to generate Excel file.").getBytes(), HttpStatus.INTERNAL_SERVER_ERROR);
+        }
+    }
 }

--
Gitblit v1.9.3