From 5c288f2960d4256e1a5cfff37dcc976e62bd86ce Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Tue, 14 May 2024 14:03:56 +0800
Subject: [PATCH] 完善定时器,修改相关信息

---
 src/main/java/com/dji/sample/patches/utils/TimerUtil.java |  177 ++++++++++++++++++++++++++++++++++++++++-------------------
 1 files changed, 120 insertions(+), 57 deletions(-)

diff --git a/src/main/java/com/dji/sample/patches/utils/TimerUtil.java b/src/main/java/com/dji/sample/patches/utils/TimerUtil.java
index 22cf892..7bae1d2 100644
--- a/src/main/java/com/dji/sample/patches/utils/TimerUtil.java
+++ b/src/main/java/com/dji/sample/patches/utils/TimerUtil.java
@@ -3,6 +3,7 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.dji.sample.common.model.CustomClaim;
+import com.dji.sample.common.model.ResponseResult;
 import com.dji.sample.media.dao.IFileMapper;
 import com.dji.sample.media.model.MediaFileEntity;
 import com.dji.sample.patches.config.pojo.PatchesConfigPojo;
@@ -28,6 +29,8 @@
 import org.springframework.web.client.RestTemplate;
 import org.springframework.web.multipart.MultipartFile;
 
+import javax.xml.transform.Result;
+import java.nio.file.StandardCopyOption;
 import java.time.LocalDateTime;
 
 import java.io.*;
@@ -38,6 +41,7 @@
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.sql.SQLException;
+import java.time.format.DateTimeFormatter;
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
@@ -68,43 +72,59 @@
     @Autowired
     private TerritoryConfigPojo territoryConfigPojo;
 
+
     /**
-     * 定时
+     * 定时器,将没有规划的图斑生成航线,并将航线飞完后的成果数据进行保存推送
      *
      * @throws IOException
      * @throws SQLException
      */
-    @Scheduled(cron = "45 1 16 * * ?")
+    @Scheduled(cron = "0 0 0 * * ?")
     public void myTask() throws Exception {
+        String taskId="";
         String workspaceId = "4a574db8-4ad3-48f7-9f16-3edbcd8056e1";
-        //获取未规划的图斑集合
-//        List<List<LotInfo>> lists = getNoPlan();
-//        List<List<LotInfo>> convertedLists = convertToLists(lists);
-//        for (List<LotInfo> list : convertedLists) {
-//            long timestamp = System.currentTimeMillis();
-//            String waylineName = "云飞行调查" + timestamp;
-//            //根据获取的图斑集合获得kmz航线文件
-//            MultipartFile multipartFile = getFile(waylineName, list);
-//            //上传航线文件
-//            String waylineId = backWayline(multipartFile, waylineName, workspaceId, backclaim().getUsername());
-//            //将为规划的图斑状态更新为已规划
-//            updatePatchesStatu(list);
-//            //调用publshFlightTask方法创建定时任务
-//            waylineJobService.publishFlightTask(JobParam(waylineId), backclaim());
-//        }
+//        获取未规划的图斑集合
+        List<List<LotInfo>> lists = getNoPlan();
+        List<List<LotInfo>> convertedLists = convertToLists(lists);
+        for (List<LotInfo> list : convertedLists) {
+            long timestamp = System.currentTimeMillis();
+            String waylineName = "云飞行调查" + timestamp;
+            //根据获取的图斑集合获得kmz航线文件
+            MultipartFile multipartFile = getFile(waylineName, list);
+            //上传航线文件
+            backWayline(multipartFile, waylineName, workspaceId, backclaim().getUsername());
+            //将为规划的图斑状态更新为已规划
+            updatePatchesStatu(list);
+        }
         List<List<MediaFileEntity>> list = getNoadd();
-        List<List<MediaFileEntity>> media = convertToLists(list);
-        for (int i = 0; i < media.size(); i++) {
+        List<List<MediaFileEntity>> medias = convertToLists(list);
+        for (List<MediaFileEntity> media : medias) {
+            //清空FJ表
             tbFJService.deleteData();
-            String dkbh = getDkbh(media.get(i).get(i).getFileName());
-            String taskId = getTaskId(media.get(i).get(i).getFileName());
-            LotInfo lotInfo = getPatchesService.getLotinfo(dkbh, workspaceId);
-            tbFJService.insertData(media.get(i), lotInfo);
-            updateMediaStatu(media.get(i));
+            for (MediaFileEntity mediaFile : media) {
+                //获取媒体文件名里的dkbh和taskId
+                String dkbh = getDkbh(mediaFile.getFileName());
+                taskId = getTaskId(mediaFile.getFileName());
+                //获取该媒体文件的图斑信息
+                LotInfo lotInfo = getPatchesService.getLotinfo(dkbh, workspaceId);
+                tbFJService.insertOneData(mediaFile, lotInfo);
+                //更新媒体文件的是否添加状态
+                updateMediaStatu(mediaFile);
+
+            }
+            dbSave(territoryConfigPojo.getResult(), territoryConfigPojo.getResultsave(), taskId);
+            //发送请求给第三方接口
             sendPostWithFileAndParameter(territoryConfigPojo.getResult(), taskId);
         }
     }
 
+    /**
+     * 创建航线
+     * @param waylineName
+     * @param list
+     * @return
+     * @throws IOException
+     */
     public MultipartFile getFile(String waylineName, List<LotInfo> list) throws IOException {
         List<PointPO> coordinates = GeoToolsUtil.getRoutePointOrder(list, 28.62703, 115.867719);
         XMLTemplateModel xmlModel = XMLTemplateModel.init(coordinates, list);
@@ -115,6 +135,14 @@
         return convert(new File(destKMZFile));
     }
 
+    /**
+     * 将航线上传Oss
+     * @param multipartFile
+     * @param waylineName
+     * @param workspaceId
+     * @param username
+     * @return
+     */
     public String backWayline(MultipartFile multipartFile, String waylineName, String workspaceId, String username) {
         waylineFileService.importKmzFile(multipartFile, workspaceId, username);
         WaylineFileEntity entity = waylineFileService.selectByName(waylineName);
@@ -153,11 +181,9 @@
         }
     }
 
