From 3430a3dd889ba51d06cbdb77014aa63d0a070e48 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Fri, 16 Aug 2024 16:38:08 +0800
Subject: [PATCH] 优化国土对接接口

---
 src/main/java/com/dji/sample/patches/utils/TimerUtil.java |  209 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 206 insertions(+), 3 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 005ea33..b8c1fe2 100644
--- a/src/main/java/com/dji/sample/patches/utils/TimerUtil.java
+++ b/src/main/java/com/dji/sample/patches/utils/TimerUtil.java
@@ -2,6 +2,10 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.dji.sample.common.model.CustomClaim;
+import com.dji.sample.droneairport.dao.DbUploadMapper;
+import com.dji.sample.droneairport.model.Entity.DbUploadEntity;
+import com.dji.sample.manage.dao.IDeviceMapper;
+import com.dji.sample.manage.model.entity.DeviceEntity;
 import com.dji.sample.media.dao.IFileMapper;
 import com.dji.sample.media.model.MediaFileEntity;
 import com.dji.sample.patches.config.pojo.PatchesConfigPojo;
@@ -53,6 +57,7 @@
 
 import static com.dji.sample.patches.utils.MultipartFileTOFileUtil.convert;
 import static com.dji.sample.patches.utils.ZipUtil.zipFolder;
+import static com.dji.sample.territory.utils.SM3.sm3;
 import static com.dji.sample.wayline.model.enums.WaylineTaskTypeEnum.TIMED;
 import static com.dji.sample.wayline.model.enums.WaylineTemplateTypeEnum.WAYPOINT;
 
@@ -78,7 +83,10 @@
     private TerritoryConfigPojo territoryConfigPojo;
     @Autowired
     private IWaylineJobBreakPointService jobBreakPointService;
-
+    @Autowired
+    private IDeviceMapper deviceMapper;
+    @Autowired
+    private DbUploadMapper dbUploadMapper;
     /**
      * 定时器,将没有规划的图斑生成航线,并将航线飞完后的成果数据进行保存推送
      *
@@ -98,10 +106,11 @@
             //根据获取的图斑集合获得kmz航线文件
             MultipartFile multipartFile = null;
             try {
-                multipartFile = getFile(waylineName, list,28.62452712442823, 115.85666327144976);
+                multipartFile = getFile(waylineName, list, 28.62452712442823, 115.85666327144976);
             } catch (IOException e) {
                 throw new RuntimeException(e);
             }
+            taskId = list.get(0).getTaskId();
             //上传航线文件
             WaylineFileEntity waylineFile = backWayline(multipartFile, waylineName, workspaceId, backclaim().getUsername());
             //将为规划的图斑状态更新为已规划
@@ -236,6 +245,7 @@
 //            }
 //        }
 //    }
+
     /**
      * 完成对未推送的图斑数据进行整合发送
      *
@@ -247,6 +257,9 @@
         String taskId = "";
         String dkbh = null;
         List<List<LotInfo>> lists = convertToLists(getNoPush(workspaceId));
+        if (lists.isEmpty()){
+            return;
+        }
         for (List<LotInfo> lotInfo : lists) {
             for (LotInfo lotInfo1 : lotInfo) {
                 List<MediaFileEntity> media = getPatchesService.listPohto(lotInfo1.getDkbh(), workspaceId);
@@ -270,6 +283,56 @@
 
         }
     }
+    @Scheduled(cron = "0 3 17 * * ?")
+   public void GtdbOperation() throws Exception {
+    String taskId = "";
+    String dkbh = null;
+    String code="";
+    int count = 0;
+    Set<String> dkbhSet = new HashSet<>(); // 用于存储不同的dkbh
+    List<List<LotInfo>> lists = convertToLists(getNoPush());
+        if (lists.isEmpty()){
+            return;
+        }
+        tbFJService.deleteData();
+    for (List<LotInfo> lotInfo : lists) {
+        for (LotInfo lotInfo1 : lotInfo) {
+            List<MediaFileEntity> media = getPatchesService.listPohto(lotInfo1.getDkbh());
+            // 清空FJ表
+            for (MediaFileEntity mediaFile : media) {
+                // 获取媒体文件名里的dkbh和taskId
+                dkbh = getDkbh(mediaFile.getFileName());
+                dkbhSet.add(dkbh); // 添加dkbh到set中
+                // 获取该媒体文件的图斑信息
+                LotInfo lotInfo2 = getPatchesService.getLotinfo(dkbh);
+                taskId = lotInfo2.getTaskId();
+                code=lotInfo2.getXzqdm();
+                count++;
+                // 插入数据到数据库
+                tbFJService.insertOneData(mediaFile, lotInfo2);
+                getPatchesService.patchesPushed(taskId, dkbh);
+            }
+        }
+
+        // 存储db到服务器
+        String dbName = dbSaveGt(territoryConfigPojo.getResult(), territoryConfigPojo.getResultGtsave(), taskId);
+        DbUploadEntity entity = new DbUploadEntity();
+        File file = new File(territoryConfigPojo.getResult());
+        byte[] bytesArray = Files.readAllBytes(file.toPath());
+        long currentTimeMillis = System.currentTimeMillis();
+        entity.setTimestamp(currentTimeMillis);
+        entity.setSize(getFileSize(file))  ;
+        entity.setHash(sm3(bytesArray));
+        entity.setCount(dkbhSet.size());// 统计dkbh的数量
+        entity.setUrl(territoryConfigPojo.getResultGtsave()+"/"+dbName);
+        entity.setRegioncode(code);
+        entity.setExtension(".db");
+        entity.setAttachmentCount(count);
+        entity.setTaskId(taskId);
+        dbUploadMapper.insert(entity);
+    }
+}
+
 
     /**
      * 创建航线
@@ -279,7 +342,7 @@
      * @return
      * @throws IOException
      */
