From dd73bbc7132f7b126b8fc2faabf2827b49622d21 Mon Sep 17 00:00:00 2001
From: guoshilong <123456>
Date: Mon, 23 Oct 2023 09:28:40 +0800
Subject: [PATCH] 获取计划最近一条

---
 src/main/java/com/dji/sample/wayline/service/impl/WaylineJobServiceImpl.java |  125 +++++++++++++++++++++++++++++++++++++----
 1 files changed, 113 insertions(+), 12 deletions(-)

diff --git a/src/main/java/com/dji/sample/wayline/service/impl/WaylineJobServiceImpl.java b/src/main/java/com/dji/sample/wayline/service/impl/WaylineJobServiceImpl.java
index 6270787..c149023 100644
--- a/src/main/java/com/dji/sample/wayline/service/impl/WaylineJobServiceImpl.java
+++ b/src/main/java/com/dji/sample/wayline/service/impl/WaylineJobServiceImpl.java
@@ -143,6 +143,7 @@
 
     /**
      * 对于即时任务,以服务器时间为准。
+     *
      * @param param
      */
     private void fillImmediateTime(CreateJobParam param) {
@@ -191,12 +192,96 @@
         return ResponseResult.success();
     }
 
+    @Override
+    public ResponseResult publishFlightTaskCondition(CreateJobParam param, CustomClaim customClaim) throws SQLException {
+        //开始日期
+        LocalDate startDate = LocalDate.ofInstant(Instant.ofEpochSecond(param.getTaskDays().get(0)), ZoneId.systemDefault());
+        long start = LocalDateTime.of(startDate, LocalTime.ofInstant(Instant.ofEpochSecond(param.getExecuteStartTimeArr().get(0).get(0)), ZoneId.systemDefault()))
+                .atZone(ZoneId.systemDefault()).toInstant().toEpochMilli();
+
+        //结束日期
+        LocalDate endDate = LocalDate.ofInstant(Instant.ofEpochSecond(param.getTaskDays().get(1)), ZoneId.systemDefault());
+        long end = LocalDateTime.of(endDate, LocalTime.ofInstant(Instant.ofEpochSecond(param.getExecuteStartTimeArr().get(param.getExecuteStartTimeArr().size() - 1).get(0)), ZoneId.systemDefault()))
+                .atZone(ZoneId.systemDefault()).toInstant().toEpochMilli();
+
+        //保存数据
+        WaylineJobEntity waylineJobEntity = WaylineJobEntity.builder()
+                .jobId(UUID.randomUUID().toString())
+                .name(param.getName())
+                .dockSn(param.getDockSn())
+                .fileId(param.getFileId())
+                .username(customClaim.getUsername())
+                .workspaceId(customClaim.getWorkspaceId())
+
+                .beginTime(start)
+                .endTime(end)
+                .status(WaylineJobStatusEnum.PENDING.getVal())
+                .taskType(param.getTaskType().getVal())
+                .waylineType(param.getWaylineType().getVal())
+                .outOfControlAction(param.getOutOfControlAction())
+                .batteryCapacity(param.getMinBatteryCapacity())
+                .rthAltitude(param.getRthAltitude())
+                .mediaCount(0)
+                .repFreVal(param.getRepFreVal())
+                .repFreType(param.getRepFreType())
+                .repRuleType(param.getRepRuleType())
+                .repRuleVal(param.getRepRuleVal())
+                .executeTimeArr(param.getTaskPeriods())
+                .executeStartTimeArr(param.getExecuteStartTimeArr())
+                .build();
+
+        Optional<WaylineJobDTO> waylineJobOpt = insertWaylineJob(waylineJobEntity);
+
+        if (waylineJobOpt.isEmpty()) {
+            throw new SQLException("任务创建失败");
+        }
+
+        WaylineJobDTO waylineJob = waylineJobOpt.get();
+
+        //循环执行时间
+        for (List<Long> timeArr : param.getExecuteStartTimeArr()) {
+            LocalDate date = LocalDate.ofInstant(Instant.ofEpochSecond(timeArr.get(0)), ZoneId.systemDefault());
+            long beginTime = LocalDateTime.of(date, LocalTime.ofInstant(Instant.ofEpochSecond(timeArr.get(0)), ZoneId.systemDefault()))
+                    .atZone(ZoneId.systemDefault()).toInstant().toEpochMilli();
+
+            long endTime = timeArr.size() > 1 && Objects.nonNull(timeArr.get(1)) ?
+                    LocalDateTime.of(date, LocalTime.ofInstant(Instant.ofEpochSecond(timeArr.get(1)), ZoneId.systemDefault()))
+                            .atZone(ZoneId.systemDefault()).toInstant().toEpochMilli() : beginTime;
+
+            if (WaylineTaskTypeEnum.IMMEDIATE != param.getTaskType() && endTime < System.currentTimeMillis()) {
+                //中断当前循环执行下一个
+                continue;
+            }
+
+
+            //条件任务
+            if (param.getTaskType() == WaylineTaskTypeEnum.CONDITION) {
+                //如果是条件任务类型,需要在任务参数中添加条件。
+                waylineJob.setConditions(
+                        WaylineTaskConditionDTO.builder()
+                                .executableConditions(Objects.nonNull(param.getMinStorageCapacity()) ? WaylineTaskExecutableConditionDTO.builder().storageCapacity(param.getMinStorageCapacity()).build() : null)
+                                .readyConditions(WaylineTaskReadyConditionDTO.builder()
+                                        .batteryCapacity(param.getMinBatteryCapacity())
+                                        .beginTime(beginTime)
+                                        .endTime(endTime)
+                                        .build())
+                                .build());
+            }
+
+            ResponseResult response = this.publishOneFlightTask(waylineJob);
+            if (ResponseResult.CODE_SUCCESS != response.getCode()) {
+                return response;
+            }
+
+        }
+        return ResponseResult.success();
+    }
+
     private void addPreparedJob(WaylineJobDTO waylineJob, CreateJobParam param, Long beginTime, Long endTime) {
         if (WaylineTaskTypeEnum.CONDITION == param.getTaskType()) {
             waylineJob.setConditions(
                     WaylineTaskConditionDTO.builder()
-                            .executableConditions(Objects.nonNull(param.getMinStorageCapacity()) ?
-                                    WaylineTaskExecutableConditionDTO.builder().storageCapacity(param.getMinStorageCapacity()).build() : null)
+                            .executableConditions(Objects.nonNull(param.getMinStorageCapacity()) ? WaylineTaskExecutableConditionDTO.builder().storageCapacity(param.getMinStorageCapacity()).build() : null)
                             .readyConditions(WaylineTaskReadyConditionDTO.builder()
                                     .batteryCapacity(param.getMinBatteryCapacity())
                                     .beginTime(beginTime)
@@ -338,7 +423,7 @@
 
         Set<String> waylineJobIds = waylineJobs.stream().map(WaylineJobDTO::getJobId).collect(Collectors.toSet());
         // Check if the task status is correct.
-        boolean isErr = !jobIds.removeAll(waylineJobIds) || !jobIds.isEmpty() ;
+        boolean isErr = !jobIds.removeAll(waylineJobIds) || !jobIds.isEmpty();
         if (isErr) {
             throw new IllegalArgumentException("以下任务状态不正确,不能取消" + Arrays.toString(jobIds.toArray()));
         }
@@ -377,10 +462,10 @@
 
     public List<WaylineJobDTO> getJobsByConditions(String workspaceId, Collection<String> jobIds, WaylineJobStatusEnum status) {
         return mapper.selectList(
-                new LambdaQueryWrapper<WaylineJobEntity>()
-                        .eq(WaylineJobEntity::getWorkspaceId, workspaceId)
-                        .eq(Objects.nonNull(status), WaylineJobEntity::getStatus, status.getVal())
-                        .in(!CollectionUtils.isEmpty(jobIds), WaylineJobEntity::getJobId, jobIds))
+                        new LambdaQueryWrapper<WaylineJobEntity>()
+                                .eq(WaylineJobEntity::getWorkspaceId, workspaceId)
+                                .eq(Objects.nonNull(status), WaylineJobEntity::getStatus, status.getVal())
+                                .in(!CollectionUtils.isEmpty(jobIds), WaylineJobEntity::getJobId, jobIds))
                 .stream()
                 .map(this::entity2Dto)
                 .collect(Collectors.toList());
@@ -415,8 +500,7 @@
 //
 //                        .orderByDesc(WaylineJobEntity::getId));
 
-        Page<WaylineJobEntity> pageData = mapper.getPage(new Page<WaylineJobEntity>(page, pageSize),waylineJobQueryParam,workspaceId);
-
+        Page<WaylineJobEntity> pageData = mapper.getPage(new Page<WaylineJobEntity>(page, pageSize), waylineJobQueryParam, workspaceId);
 
 
         List<WaylineJobDTO> records = pageData.getRecords()
@@ -432,7 +516,8 @@
     @ServiceActivator(inputChannel = ChannelName.INBOUND_REQUESTS_FLIGHT_TASK_RESOURCE_GET, outputChannel = ChannelName.OUTBOUND)
     @Transactional(isolation = Isolation.READ_UNCOMMITTED)
     public void flightTaskResourceGet(CommonTopicReceiver receiver, MessageHeaders headers) {
-        Map<String, String> jobIdMap = objectMapper.convertValue(receiver.getData(), new TypeReference<Map<String, String>>() {});
+        Map<String, String> jobIdMap = objectMapper.convertValue(receiver.getData(), new TypeReference<Map<String, String>>() {
+        });
         String jobId = jobIdMap.get(MapKeyConst.FLIGHT_ID);
 
         CommonTopicResponse.CommonTopicResponseBuilder<RequestsReply> builder = CommonTopicResponse.<RequestsReply>builder()
@@ -588,6 +673,22 @@
         return WaylineJobStatusEnum.UNKNOWN;
     }
 
+    @Override
+    public WaylineJobEntity getLatestJob(String workspaceId, WaylineJobQueryParam waylineJobQueryParam) {
+//        List<WaylineJobEntity> waylineJobEntities = mapper.selectList(new LambdaQueryWrapper<>(WaylineJobEntity.class)
+//                .eq(WaylineJobEntity::getWorkspaceId, workspaceId)
+//                .eq(WaylineJobEntity::getDockSn, waylineJobQueryParam.getDockSn())
+//                //获取状态为待执行
+//                        .eq(WaylineJobEntity::getStatus,1)
+//                .orderByDesc(WaylineJobEntity::getBeginTime)
+//        );
+
+        WaylineJobEntity waylineJobEntity = mapper.getLatest(workspaceId,waylineJobQueryParam);
+
+
+       return waylineJobEntity;
+    }
+
     private void pauseJob(String workspaceId, String dockSn, String jobId, WaylineJobStatusEnum statusEnum) {
         if (WaylineJobStatusEnum.PAUSED == statusEnum && jobId.equals(waylineRedisService.getPausedWaylineJobId(dockSn))) {
             waylineRedisService.setPausedWaylineJob(dockSn, jobId);
@@ -643,7 +744,7 @@
                 .workspaceId(entity.getWorkspaceId())
                 .status(WaylineJobStatusEnum.IN_PROGRESS.getVal() == entity.getStatus() &&
                         entity.getJobId().equals(waylineRedisService.getPausedWaylineJobId(entity.getDockSn())) ?
-                                WaylineJobStatusEnum.PAUSED.getVal() : entity.getStatus())
+                        WaylineJobStatusEnum.PAUSED.getVal() : entity.getStatus())
                 .code(entity.getErrorCode())
                 .beginTime(LocalDateTime.ofInstant(Instant.ofEpochMilli(entity.getBeginTime()), ZoneId.systemDefault()))
                 .endTime(Objects.nonNull(entity.getEndTime()) ?
@@ -679,7 +780,7 @@
         Object mediaFileCount = RedisOpsUtils.hashGet(countKey, entity.getJobId());
         if (Objects.nonNull(mediaFileCount)) {
             builder.uploadedCount(((MediaFileCountDTO) mediaFileCount).getUploadedCount())
-                    .uploading(RedisOpsUtils.checkExist(key) && entity.getJobId().equals(((MediaFileCountDTO)RedisOpsUtils.get(key)).getJobId()));
+                    .uploading(RedisOpsUtils.checkExist(key) && entity.getJobId().equals(((MediaFileCountDTO) RedisOpsUtils.get(key)).getJobId()));
             return builder.build();
         }
 

--
Gitblit v1.9.3