From 8d9a2d656e4ae007590c622e5f7c228adacdca49 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Fri, 14 Jun 2024 10:11:36 +0800
Subject: [PATCH] 统一风格
---
src/main/java/com/dji/sample/patches/utils/TimerUtil.java | 418 +++++++++++++++++++++++++++++++++++++++++++++++++++--------
1 files changed, 361 insertions(+), 57 deletions(-)
diff --git a/src/main/java/com/dji/sample/patches/utils/TimerUtil.java b/src/main/java/com/dji/sample/patches/utils/TimerUtil.java
index 582cb18..4ec1d62 100644
--- a/src/main/java/com/dji/sample/patches/utils/TimerUtil.java
+++ b/src/main/java/com/dji/sample/patches/utils/TimerUtil.java
@@ -1,30 +1,61 @@
package com.dji.sample.patches.utils;
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;
import com.dji.sample.patches.dao.ShpToDataSourceMapper;
import com.dji.sample.patches.model.entity.LotInfo;
+import com.dji.sample.patches.service.GetPatchesService;
import com.dji.sample.patches.xml.mode.XMLTemplateModel;
import com.dji.sample.patches.xml.utils.CreateWaylineFileUtils;
-import com.dji.sample.wayline.model.dto.WaylineFileDTO;
+import com.dji.sample.territory.pojo.TerritoryConfigPojo;
+import com.dji.sample.territory.service.ITbFJService;
import com.dji.sample.wayline.model.entity.WaylineFileEntity;
-import com.dji.sample.wayline.model.enums.WaylineTaskTypeEnum;
import com.dji.sample.wayline.model.param.CreateJobParam;
-import com.dji.sample.wayline.service.IWayLineTaskService;
import com.dji.sample.wayline.service.IWaylineFileService;
import com.dji.sample.wayline.service.IWaylineJobService;
-import org.locationtech.jts.geom.Coordinate;
+
+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 java.io.File;
-import java.io.IOException;
+import org.springframework.web.util.UriComponentsBuilder;
+
+import javax.xml.transform.Result;
+import java.nio.file.StandardCopyOption;
+import java.time.LocalDateTime;
+
+import java.io.*;
+import java.net.HttpURLConnection;
+import java.net.URL;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+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.UUID;
+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;
import static com.dji.sample.patches.utils.ZipUtil.zipFolder;
@@ -40,44 +71,119 @@
@Autowired
public IWaylineFileService waylineFileService;
@Autowired
+ private IFileMapper fileMapper;
+ @Autowired
+ private ITbFJService tbFJService;
+ @Autowired
+ private GetPatchesService getPatchesService;
+ @Autowired
private IWaylineJobService waylineJobService;
@Autowired
- private IWayLineTaskService wayLineTaskService;
+ private TerritoryConfigPojo territoryConfigPojo;
+
/**
- * 定时
+ * 定时器，将没有规划的图斑生成航线，并将航线飞完后的成果数据进行保存推送
+ *
* @throws IOException
* @throws SQLException
*/
- @Scheduled(cron = "0 25 14 * * ?")
- public void myTask() throws IOException, SQLException {
- //获取未规划的图斑集合
- List<LotInfo> list = getNoPlan();
- long timestamp = System.currentTimeMillis();
- String waylineName = "云飞行调查" + timestamp;
- String workspaceId ="4a574db8-4ad3-48f7-9f16-3edbcd8056e1";
- //根据获取的图斑集合获得kmz航线文件
- MultipartFile multipartFile = getFile(waylineName, list);
- //上传航线文件
- String waylineId=backWayline(multipartFile, waylineName, workspaceId, backclaim().getUsername());
- //将为规划的图斑状态更新为已规划
- updateStatu(list);
- //调用publshFlightTask方法创建定时任务
- waylineJobService.publishFlightTask(JobParam(waylineId), backclaim());
+ @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 = 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>> medias = convertToLists(list);
+ for (List<MediaFileEntity> media : medias) {
+ //清空FJ表
+ tbFJService.deleteData();
+ 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);
+ }
+ //存储db到服务器
+ dbSave(territoryConfigPojo.getResult(), territoryConfigPojo.getResultsave(), taskId);
+
+ //发送请求给第三方接口
+ sendPostWithFileAndParameter(territoryConfigPojo.getResult(), 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.218512, 115.856725497);
+ 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());
- String destKMZFile = patchesConfigPojo.getDestKMZFile() + waylineName + ".kmz";
+ // 压缩文件夹中的内容
+ String destKMZFile = patchesConfigPojo.getDestKMZFile() + waylineName + ".kmz"; // 输出的KMZ文件路径
zipFolder(patchesConfigPojo.getSourceDir(), destKMZFile);
- MultipartFile multipartFile = convert(new File(destKMZFile));
- return multipartFile;
+ return convert(new File(destKMZFile));
}
- public String backWayline(MultipartFile multipartFile, String waylineName, String workspaceId,String username) {
- waylineFileService.importKmzFileBack(multipartFile, workspaceId,username);
+ /**
+ * 将航线上传Oss
+ *
+ * @param multipartFile
+ * @param waylineName
+ * @param workspaceId
+ * @param username
+ * @return
+ */
+ public String backWayline(MultipartFile multipartFile, String waylineName, String workspaceId, String username) {
+ waylineFileService.importKmzFileBack(multipartFile, workspaceId, username);
WaylineFileEntity entity = waylineFileService.selectByName(waylineName);
try {
waylineFileService.getObjectUrl(workspaceId, entity.getWaylineId());
@@ -87,45 +193,243 @@
return entity.getWaylineId();
}
- public List<LotInfo> getNoPlan() {
- return shpToDataSourceMapper.selectList(new LambdaQueryWrapper<LotInfo>().eq(LotInfo::getIsplan, 0));
+ /**
+ * 获取未规划的图斑集合
+ *
+ * @return
+ */
+ public List<List<LotInfo>> getNoPlan() {
+ List<LotInfo> list = shpToDataSourceMapper.selectList(new LambdaQueryWrapper<LotInfo>().eq(LotInfo::getIsPlan, 0));
+ List<List<LotInfo>> combinedTasks = list.stream()
+ .collect(Collectors.groupingBy(LotInfo::getTaskId))
+ .values().stream()
+ .map(ArrayList::new)
+ .collect(Collectors.toList());
+ return combinedTasks;
}
- public void updateStatu(List<LotInfo> list) {
+ /**
+ * 更新图斑执行状态
+ *
+ * @param list
+ */
+ public void updatePatchesStatu(List<LotInfo> list) {
for (LotInfo lotInfo : list) {
- lotInfo.setIsplan(1);
+ lotInfo.setIsPlan(1);
shpToDataSourceMapper.updateById(lotInfo);
}
}
- public CreateJobParam JobParam(String waylineId) {
- List<List<Long>> listOfLists = new ArrayList<>();
- List<Long> sublist = new ArrayList<>();
- sublist.add(1714944600L); // 添加整数值
- listOfLists.add(sublist);
- List<Long> list = new ArrayList<>();
- list.add(1714944600L);
- CreateJobParam param = new CreateJobParam();
- param.setName("云飞行调查");
- param.setTaskType(TIMED);
- param.setOutOfControlAction(0);
- param.setRepFreType(3);
- param.setRepFreVal(1);
- param.setRepRuleType(1);
- param.setExecuteStartTimeArr(listOfLists);
- param.setRthAltitude(80);
- param.setWaylineType(WAYPOINT);
- param.setTaskPeriods(listOfLists);
- param.setTaskDays(list);
- param.setFileId(waylineId);
- param.setDockSn("4TADKCMB010016");
- return param;
+
+ 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)
+ .eq(MediaFileEntity::getExamine, 1)
+ .like(MediaFileEntity::getFileName, "~"));
+ List<List<MediaFileEntity>> groupedFilenames = groupTasks(list);
+ return groupedFilenames;
+ }
+
public CustomClaim backclaim() {
- CustomClaim claim=new CustomClaim();
+ 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(1715941980L); // 添加整数值
+ listOfLists.add(sublist);
+ List<Long> list = new ArrayList<>();
+ list.add(1715941980L);
+ CreateJobParam param = new CreateJobParam();
+ param.setName("云飞行调查" + timestamp);
+ param.setExecuteStartTimeArr(listOfLists);
+ param.setTaskType(TIMED);
+ param.setOutOfControlAction(0);
+ param.setRepFreType(3);
+ param.setRepFreVal(1);
+ param.setRepRuleType(1);
+ param.setRthAltitude(80);
+ param.setWaylineType(WAYPOINT);
+ param.setTaskPeriods(listOfLists);
+ param.setTaskDays(list);
+ param.setFileId(waylineId);
+ param.setDockSn("4TADKCMB0010016");
+ return param;
+ }
+
+ public static <T> List<List<T>> convertToLists(List<List<T>> listOfLists) {
+ List<List<T>> convertedLists = new ArrayList<>();
+ for (List<T> list : listOfLists) {
+ convertedLists.add(new ArrayList<>(list));
+ }
+ return convertedLists;
+ }
+
+
+ // 对任务ID进行分组
+ public static List<List<MediaFileEntity>> groupTasks(List<MediaFileEntity> mediaFiles) {
+ Map<String, List<MediaFileEntity>> groupedTasks = new HashMap<>();
+ for (MediaFileEntity mediaFile : mediaFiles) {
+ String taskId = getTaskId(mediaFile.getFileName());
+ if (!groupedTasks.containsKey(taskId)) {
+ groupedTasks.put(taskId, new ArrayList<>());
+ }
+ groupedTasks.get(taskId).add(mediaFile);
+ }
+ return new ArrayList<>(groupedTasks.values());
+ }
+
+ // 从filename字段中提取任务ID（任务ID在文件名中以"~"开头、"."结尾）
+ public static String getTaskId(String filename) {
+ // 这里只是一个示例，你需要根据实际情况编写提取task_id的逻辑
+ int startIndex = filename.indexOf("~") + 1;
+ int endIndex = filename.indexOf(".");
+ return filename.substring(startIndex, endIndex);
+ }
+
+ public static String getDkbh(String filename) {
+ // 这里只是一个示例，你需要根据实际情况编写提取task_id的逻辑
+ int startIndex = filename.indexOf("点") + 1;
+ int endIndex = filename.indexOf("~");
+ return filename.substring(startIndex, endIndex);
+ }
+
+ /**
+ * 将成果发送post请求到第三方接口
+ *
+ * @param filePath
+ * @param taskId
+ * @throws IOException
+ */
+ public static void sendPostWithFileAndParameter(String filePath, String taskId) throws IOException {
+ // 创建 RestTemplate 实例
+ try {
+ RestTemplate restTemplate = new RestTemplate();
+
+ // 读取文件内容为字节数组
+ byte[] fileContent = readFileToBytes(filePath);
+
+ // 构建请求体
+ MultiValueMap<String, Object> body = buildRequestBody(taskId, fileContent, filePath);
+
+ // 设置请求头
+ 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);
+
+ // 发送请求
+ 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);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ System.out.println(response);
+ } catch (Exception e) {
+ throw new IllegalArgumentException("db推送失败" + e.getMessage());
+ }
+ }
+
+ private static MultiValueMap<String, Object> buildRequestBody(String taskId, byte[] fileContent, String filePath) {
+ MultiValueMap<String, Object> body = new LinkedMultiValueMap<>();
+ body.add("taskId", taskId);
+ body.add("file", new ByteArrayResource(fileContent) {
+ @Override
+ public String getFilename() {
+ return Paths.get(filePath).getFileName().toString();
+ }
+ });
+ return body;
+ }
+
+ private static byte[] readFileToBytes(String filePath) throws IOException {
+ Path path = Paths.get(filePath);
+ 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;
+ }
+
}
--
Gitblit v1.9.3