-    public void updateMediaStatu(List<MediaFileEntity> list) {
-        for (MediaFileEntity mediaFile : list) {
+    public void updateMediaStatu(MediaFileEntity mediaFile) {
             mediaFile.setIsadd(1);
             fileMapper.updateById(mediaFile);
-        }
     }
 
     public List<List<MediaFileEntity>> getNoadd() {
@@ -165,29 +191,6 @@
                 .eq(MediaFileEntity::getIsadd, 0).like(MediaFileEntity::getFileName, "~"));
         List<List<MediaFileEntity>> groupedFilenames = groupTasks(list);
         return groupedFilenames;
-    }
-
-    public CreateJobParam JobParam(String waylineId) {
-        List<List<Long>> listOfLists = new ArrayList<>();
-        List<Long> sublist = new ArrayList<>();
-        sublist.add(177271980L); // 添加整数值
-        listOfLists.add(sublist);
-        List<Long> list = new ArrayList<>();
-        list.add(177271980L);
-        CreateJobParam param = new CreateJobParam();
-        param.setName("云飞行调查");
-        param.setTaskType(TIMED);
-        param.setOutOfControlAction(0);
-        param.setRepFreType(3);
-        param.setRepFreVal(1);
-        param.setRepRuleType(1);
-        param.setRthAltitude(80);
-        param.setWaylineType(WAYPOINT);
-        param.setTaskPeriods(listOfLists);
-        param.setTaskDays(list);
-        param.setFileId(waylineId);
-        param.setDockSn("4TADKCMB010016");
-        return param;
     }
 
     public CustomClaim backclaim() {
@@ -221,7 +224,7 @@
         return new ArrayList<>(groupedTasks.values());
     }
 
