无人机java后台代码
blame | history | patch | commit | commitdiff | ignore whitespace
aix
2024-08-08 39ffdf5754b4ebca682bd3c925f7a500e8a1bbda 
 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());