无人机java后台代码
blame | history | patch | commit | commitdiff | ignore whitespace
rain
2024-08-20 74d1c21ebb6b3b916904d95d13d289df23dcdedd 
 src/main/java/com/dji/sample/patches/utils/TimerUtil.java


@@ -2,6 +2,8 @@




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;


@@ -15,9 +17,13 @@


import com.dji.sample.patches.xml.utils.CreateWaylineFileUtils;


import com.dji.sample.territory.pojo.TerritoryConfigPojo;


import com.dji.sample.territory.service.ITbFJService;


import com.dji.sample.wayline.dao.IWaylineJobBreakPointMapper;


import com.dji.sample.wayline.dao.IWaylineJobMapper;


import com.dji.sample.wayline.model.dto.WaylineFileDTO;


import com.dji.sample.wayline.model.dto.WaylineJobDTO;


import com.dji.sample.wayline.model.entity.WaylineFileEntity;


import com.dji.sample.wayline.model.entity.WaylineJobBreakPointEntity;


import com.dji.sample.wayline.model.entity.WaylineJobEntity;


import com.dji.sample.wayline.model.param.CreateJobParam;


import com.dji.sample.wayline.service.IWaylineFileService;


import com.dji.sample.wayline.service.IWaylineJobBreakPointService;


@@ -37,16 +43,13 @@


import org.springframework.web.multipart.MultipartFile;




import java.nio.file.StandardCopyOption;


import java.time.Instant;


import java.time.LocalDateTime;


import java.time.*;




import java.io.*;


import java.nio.file.Files;


import java.nio.file.Path;


import java.nio.file.Paths;


import java.sql.SQLException;


import java.time.ZoneId;


import java.time.ZonedDateTime;


import java.time.format.DateTimeFormatter;


import java.util.*;


import java.util.concurrent.ExecutorService;


@@ -55,6 +58,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;




@@ -66,7 +70,8 @@


    private PatchesConfigPojo patchesConfigPojo;


    @Autowired


    public IWaylineFileService waylineFileService;




    @Autowired


    private IWaylineJobService waylineJobService;


    @Autowired


    private ITbFJService tbFJService;


    @Autowired


@@ -75,13 +80,19 @@


    @Autowired


    private GetPatchesMapper patchesMapper;


    @Autowired


    private IWaylineJobService waylineJobService;


    private IWaylineJobMapper waylineJobMapper;




    @Autowired


    private IWaylineJobBreakPointMapper jobBreakPointMapper;


    @Autowired


    private TerritoryConfigPojo territoryConfigPojo;


    @Autowired


    private IWaylineJobBreakPointService jobBreakPointService;


    @Autowired


    private IDeviceMapper deviceMapper;


    @Autowired


    private DbUploadMapper dbUploadMapper;




    /**


     * 定时器，将没有规划的图斑生成航线，并将航线飞完后的成果数据进行保存推送


     *


@@ -136,6 +147,16 @@


        executor.shutdown();


    }




    @Scheduled(cron = "0 45 8-18 * * ?")


    public void runTask() throws SQLException {


        WaylineJobEntity job = processWaylineJobs();


        if (job != null) {


            // 调用其他方法进行后续处理


            handleJobId(job);


        } else {


            System.out.println("未找到符合条件的对象");


        }


    }


    //    public void myTask3() throws SQLException {


//        String workspaceId = "4a574db8-4ad3-48f7-9f16-3edbcd8056e1";


//        String name = getNowDay();


@@ -252,6 +273,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);


@@ -275,30 +299,57 @@




        }


    }




    @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表


                tbFJService.deleteData();


                // 清空FJ表


                for (MediaFileEntity mediaFile : media) {


                    //获取媒体文件名里的dkbh和taskId


                    // 获取媒体文件名里的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到服务器


            dbSaveGt(territoryConfigPojo.getResult(), territoryConfigPojo.getResultGtsave(), taskId);




            // 存储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);


        }


    }






    /**


     * 创建航线


@@ -330,7 +381,7 @@


     * @return


     */


    public WaylineFileEntity backWayline(MultipartFile multipartFile, String waylineName, String workspaceId, String username) {


        waylineFileService.importKmzFileBack(multipartFile, workspaceId, username);


        waylineFileService.importKmzFileBack(multipartFile, workspaceId, username, 1);


        WaylineFileEntity entity = waylineFileService.selectByName(waylineName);


        try {


            waylineFileService.getObjectUrl(workspaceId, entity.getWaylineId());


@@ -378,15 +429,16 @@


    public List<List<LotInfo>> getNoPush(String workspaceId) {


        List<LotInfo> list = patchesMapper.selectList(new LambdaQueryWrapper<LotInfo>()


                .eq(LotInfo::getIsPush, 0)


                .eq(LotInfo::getType, 1)


                .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, 0)


                .eq(LotInfo::getType, 1)


                .eq(LotInfo::getInvestigate, 1));


        return groupTasks(list);


    }


@@ -528,13 +580,13 @@


        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 {


    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);


            String jsonBody = buildRequestBody(name, fileId, time, taskDays, taskPeriods, jobId, docksn);


            String workspaceId = getWorkspaceIdByDocksn(docksn);


            // 设置请求头


            HttpHeaders headers = new HttpHeaders();


            headers.setContentType(MediaType.APPLICATION_JSON);


@@ -549,7 +601,7 @@


            // 发送请求


            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",


                    "http://localhost:6789/wayline/api/v1/workspaces/" + workspaceId + "/flight-tasks",


                    HttpMethod.POST,


                    requestEntity,


                    String.class);


