~nishiwei/ard-uav-server.git

			@@ -2,21 +2,18 @@

			import cn.hutool.core.bean.BeanUtil;
			import cn.hutool.core.collection.CollectionUtil;
			import cn.hutool.core.io.FileUtil;
			import com.ard.common.utils.SecurityUtils;
			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.ArdRouteInformation;
			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.IArdRouteInformationService;
			import com.ard.route.util.FileUtils;
			import com.ard.route.service.IArdRouteInfoService;
			import com.ard.route.util.InitializationUtils;
			import com.ard.route.util.RouteFileUtils;
			import com.ard.route.service.UavRouteService;
			@@ -27,7 +24,6 @@
			import org.apache.commons.compress.archivers.zip.ZipArchiveInputStream;
			import org.apache.commons.lang3.StringUtils;
			import org.springframework.beans.BeanUtils;
			import org.springframework.beans.factory.annotation.Value;
			import org.springframework.mock.web.MockMultipartFile;
			import org.springframework.stereotype.Service;
			import org.springframework.web.multipart.MultipartFile;
			@@ -53,7 +49,7 @@
			private OssServiceContext ossService;

			@Resource
			private IArdRouteInformationService ardRouteInformationService;
			private IArdRouteInfoService ardRouteInformationService;

			InitializationUtils init = new InitializationUtils();

			@@ -64,15 +60,15 @@
			public KmzResultVo updateKmz(UavRouteReq uavRouteReq,String fileName){
			KmzResultVo result = new KmzResultVo();
			// 校验修改的文件是否存在
			ArdRouteInformation routeInformation = ardRouteInformationService.selectArdRouteInformationByName(fileName);
			ArdRouteInfo routeInformation = ardRouteInformationService.selectArdRouteInfoByName(fileName);
			if ( routeInformation == null ){
			result.setData("修改航线文件失败，请检查文件是否存在");
			return result;
			}

			// 获取文件
			String minioUrl = routeInformation.getMinioUpPath();
			try (InputStream inputStream = ossService.getObject(kmzPath,minioUrl);
			String minioUrlName = subStringMinioUrlName(routeInformation.getPath());
			try (InputStream inputStream = ossService.getObject(kmzPath,minioUrlName);
			ArchiveInputStream archiveInputStream = new ZipArchiveInputStream(inputStream)){
			ArchiveEntry entry;
			KmlInfo kmlInfo = new KmlInfo();
			@@ -90,8 +86,8 @@
			}
			}
			// 删除文件
			ossService.deleteObject(kmzPath, minioUrl);
			return buildKmzMethod(routeInformation.getFileName(), kmlInfo, wpmlInfo);
			ossService.deleteObject(kmzPath, minioUrlName);
			return buildKmzMethod(routeInformation.getName(), kmlInfo, wpmlInfo);
			} catch (Exception e) {
			e.printStackTrace();
			result.setData("更新航线kmz文件失败");
			@@ -144,13 +140,13 @@
			* 生成kmz文件(带航点)
			*/
			@Override
			public KmzResultVo buildKmz(UavRouteReq uavRouteReq, String fileName) {
			public KmzResultVo buildKmz(UavRouteReq uavRouteReq, String kmzName) {
			// 赋值默认参数
			init.initialization(uavRouteReq);
			KmlParams kmlParams = new KmlParams();
			BeanUtils.copyProperties(uavRouteReq, kmlParams);
			kmlParams.setRoutePointList(BeanUtil.copyToList(uavRouteReq.getRoutePointList(), RoutePointInfo.class));
			return buildKmz(fileName, kmlParams);
			return buildKmz(kmzName, kmlParams);
			}

			/**
			@@ -182,57 +178,32 @@

			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.setFileNameNew(getFileNameWithoutExtension(kmzFilePath));
			result.setMinioUpPath(extractFilePath(kmzFilePath));
			result.setUpPath(kmzFilePath);
			result.setUpFileSize(fileSize);
			return result;
			} catch (Exception e) {
			throw new RuntimeException("KMZ文件生成或上传失败", e);
			}
			} catch (Exception e) {
			throw new RuntimeException("KMZ文件生成或上传失败", e);
			}
			}

			@@ -240,8 +211,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;
			@@ -260,42 +231,12 @@
			return null;
			}

			/**
			* KmzInfoVo实体类转换为UavRouteReq实体类
			*/
			@Override
			public KmzResultVo updateKmzNew(UavRouteReq uavRouteReq, String oldFile, String newFile) {
			KmzResultVo result = new KmzResultVo();
			// 校验修改的文件是否存在
			ArdRouteInformation routeInformation = ardRouteInformationService.selectArdRouteInformationByName(oldFile);
			if ( routeInformation == null ){
			result.setData("修改航线文件失败，请检查文件是否存在");
			return result;
			}

			// 获取文件
			String minioUrl = routeInformation.getMinioUpPath();
			try (InputStream inputStream = ossService.getObject(kmzPath,minioUrl);
			ArchiveInputStream archiveInputStream = new ZipArchiveInputStream(inputStream)){
			ArchiveEntry entry;
			KmlInfo kmlInfo = new KmlInfo();
			KmlInfo wpmlInfo = new KmlInfo();
			KmlParams kmlParams = new KmlParams();
			while (!Objects.isNull(entry = archiveInputStream.getNextEntry())) {
			String name = entry.getName();
			if (name.toLowerCase().endsWith(".kml")) {
			kmlInfo = RouteFileUtils.parseKml(archiveInputStream);
			buildKmlParams(kmlParams, kmlInfo);
			this.handleRouteUpdate(kmlInfo, uavRouteReq, FileTypeConstants.KML, kmlParams);
			} else if (name.toLowerCase().endsWith(".wpml")) {
			wpmlInfo = RouteFileUtils.parseKml(archiveInputStream);
			this.handleRouteUpdate(wpmlInfo, uavRouteReq, FileTypeConstants.WPML, kmlParams);
			}
			}
			// 删除文件
			ossService.deleteObject(kmzPath, minioUrl);
			return buildKmzMethod(newFile, kmlInfo, wpmlInfo);
			} catch (Exception e) {
			result.setData("更新航线kmz文件失败");
			return result;
			}
			public UavRouteReq kmzInfoVoToUavRouteReq(KmzInfoVo kmzInfoVo) {
			return kmzInfoVoToUavRouteReq(kmzInfoVo);
			}

			}