-    // 从filename字段中提取任务ID(假设任务ID在文件名中以"~"开头、"."结尾)
+    // 从filename字段中提取任务ID(任务ID在文件名中以"~"开头、"."结尾)
     public static String getTaskId(String filename) {
         // 这里只是一个示例,你需要根据实际情况编写提取task_id的逻辑
         int startIndex = filename.indexOf("~") + 1;
@@ -236,6 +239,12 @@
         return filename.substring(startIndex, endIndex);
     }
 
+    /**
+     * 将成果发送post请求到第三方接口
+     * @param filePath
+     * @param taskId
+     * @throws IOException
+     */
     public static void sendPostWithFileAndParameter(String filePath, String taskId) throws IOException {
         // 创建 RestTemplate 实例
         RestTemplate restTemplate = new RestTemplate();
@@ -249,20 +258,19 @@
         // 设置请求头
         HttpHeaders headers = new HttpHeaders();
         headers.setContentType(MediaType.MULTIPART_FORM_DATA);
-
+        headers.set("X-SHARE-TOKEN", "981D9B822A9A09DE28B66243AA330DC666E1A82826B58EC3FE63FEB1EC8AA190");
+        headers.set("X-SHARE-AK", "gwc952d483-bfe5-476b-8431-ae15fe94c006");
 
         // 构建请求实体
         HttpEntity<MultiValueMap<String, Object>> requestEntity = new HttpEntity<>(body, headers);
 
         // 发送请求
-        ResponseEntity<String> response = restTemplate.exchange(
-                "http://localhost:6789/territory/tbdkjbxx/upload",
+                ResponseEntity<String> result=restTemplate.exchange(
+                "http://39.98.48.180:8087/landCloudWork/artifact/media/upload.action",
+//                "http://localhost:6789/territory/tbdkjbxx/upload",
                 HttpMethod.POST,
                 requestEntity,
                 String.class);
-
-        // 输出响应结果
-        System.out.println("Response: " + response.getBody());
     }
 
     private static MultiValueMap<String, Object> buildRequestBody(String taskId, byte[] fileContent, String filePath) {
@@ -282,4 +290,59 @@
         return Files.readAllBytes(path);
     }
 
+    /**
+     * 将上传的db保存一份到本地
+     * @param path
+     * @param folder
+     * @param taskId
+     */
+    public void dbSave(String path, String folder, String taskId) {
+        String sourceFilePath = path; // 源文件路径
+        String targetFolderPath = fileFold(folder).getPath(); // 目标文件夹路径
+
+        try {
+            // 创建目标文件夹
+            File targetFolder = new File(targetFolderPath);
+            if (!targetFolder.exists()) {
+                targetFolder.mkdirs();
+            }
+
+            // 获取当前时间
+            LocalDateTime currentTime = LocalDateTime.now();
+
+            // 格式化时间,生成文件名
+            String timeName = currentTime.format(DateTimeFormatter.ofPattern("HHmmss"));
+            String fileName = timeName + "_" + taskId + ".db";
+            // 构建目标文件路径
+            String targetFilePath = targetFolderPath + File.separator + fileName;
+
+            // 复制文件到目标文件夹
+            Path sourcePath = Paths.get(sourceFilePath);
+            Path targetPath = Paths.get(targetFilePath);
+            Files.copy(sourcePath, targetPath, StandardCopyOption.REPLACE_EXISTING);
+        } catch (IOException e) {
+            throw new RuntimeException("Db文件复制失败", e); // 捕获并抛出原始异常
+        }
+    }
+
+    private static File fileFold(String fold) {
+        // 获取当前时间
+        LocalDateTime currentTime = LocalDateTime.now();
+
+        // 格式化时间,生成文件夹名字
+        String folderName = currentTime.format(DateTimeFormatter.ofPattern("yyyyMMdd"));
+
+        // 拼接文件夹路径
+        String folderPath = fold + File.separator + folderName;
+
+        // 创建文件夹
+        File folder = new File(folderPath);
+        if (!folder.exists()) {
+            if (!folder.mkdirs()) {
+                throw new RuntimeException("文件创建失败" + folderPath);
+            }
+        }
+        return folder;
+    }
+
 }

--
Gitblit v1.9.3