@@ -558,10 +610,11 @@


        }


    }




    public String getWorkspaceIdByDocksn(String docksn){


      DeviceEntity entity=  deviceMapper.selectOne(new LambdaQueryWrapper<DeviceEntity>().eq(DeviceEntity::getDeviceSn,docksn));


      return entity.getWorkspaceId();


    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();


@@ -691,30 +744,18 @@


            throw new RuntimeException("Db文件复制失败", e); // 捕获并抛出原始异常


        }


    }


    public void dbSaveGt(String path, String folder, String taskId) {


        String sourceFilePath = path; // 源文件路径


        String targetFolderPath = fileFold(folder,taskId).getPath(); // 目标文件夹路径




    public String dbSaveGt(String path, String folder, String taskId) {


        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 fileName = taskId + ".db";


            // 构建目标文件路径


            String targetFilePath = targetFolderPath + File.separator + fileName;


            String targetFilePath = folder + File.separator + fileName;




            // 复制文件到目标文件夹


            Path sourcePath = Paths.get(sourceFilePath);


            // 复制文件到目标位置


            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); // 捕获并抛出原始异常


        }


@@ -745,7 +786,8 @@


        }


        return folder;


    }


    private static File fileFold(String fold,String taskId) {




    private static File fileFold(String fold, String taskId) {


        // 获取当前时间


        LocalDateTime currentTime = LocalDateTime.now();




@@ -761,6 +803,7 @@


        }


        return folder;


    }




    public static Long getCurrentTimestampPlus8Hours(long hours) {


        // 获取当前时间


        LocalDateTime now = LocalDateTime.now();


@@ -776,6 +819,32 @@


        return timestamp / 1000;


    }




    public static long getNextDayEightAMTimestamp(int hour) {


        // 获取当前时间


        LocalDateTime now = LocalDateTime.now();




        // 获取第二天的日期并设置时间为08:00


        LocalDateTime nextDayEightAM = now.plusDays(1).withHour(8).withMinute(30).withSecond(0).withNano(0);




        // 根据传入的hour参数增加小时


        LocalDateTime adjustedTime = nextDayEightAM.plusHours(hour);




        // 转换为时间戳（秒级）


        ZonedDateTime zonedDateTime = adjustedTime.atZone(ZoneId.systemDefault());


        return zonedDateTime.toEpochSecond();


    }




    public static long getAdjustedTimestamp(int hour) {


        // 获取当前时间


        LocalDateTime now = LocalDateTime.now();




        // 根据传入的hour参数增加小时


        LocalDateTime adjustedTime = now.plusSeconds(hour);




        // 转换为时间戳（秒级）


        ZonedDateTime zonedDateTime = adjustedTime.atZone(ZoneId.systemDefault());


        return zonedDateTime.toEpochSecond();


    }


    public static Long addOneHourToTimestamp(Long timestampInSeconds, long hours, long min) {


        // 将输入的秒级时间戳转换为Instant对象


        Instant instant = Instant.ofEpochSecond(timestampInSeconds);


@@ -821,4 +890,59 @@


        // 格式化时间，生成当前时间


        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);




    }




    public WaylineJobEntity processWaylineJobs() {


        // 获取昨天 00:00 和 23:59:59 的时间戳


        LocalDate yesterday = LocalDate.now().minusDays(1);


        long startOfDayMillis = LocalDateTime.of(yesterday, LocalTime.MIN).toInstant(ZoneOffset.UTC).toEpochMilli();


        long endOfDayMillis = LocalDateTime.of(yesterday, LocalTime.MAX).toInstant(ZoneOffset.UTC).toEpochMilli();


        // 查询符合条件的 wayline_job_break_point 对象


        List<WaylineJobBreakPointEntity> breakPoints = jobBreakPointMapper.selectList(


                new LambdaQueryWrapper<WaylineJobBreakPointEntity>()


                        .ge(WaylineJobBreakPointEntity::getCreateTime, startOfDayMillis)


                        .le(WaylineJobBreakPointEntity::getCreateTime, endOfDayMillis)


                        .like(WaylineJobBreakPointEntity::getJobId, "~")


                        .orderByAsc(WaylineJobBreakPointEntity::getCreateTime)


        );




        for (WaylineJobBreakPointEntity breakPoint : breakPoints) {


            String jobId = breakPoint.getJobId();




            // 根据 jobId 查询 wayline_job 表


            WaylineJobEntity job = waylineJobMapper.selectOne(


                    new LambdaQueryWrapper<WaylineJobEntity>()


                            .eq(WaylineJobEntity::getJobId, jobId)


            );




            // 检查 statues 是否等于 3


            if (job != null && job.getStatus() != 3) {


                // 返回符合条件的对象


                return job;


            }


        }


        return null;


    }




    private void handleJobId(WaylineJobEntity entity) throws SQLException {


        Optional<WaylineJobDTO> waylineJobDTO = waylineJobService.getJobByJobId(entity.getWorkspaceId(), entity.getJobId(),true);


        waylineJobService.publishOneFlightTask(waylineJobDTO.get());


    }


}