From 39ffdf5754b4ebca682bd3c925f7a500e8a1bbda Mon Sep 17 00:00:00 2001
From: aix <vip_xiaobin810@163.com>
Date: Thu, 08 Aug 2024 16:56:36 +0800
Subject: [PATCH] 国土调查云身份凭证token

---
 src/main/java/com/dji/sample/wayline/service/impl/WaylineJobServiceImpl.java |   60 ++++++++++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 54 insertions(+), 6 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 f55eefa..459ddb0 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
@@ -15,15 +15,11 @@
 import com.dji.sample.common.model.PaginationData;
 import com.dji.sample.common.model.ResponseResult;
 import com.dji.sample.common.util.MinioUrlUtils;
-import com.dji.sample.common.util.SpringBeanUtils;
 import com.dji.sample.component.mqtt.model.*;
 import com.dji.sample.component.mqtt.service.IMessageSenderService;
 import com.dji.sample.component.redis.RedisConst;
 import com.dji.sample.component.redis.RedisOpsUtils;
 import com.dji.sample.control.model.dto.PointDTO;
-import com.dji.sample.control.model.enums.CameraModeEnum;
-import com.dji.sample.control.model.enums.DroneAuthorityEnum;
-import com.dji.sample.control.model.enums.PayloadCommandsEnum;
 import com.dji.sample.control.model.param.*;
 import com.dji.sample.control.service.IControlService;
 import com.dji.sample.control.service.IDrcService;
@@ -53,6 +49,7 @@
 import com.dji.sample.wayline.service.IWaylineJobBreakPointService;
 import com.dji.sample.wayline.service.IWaylineJobService;
 import com.dji.sample.wayline.service.IWaylineRedisService;
+import com.dji.sample.wayline.util.ErrorCodeUtil;
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import lombok.extern.slf4j.Slf4j;
@@ -66,7 +63,6 @@
 import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
 
-import java.io.File;
 import java.net.URL;
 import java.sql.SQLException;
 import java.time.*;
@@ -147,6 +143,7 @@
                 .waylineType(param.getWaylineType().getVal())
                 .outOfControlAction(param.getOutOfControlAction())
                 .rthAltitude(param.getRthAltitude())
+                .rthMode(param.getRthMode())
                 .mediaCount(0)
                 .build();
         if (StringUtils.hasText(param.getJobId())) {
@@ -170,6 +167,11 @@
         jobEntity.setParentId(parentId);
 
         return this.insertWaylineJob(jobEntity);
+    }
+    @Override
+    public String getWaylineId(String jobId){
+        WaylineJobEntity waylineJob=mapper.selectOne(new LambdaQueryWrapper<WaylineJobEntity>().select(WaylineJobEntity::getFileId).eq(WaylineJobEntity::getJobId,jobId));
+        return waylineJob.getFileId();
     }
 
     /**
@@ -366,6 +368,7 @@
                 .taskType(waylineJob.getTaskType())
                 .waylineType(waylineJob.getWaylineType())
                 .rthAltitude(waylineJob.getRthAltitude())
+                .rthMode(waylineJob.getRthMode())
                 .outOfControlAction(waylineJob.getOutOfControlAction())
                 .file(WaylineTaskFileDTO.builder()
                         .url(MinioUrlUtils.getUrl(url))
@@ -664,7 +667,6 @@
     }
 
 
-
     @Override
     public void uploadMediaHighestPriority(String workspaceId, String jobId) {
         Optional<WaylineJobDTO> jobOpt = getJobByJobId(workspaceId, jobId);
@@ -742,6 +744,15 @@
 
     }
 
+    @Override
+    public List<String> getJobNamesByPartialName(String partialJobName) {
+        LambdaQueryWrapper<WaylineJobEntity> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.like(WaylineJobEntity::getName, partialJobName);
+        return mapper.selectList(queryWrapper).stream()
+                .map(WaylineJobEntity::getName)
+                .collect(Collectors.toList());
+    }
+
     public WaylineJobStatusEnum getWaylineState(String dockSn) {
         Optional<DeviceDTO> dockOpt = deviceRedisService.getDeviceOnline(dockSn);
         if (dockOpt.isEmpty() || !StringUtils.hasText(dockOpt.get().getChildDeviceSn())) {
@@ -766,6 +777,41 @@
         }
         return WaylineJobStatusEnum.UNKNOWN;
     }
+
+    @Override
+    public String getName(String jobId) {
+        WaylineJobEntity entity = mapper.selectOne(
+                new LambdaQueryWrapper<WaylineJobEntity>()
+                        .select(WaylineJobEntity::getName)
+                        .eq(WaylineJobEntity::getJobId, jobId)
+        );
+        return entity != null ? entity.getName() : null;
+    }
+
+    @Override
+    public List<String> getJobIds(String jobNames) {
+        List<String> names = getJobNamesByPartialName(jobNames);
+
+        // 如果 names 列表为空,直接返回空的 jobIdStringList
+        if (names.isEmpty()) {
+            return new ArrayList<>();
+        }
+
+        // 使用 LambdaQueryWrapper 进行查询
+        List<Object> jobIdList = mapper.selectObjs(
+                new LambdaQueryWrapper<WaylineJobEntity>()
+                        .select(WaylineJobEntity::getJobId)
+                        .in(WaylineJobEntity::getName, names) // 使用 IN 子句
+        );
+
+        // 将 Object 类型的 jobId 转换为 String 类型,并放入 List<String> 中
+        List<String> jobIdStringList = new ArrayList<>();
+        for (Object jobIdObj : jobIdList) {
+            jobIdStringList.add(String.valueOf(jobIdObj));
+        }
+        return jobIdStringList;
+    }
+
 
     @Override
     public WaylineJobEntity getLatestJob(String workspaceId, WaylineJobQueryParam waylineJobQueryParam) {
@@ -1041,6 +1087,7 @@
 
         WaylineJobDTO.WaylineJobDTOBuilder builder = WaylineJobDTO.builder()
                 .jobId(entity.getJobId())
+                .reason(ErrorCodeUtil.codeToReason(entity.getErrorCode()))
                 .jobName(entity.getName())
                 .fileId(entity.getFileId())
                 .fileName(waylineFileService.getWaylineByWaylineId(entity.getWorkspaceId(), entity.getFileId())
@@ -1064,6 +1111,7 @@
                 .taskType(WaylineTaskTypeEnum.find(entity.getTaskType()))
                 .waylineType(WaylineTemplateTypeEnum.find(entity.getWaylineType()))
                 .rthAltitude(entity.getRthAltitude())
+                .rthMode(entity.getRthMode())
                 .outOfControlAction(entity.getOutOfControlAction())
                 .mediaCount(entity.getMediaCount())
                 .hasChildren(entity.getHasChildren());

--
Gitblit v1.9.3