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


@@ -3,6 +3,7 @@


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;


import com.dji.sample.common.model.CustomClaim;


import com.dji.sample.common.model.ResponseResult;


import com.dji.sample.media.dao.IFileMapper;


import com.dji.sample.media.model.MediaFileEntity;


import com.dji.sample.patches.config.pojo.PatchesConfigPojo;


@@ -18,16 +19,24 @@


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


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




import com.fasterxml.jackson.core.JsonProcessingException;


import com.fasterxml.jackson.databind.JsonNode;


import com.fasterxml.jackson.databind.ObjectMapper;


import com.fasterxml.jackson.databind.node.ObjectNode;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.core.io.ByteArrayResource;


import org.springframework.http.*;


import org.springframework.scheduling.annotation.Scheduled;


import org.springframework.stereotype.Component;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.util.LinkedMultiValueMap;


import org.springframework.util.MultiValueMap;


import org.springframework.web.client.RestTemplate;


import org.springframework.web.multipart.MultipartFile;


import org.springframework.web.util.UriComponentsBuilder;




import javax.xml.transform.Result;


import java.nio.file.StandardCopyOption;


import java.time.LocalDateTime;




import java.io.*;


@@ -38,10 +47,14 @@


import java.nio.file.Path;


import java.nio.file.Paths;


import java.sql.SQLException;


import java.time.format.DateTimeFormatter;


import java.util.ArrayList;


import java.util.HashMap;


import java.util.List;


import java.util.Map;


import java.util.concurrent.ExecutorService;


import java.util.concurrent.Executors;


import java.util.concurrent.Future;


import java.util.stream.Collectors;




import static com.dji.sample.patches.utils.MultipartFileTOFileUtil.convert;