-    public MultipartFile getFile(String waylineName, List<LotInfo> list,double lat, double lon) throws IOException {
+    public MultipartFile getFile(String waylineName, List<LotInfo> list, double lat, double lon) throws IOException {
         //南昌28.62452712442823, 115.85666327144976  瑞金 25.8917266,116.020940643
         List<PointPO> coordinates = GeoToolsUtil.getRoutePointOrder(list, lat, lon);
         XMLTemplateModel xmlModel = XMLTemplateModel.init(coordinates, list);
@@ -349,8 +412,16 @@
     public List<List<LotInfo>> getNoPush(String workspaceId) {
         List<LotInfo> list = patchesMapper.selectList(new LambdaQueryWrapper<LotInfo>()
                 .eq(LotInfo::getIsPush, 0)
+                .eq(LotInfo::getType, 2)
                 .eq(LotInfo::getInvestigate, 1)
                 .eq(LotInfo::getWorkspaceId, workspaceId));
+        return groupTasks(list);
+    }
+    public List<List<LotInfo>> getNoPush() {
+        List<LotInfo> list = patchesMapper.selectList(new LambdaQueryWrapper<LotInfo>()
+                .eq(LotInfo::getIsPush, 0)
+                .eq(LotInfo::getType, 1)
+                .eq(LotInfo::getInvestigate, 1));
         return groupTasks(list);
     }
 
@@ -491,6 +562,40 @@
         return Files.readAllBytes(path);
     }
 
+    public  void sendPostWithParameters(String name, String fileId, String time, List<Long> taskDays, List<List<Long>> taskPeriods, String jobId,String docksn) throws IOException {
+        try {
+            RestTemplate restTemplate = new RestTemplate();
+
+            // 构建请求体
+            String jsonBody = buildRequestBody(name, fileId, time, taskDays, taskPeriods, jobId,docksn);
+            String workspaceId =getWorkspaceIdByDocksn(docksn);
+            // 设置请求头
+            HttpHeaders headers = new HttpHeaders();
+            headers.setContentType(MediaType.APPLICATION_JSON);
+            headers.setAccept(List.of(MediaType.APPLICATION_JSON, MediaType.TEXT_PLAIN, MediaType.ALL));
+            headers.set("Accept-Language", "zh-CN,zh;q=0.9");
+            headers.setConnection("keep-alive");
+            headers.set("x-auth-token", "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ3b3Jrc3BhY2VfaWQiOiJlM2RlYTBmNS0zN2YyLTRkNzktYWU1OC00OTBhZjMyMjgwNjkiLCJzdWIiOiJDbG91ZEFwaVNhbXBsZSIsInVzZXJfdHlwZSI6IjEiLCJuYmYiOjE3MTg1ODUxNTQsImxvZyI6IkxvZ2dlcltjb20uZGppLnNhbXBsZS5jb21tb24ubW9kZWwuQ3VzdG9tQ2xhaW1dIiwiaXNzIjoiREpJIiwiaWQiOiI5Y2MwMGY3Zi0yODMwLTRmM2UtYjdmNC1lN2E1ZjIzYjNjNGYiLCJleHAiOjE4MDQ5ODUxNTQsImlhdCI6MTcxODU4NTE1NCwidXNlcm5hbWUiOiJhZG1pbkRQIn0.eTi9DTYGwcTvnIZasKTFq9OHQgMFS1-m840terb7ss0");
+
+            // 构建请求实体
+            HttpEntity<String> requestEntity = new HttpEntity<>(jsonBody, headers);
+
+            // 发送请求
+            ResponseEntity<String> response = restTemplate.exchange(
+//                    "https://wrj.shuixiongit.com/drone-api/wayline/api/v1/workspaces/4a574db8-4ad3-48f7-9f16-3edbcd8056e1/flight-tasks",
+                    "http://localhost:6789/wayline/api/v1/workspaces/"+workspaceId+"/flight-tasks",
+                    HttpMethod.POST,
+                    requestEntity,
+                    String.class);
+        } catch (Exception e) {
+            throw new IllegalArgumentException("Request failed: " + e.getMessage());
+        }
+    }
+
+    public String getWorkspaceIdByDocksn(String docksn){
+      DeviceEntity entity=  deviceMapper.selectOne(new LambdaQueryWrapper<DeviceEntity>().eq(DeviceEntity::getDeviceSn,docksn));
+      return entity.getWorkspaceId();
+    }
     public static void sendPostWithParameters(String name, String fileId, String time, List<Long> taskDays, List<List<Long>> taskPeriods) throws IOException {
         try {
             RestTemplate restTemplate = new RestTemplate();
@@ -554,6 +659,37 @@
                 + "}", name, fileId, select_execute_time, taskDays, taskPeriods);
     }
 
+    private static String buildRequestBody(String name, String fileId, String select_execute_time, List<Long> taskDays, List<List<Long>> taskPeriods, String jobid, String dockSn) {
+        // 将 List 和 String 类型的参数格式化为 JSON 字符串
+        String taskDaysJson = taskDays != null ? taskDays.toString() : "[]";
+        String taskPeriodsJson = taskPeriods != null ? taskPeriods.toString() : "[]";
+
+        // 使用 String.format 来生成 JSON 字符串
+        return String.format("{"
+                + "\"name\":\"%s\","
+                + "\"file_id\":\"%s\","
+                + "\"dock_sn\":\"%s\","
+                + "\"task_type\":1,"
+                + "\"select_execute_time\":\"%s\","
+                + "\"rth_altitude\":100,"
+                + "\"out_of_control_action\":0,"
+                + "\"executeTimeContinuousArr\":[{\"index\":1,\"value\":[]}],"
+                + "\"executeTimeRepeatArr\":[{\"index\":1,\"value\":\"\"}],"
+                + "\"rep_fre_val\":1,"
+                + "\"rep_fre_type\":3,"
+                + "\"rep_rule_type\":1,"
+                + "\"time_range\":[],"
+                + "\"rth_mode\":1,"
+                + "\"1conValue\":[],"
+                + "\"1repValue\":\"\","
+                + "\"task_days\":%s,"
+                + "\"task_periods\":%s,"
+                + "\"wayline_type\":0,"
+                + "\"job_id\":\"%s\""
+                + "}", name, fileId, dockSn, select_execute_time, taskDaysJson, taskPeriodsJson, jobid);
+    }
+
+
     /**
      * 将上传的db保存一份到本地
      *
@@ -589,6 +725,21 @@
             throw new RuntimeException("Db文件复制失败", e); // 捕获并抛出原始异常
         }
     }
+    public String dbSaveGt(String path, String folder, String taskId) {
+        try {
+            String fileName = taskId + ".db";
+            // 构建目标文件路径
+            String targetFilePath = folder + File.separator + fileName;
+
+            // 复制文件到目标位置
+            Path sourcePath = Paths.get(path);
+            Path targetPath = Paths.get(targetFilePath);
+            Files.copy(sourcePath, targetPath, StandardCopyOption.REPLACE_EXISTING);
+            return fileName;
+        } catch (IOException e) {
+            throw new RuntimeException("Db文件复制失败", e); // 捕获并抛出原始异常
+        }
+    }
 
     /**
      * 根据时间创建db存储时的文件夹
@@ -615,7 +766,22 @@
         }
         return folder;
     }
+    private static File fileFold(String fold,String taskId) {
+        // 获取当前时间
+        LocalDateTime currentTime = LocalDateTime.now();
 
+        // 拼接文件夹路径
+        String folderPath = fold + File.separator + taskId;
+
+        // 创建文件夹
+        File folder = new File(folderPath);
+        if (!folder.exists()) {
+            if (!folder.mkdirs()) {
+                throw new RuntimeException("文件创建失败" + folderPath);
+            }
+        }
+        return folder;
+    }
     public static Long getCurrentTimestampPlus8Hours(long hours) {
         // 获取当前时间
         LocalDateTime now = LocalDateTime.now();
@@ -630,6 +796,21 @@
         // 将毫秒级时间戳转换为秒级时间戳
         return timestamp / 1000;
     }
+    public static long getNextDayEightAMTimestamp(int hour) {
+        // 获取当前时间
+        LocalDateTime now = LocalDateTime.now();
+
+        // 获取第二天的日期并设置时间为08:00
+        LocalDateTime nextDayEightAM = now.plusDays(1).withHour(8).withMinute(0).withSecond(0).withNano(0);
+
+        // 根据传入的hour参数增加小时
+        LocalDateTime adjustedTime = nextDayEightAM.plusHours(hour);
+
+        // 转换为时间戳(秒级)
+        ZonedDateTime zonedDateTime = adjustedTime.atZone(ZoneId.systemDefault());
+        return zonedDateTime.toEpochSecond();
+    }
+
 
     public static Long addOneHourToTimestamp(Long timestampInSeconds, long hours, long min) {
         // 将输入的秒级时间戳转换为Instant对象
@@ -664,10 +845,32 @@
         return "云飞行计划" + currentTime.format(DateTimeFormatter.ofPattern("MMddHHmmssSS"));
     }
 
+    public static String getTimeName() {
+        LocalDateTime currentTime = LocalDateTime.now();
+        // 格式化时间,生成当前时间
+        return "国土调查云" + currentTime.format(DateTimeFormatter.ofPattern("MMddHHmmssSS"));
+    }
+
     public static String getNowDay() {
         LocalDateTime currentTime = LocalDateTime.now();
 
         // 格式化时间,生成当前时间
         return currentTime.format(DateTimeFormatter.ofPattern("MMddHHmmssSSS"));
     }
+    public static long getFileSize(File file) {
+        if (file == null || !file.exists() || !file.isFile()) {
+            return -1L; // 文件不存在或不是普通文件
+        }
+        return file.length();
+    }
+
+    public static void main(String[] args) {
+        File file = new File("DB/result_db.db");
+        long size = getFileSize(file);
+        System.out.println("文件大小: " + size + " 字节");
+    }
+    public static void  getFileInfo(File file){
+        long size=getFileSize(file);
+
+    }
 }

--
Gitblit v1.9.3