forked from uav/ard-uav-server
ard-route/src/main/java/com/ard/route/service/impl/UavRouteServiceImpl.java
@@ -4,16 +4,16 @@ import cn.hutool.core.collection.CollectionUtil; import com.ard.oss.service.impl.OssServiceContext; import com.ard.route.constant.FileTypeConstants; import com.ard.route.domain.RoutePointReq; import com.ard.route.domain.UavRouteReq; import com.ard.route.domain.WaypointHeadingReq; import com.ard.route.domain.WaypointTurnReq; import com.ard.route.domain.kml.*; import com.ard.route.domain.route.ArdRouteInfor; import com.ard.route.domain.vo.KmzInfoVO; import com.ard.route.domain.djuav.RoutePointReq; import com.ard.route.domain.djuav.UavRouteReq; import com.ard.route.domain.djuav.WaypointHeadingReq; import com.ard.route.domain.djuav.WaypointTurnReq; import com.ard.route.domain.djuav.kml.*; import com.ard.route.domain.route.ArdRouteInfo; import com.ard.route.domain.vo.KmzInfoVo; import com.ard.route.domain.vo.KmzResultVo; import com.ard.route.enums.kml.ExitOnRCLostEnums; import com.ard.route.service.IArdRouteInforService; import com.ard.route.service.IArdRouteInfoService; import com.ard.route.util.InitializationUtils; import com.ard.route.util.RouteFileUtils; import com.ard.route.service.UavRouteService; @@ -49,7 +49,7 @@ private OssServiceContext ossService; @Resource private IArdRouteInforService ardRouteInformationService; private IArdRouteInfoService ardRouteInformationService; InitializationUtils init = new InitializationUtils(); @@ -60,14 +60,14 @@ public KmzResultVo updateKmz(UavRouteReq uavRouteReq,String fileName){ KmzResultVo result = new KmzResultVo(); // 校验修改的文件是否存在 ArdRouteInfor routeInformation = ardRouteInformationService.selectArdRouteInforByName(fileName); ArdRouteInfo routeInformation = ardRouteInformationService.selectArdRouteInfoByName(fileName); if ( routeInformation == null ){ result.setData("修改航线文件失败，请检查文件是否存在"); return result; } // 获取文件 String minioUrlName = subStringMinioUrlName(routeInformation.getUpPath()); String minioUrlName = subStringMinioUrlName(routeInformation.getPath()); try (InputStream inputStream = ossService.getObject(kmzPath,minioUrlName); ArchiveInputStream archiveInputStream = new ZipArchiveInputStream(inputStream)){ ArchiveEntry entry; @@ -87,7 +87,7 @@ } // 删除文件 ossService.deleteObject(kmzPath, minioUrlName); return buildKmzMethod(routeInformation.getKmzName(), kmlInfo, wpmlInfo); return buildKmzMethod(routeInformation.getName(), kmlInfo, wpmlInfo); } catch (Exception e) { e.printStackTrace(); result.setData("更新航线kmz文件失败"); @@ -178,54 +178,31 @@ String kml = XML_HEADER + xStream.toXML(kmlInfo); String wpml = XML_HEADER + xStream.toXML(wpmlInfo); KmzResultVo result = new KmzResultVo(); try (ByteArrayOutputStream baos = new ByteArrayOutputStream(); ZipOutputStream zipOutputStream = new ZipOutputStream(baos)) { zipOutputStream.setLevel(0); // 判断本地生成kmz文件 if (isLocal) { try (FileOutputStream fileOutputStream = new FileOutputStream(localFilePath + fileName + ".kmz"); ZipOutputStream zipOutputStream = new ZipOutputStream(fileOutputStream)) { zipOutputStream.setLevel(0); // 0 表示不压缩，存储方式 // 创建 wpmz 目录中的 template.kml 文件条目 buildZipFile("wpmz/template.kml", zipOutputStream, kml); // 创建 wpmz 目录中的 waylines.wpml 文件条目 buildZipFile("wpmz/waylines.wpml", zipOutputStream, wpml); // 创建KMZ文件内容 buildZipFile("wpmz/template.kml", zipOutputStream, kml); buildZipFile("wpmz/waylines.wpml", zipOutputStream, wpml); zipOutputStream.finish(); zipOutputStream.flush(); } catch (Exception e) { throw new RuntimeException(e); } result.setUpPath(localFilePath + fileName + ".kmz"); // 创建MultipartFile对象 MultipartFile multipartFile = new MockMultipartFile( fileName + ".kmz", fileName + ".kmz", "application/vnd.google-earth.kmz", new ByteArrayInputStream(baos.toByteArray()) ); String kmzFilePath = ossService.upload(kmzPath,multipartFile); result.setPath(kmzFilePath); return result; }else { try (ByteArrayOutputStream baos = new ByteArrayOutputStream(); ZipOutputStream zipOutputStream = new ZipOutputStream(baos)) { zipOutputStream.setLevel(0); // 创建 wpmz 目录中的 template.kml 文件条目 buildZipFile("wpmz/template.kml", zipOutputStream, kml); // 创建 wpmz 目录中的 waylines.wpml 文件条目 buildZipFile("wpmz/waylines.wpml", zipOutputStream, wpml); zipOutputStream.finish(); zipOutputStream.flush(); // 获取文件大小 Double fileSize = (double) baos.size(); // 字节数 // 创建MultipartFile对象 MultipartFile multipartFile = new MockMultipartFile( fileName + ".kmz", fileName + ".kmz", "application/vnd.google-earth.kmz", new ByteArrayInputStream(baos.toByteArray()) ); String kmzFilePath = ossService.upload(kmzPath,multipartFile); result.setUpPath(kmzFilePath); return result; } catch (Exception e) { throw new RuntimeException("KMZ文件生成或上传失败", e); } } catch (Exception e) { throw new RuntimeException("KMZ文件生成或上传失败", e); } } @@ -233,8 +210,8 @@ * 解析kmz文件 */ @Override public KmzInfoVO parseKmz(String fileUrl){ KmzInfoVO kmzInfoVO = new KmzInfoVO(); public KmzInfoVo parseKmz(String fileUrl){ KmzInfoVo kmzInfoVO = new KmzInfoVo(); try (InputStream inputStream = ossService.getObject(kmzPath,fileUrl); ArchiveInputStream archiveInputStream = new ZipArchiveInputStream(inputStream)){ ArchiveEntry entry; @@ -257,14 +234,14 @@ public KmzResultVo updateKmzNew(UavRouteReq uavRouteReq, String oldFile, String newFile) { KmzResultVo result = new KmzResultVo(); // 校验修改的文件是否存在 ArdRouteInfor routeInformation = ardRouteInformationService.selectArdRouteInforByName(oldFile); ArdRouteInfo routeInformation = ardRouteInformationService.selectArdRouteInfoByName(oldFile); if ( routeInformation == null ){ result.setData("修改航线文件失败，请检查文件是否存在"); return result; } // 获取文件 String minioUrl = subStringMinioUrlName(routeInformation.getUpPath()); String minioUrl = subStringMinioUrlName(routeInformation.getPath()); try (InputStream inputStream = ossService.getObject(kmzPath,minioUrl); ArchiveInputStream archiveInputStream = new ZipArchiveInputStream(inputStream)){ ArchiveEntry entry;