@@ -68,45 +81,88 @@


    @Autowired


    private TerritoryConfigPojo territoryConfigPojo;






    /**


     * 定时


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


     *


     * @throws IOException


     * @throws SQLException


     */


    @Scheduled(cron = "45 1 16 * * ?")


    @Scheduled(cron = "0 0 0 * * ?")


    public void myTask() throws Exception {


        String taskId = "";


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


        //获取未规划的图斑集合


//        List<List<LotInfo>> lists = getNoPlan();


//        List<List<LotInfo>> convertedLists = convertToLists(lists);


//        for (List<LotInfo> list : convertedLists) {


//            long timestamp = System.currentTimeMillis();


//            String waylineName = "云飞行调查" + timestamp;


//            //根据获取的图斑集合获得kmz航线文件


//            MultipartFile multipartFile = getFile(waylineName, list);


//            //上传航线文件


//            String waylineId = backWayline(multipartFile, waylineName, workspaceId, backclaim().getUsername());


//            //将为规划的图斑状态更新为已规划


//            updatePatchesStatu(list);


//            //调用publshFlightTask方法创建定时任务


//            waylineJobService.publishFlightTask(JobParam(waylineId), backclaim());


//        }


//        获取未规划的图斑集合


        List<List<LotInfo>> lists = getNoPlan();


        List<List<LotInfo>> convertedLists = convertToLists(lists);


        for (List<LotInfo> list : convertedLists) {


            long timestamp = System.currentTimeMillis();


            String waylineName = "云飞行调查" + timestamp;


            //根据获取的图斑集合获得kmz航线文件


            MultipartFile multipartFile = null;


            try {


                multipartFile = getFile(waylineName, list);


            } catch (IOException e) {


                throw new RuntimeException(e);


            }


            //上传航线文件


            backWayline(multipartFile, waylineName, workspaceId, backclaim().getUsername());


            //将为规划的图斑状态更新为已规划


            updatePatchesStatu(list);


//            waylineJobService.publishFlightTask(JobParam("656c62f4-cb86-4911-930d-b38089829348"),backclaim());


        }




    }




    @Scheduled(cron = "0 0 0 * * ?")


    public void myTask2() {


        ExecutorService executor = Executors.newSingleThreadExecutor();


        executor.execute(() -> {


            try {


                dbOperation();


            } catch (Exception e) {


                throw new RuntimeException(e);


            }


        });


        executor.shutdown();


    }




    @Transactional


    public void dbOperation() throws Exception {


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


        String taskId = "";


        List<List<MediaFileEntity>> list = getNoadd();


        List<List<MediaFileEntity>> media = convertToLists(list);


        for (int i = 0; i < media.size(); i++) {


        List<List<MediaFileEntity>> medias = convertToLists(list);


        for (List<MediaFileEntity> media : medias) {


            //清空FJ表


            tbFJService.deleteData();


            String dkbh = getDkbh(media.get(i).get(i).getFileName());


            String taskId = getTaskId(media.get(i).get(i).getFileName());


            LotInfo lotInfo = getPatchesService.getLotinfo(dkbh, workspaceId);


            tbFJService.insertData(media.get(i), lotInfo);


            updateMediaStatu(media.get(i));


            for (MediaFileEntity mediaFile : media) {


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


                String dkbh = getDkbh(mediaFile.getFileName());


                taskId = getTaskId(mediaFile.getFileName());


                //获取该媒体文件的图斑信息


                LotInfo lotInfo = getPatchesService.getLotinfo(dkbh, workspaceId);


                tbFJService.insertOneData(mediaFile, lotInfo);


                //更新媒体文件的是否添加状态


                updateMediaStatu(mediaFile);


            }


            //发送请求给第三方接口


            sendPostWithFileAndParameter(territoryConfigPojo.getResult(), taskId);


            //存储db到服务器


            dbSave(territoryConfigPojo.getResult(), territoryConfigPojo.getResultsave(), taskId);


        }


    }




    /**


     * 创建航线


     *


     * @param waylineName


     * @param list


     * @return


     * @throws IOException


     */


    public MultipartFile getFile(String waylineName, List<LotInfo> list) throws IOException {


        List<PointPO> coordinates = GeoToolsUtil.getRoutePointOrder(list, 28.62703, 115.867719);


        List<PointPO> coordinates = GeoToolsUtil.getRoutePointOrder(list, 28.62452712442823, 115.85666327144976);


        XMLTemplateModel xmlModel = XMLTemplateModel.init(coordinates, list);


        CreateWaylineFileUtils.createWaylineFile(xmlModel, patchesConfigPojo.getTemplate(), patchesConfigPojo.getTargetTemplate(), patchesConfigPojo.getWaylines(), patchesConfigPojo.getTargetWaylines());


        // 压缩文件夹中的内容


@@ -115,8 +171,17 @@


        return convert(new File(destKMZFile));


    }




    /**


     * 将航线上传Oss


     *


     * @param multipartFile


     * @param waylineName


     * @param workspaceId


     * @param username


     * @return


     */


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


        waylineFileService.importKmzFile(multipartFile, workspaceId, username);


        waylineFileService.importKmzFileBack(multipartFile, workspaceId, username);


        WaylineFileEntity entity = waylineFileService.selectByName(waylineName);


        try {


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


@@ -153,29 +218,40 @@


        }


    }




    public void updateMediaStatu(List<MediaFileEntity> list) {


        for (MediaFileEntity mediaFile : list) {


            mediaFile.setIsadd(1);


            fileMapper.updateById(mediaFile);


        }


    public void updateMediaStatu(MediaFileEntity mediaFile) {


        mediaFile.setIsadd(1);


        fileMapper.updateById(mediaFile);


    }




    public List<List<MediaFileEntity>> getNoadd() {


        List<MediaFileEntity> list = fileMapper.selectList(new LambdaQueryWrapper<MediaFileEntity>()


                .eq(MediaFileEntity::getIsadd, 0).like(MediaFileEntity::getFileName, "~"));


                .eq(MediaFileEntity::getIsadd, 0)


                .eq(MediaFileEntity::getExamine, 1)


                .like(MediaFileEntity::getFileName, "~"));


        List<List<MediaFileEntity>> groupedFilenames = groupTasks(list);


        return groupedFilenames;


    }




    public CreateJobParam JobParam(String waylineId) {


    public CustomClaim backclaim() {


        CustomClaim claim = new CustomClaim();


        claim.setId("1");


        claim.setUsername("adminPC");


        claim.setWorkspaceId("4a574db8-4ad3-48f7-9f16-3edbcd8056e1");


        claim.setUserType(1);


        return claim;


    }




    public static CreateJobParam JobParam(String waylineId) {


        long timestamp = System.currentTimeMillis();


        List<List<Long>> listOfLists = new ArrayList<>();


        List<Long> sublist = new ArrayList<>();


        sublist.add(177271980L); // 添加整数值


        sublist.add(1715941980L); // 添加整数值


        listOfLists.add(sublist);


        List<Long> list = new ArrayList<>();


        list.add(177271980L);


        list.add(1715941980L);


        CreateJobParam param = new CreateJobParam();


        param.setName("云飞行调查");


        param.setName("云飞行调查" + timestamp);


        param.setExecuteStartTimeArr(listOfLists);


        param.setTaskType(TIMED);


        param.setOutOfControlAction(0);


        param.setRepFreType(3);


@@ -186,17 +262,8 @@


        param.setTaskPeriods(listOfLists);


        param.setTaskDays(list);


        param.setFileId(waylineId);


        param.setDockSn("4TADKCMB010016");


        param.setDockSn("4TADKCMB0010016");


        return param;


    }




    public CustomClaim backclaim() {


        CustomClaim claim = new CustomClaim();


        claim.setId("1");


        claim.setUsername("adminPC");


        claim.setWorkspaceId("4a574db8-4ad3-48f7-9f16-3edbcd8056e1");


        claim.setUserType(1);


        return claim;


    }




    public static <T> List<List<T>> convertToLists(List<List<T>> listOfLists) {


@@ -221,7 +288,7 @@


        return new ArrayList<>(groupedTasks.values());


    }




    // 从filename字段中提取任务ID（假设任务ID在文件名中以"~"开头、"."结尾）


    // 从filename字段中提取任务ID（任务ID在文件名中以"~"开头、"."结尾）


    public static String getTaskId(String filename) {


        // 这里只是一个示例，你需要根据实际情况编写提取task_id的逻辑


        int startIndex = filename.indexOf("~") + 1;


@@ -236,33 +303,43 @@


        return filename.substring(startIndex, endIndex);


    }




    /**


     * 将成果发送post请求到第三方接口


     *


     * @param filePath


     * @param taskId


     * @throws IOException


     */


    public static void sendPostWithFileAndParameter(String filePath, String taskId) throws IOException {


        // 创建 RestTemplate 实例


        RestTemplate restTemplate = new RestTemplate();


        try {


            RestTemplate restTemplate = new RestTemplate();




        // 读取文件内容为字节数组


        byte[] fileContent = readFileToBytes(filePath);


            // 读取文件内容为字节数组


            byte[] fileContent = readFileToBytes(filePath);




        // 构建请求体


        MultiValueMap<String, Object> body = buildRequestBody(taskId, fileContent, filePath);


            // 构建请求体


            MultiValueMap<String, Object> body = buildRequestBody(taskId, fileContent, filePath);




        // 设置请求头


        HttpHeaders headers = new HttpHeaders();


        headers.setContentType(MediaType.MULTIPART_FORM_DATA);


            // 设置请求头


            HttpHeaders headers = new HttpHeaders();


            headers.setContentType(MediaType.MULTIPART_FORM_DATA);


            headers.set("X-SHARE-TOKEN", "981D9B822A9A09DE28B66243AA330DC666E1A82826B58EC3FE63FEB1EC8AA190");


            headers.set("X-SHARE-AK", "gwc952d483-bfe5-476b-8431-ae15fe94c006");




            // 构建请求实体


            HttpEntity<MultiValueMap<String, Object>> requestEntity = new HttpEntity<>(body, headers);




        // 构建请求实体


        HttpEntity<MultiValueMap<String, Object>> requestEntity = new HttpEntity<>(body, headers);




        // 发送请求


        ResponseEntity<String> response = restTemplate.exchange(


                "http://localhost:6789/territory/tbdkjbxx/upload",


                HttpMethod.POST,


                requestEntity,


                String.class);




        // 输出响应结果


        System.out.println("Response: " + response.getBody());


            // 发送请求


            ResponseEntity<String> response =restTemplate.exchange(


                    "https://jxlandcloud.org.cn/landCloudWork/artifact/media/upload.action",


//                  "http://localhost:6789/territory/tbdkjbxx/upload",


                    HttpMethod.POST,


                    requestEntity,


                    String.class);


        } catch (Exception e) {


            throw new IllegalArgumentException("db推送失败" + e.getMessage());


        }


    }




    private static MultiValueMap<String, Object> buildRequestBody(String taskId, byte[] fileContent, String filePath) {


@@ -282,4 +359,60 @@


        return Files.readAllBytes(path);


    }




    /**


     * 将上传的db保存一份到本地


     *


     * @param path


     * @param folder


     * @param taskId


     */


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


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


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




        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 targetFilePath = targetFolderPath + File.separator + fileName;




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


            Path sourcePath = Paths.get(sourceFilePath);


            Path targetPath = Paths.get(targetFilePath);


            Files.copy(sourcePath, targetPath, StandardCopyOption.REPLACE_EXISTING);


        } catch (IOException e) {


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


        }


    }




    private static File fileFold(String fold) {


        // 获取当前时间


        LocalDateTime currentTime = LocalDateTime.now();




        // 格式化时间，生成文件夹名字


        String folderName = currentTime.format(DateTimeFormatter.ofPattern("yyyyMMdd"));




        // 拼接文件夹路径


        String folderPath = fold + File.separator + folderName;




        // 创建文件夹


        File folder = new File(folderPath);


        if (!folder.exists()) {


            if (!folder.mkdirs()) {


                throw new RuntimeException("文件创建失败" + folderPath);


            }


        }


        return folder;


    }




}