140 lines
5.2 KiB
Java
140 lines
5.2 KiB
Java
package com.ruoyi.airline.service.impl;
|
||
|
||
import com.ruoyi.airline.api.domain.AirLinePointVO;
|
||
import com.ruoyi.airline.domain.api.IAirlineFileDomain;
|
||
import com.ruoyi.airline.domain.model.AirlineFile;
|
||
import com.ruoyi.airline.domain.model.kml.KmlInfo;
|
||
import com.ruoyi.airline.domain.uitl.KmlFileUtils;
|
||
import com.ruoyi.airline.domain.uitl.WayPointUitls;
|
||
import com.ruoyi.airline.service.api.IAirlineFileService;
|
||
import com.ruoyi.airline.service.convert.AirlineFileServiceConvert;
|
||
import com.ruoyi.airline.service.dto.AirLinePointDTO;
|
||
import com.ruoyi.airline.service.dto.AirlineFileDTO;
|
||
import com.ruoyi.common.core.domain.R;
|
||
import com.ruoyi.common.core.exception.base.BaseException;
|
||
import com.ruoyi.system.api.RemoteFileService;
|
||
import org.apache.commons.compress.archivers.ArchiveEntry;
|
||
import org.apache.commons.compress.archivers.ArchiveInputStream;
|
||
import org.apache.commons.compress.archivers.zip.ZipArchiveInputStream;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.stereotype.Service;
|
||
import org.springframework.web.multipart.MultipartFile;
|
||
|
||
import java.io.IOException;
|
||
import java.util.List;
|
||
import java.util.Objects;
|
||
import java.util.UUID;
|
||
|
||
/**
|
||
* 航线文件Service实现类
|
||
*
|
||
* @author ruoyi
|
||
* @date 2026-01-17
|
||
*/
|
||
@Service
|
||
public class AirlineFileServiceImpl implements IAirlineFileService {
|
||
|
||
private static final Logger log = LoggerFactory.getLogger(AirlineFileServiceImpl.class);
|
||
|
||
@Autowired
|
||
private RemoteFileService remoteFileService;
|
||
|
||
@Autowired
|
||
private IAirlineFileDomain iAirlineFileDomain;
|
||
|
||
|
||
@Override
|
||
public AirlineFileDTO save(AirlineFileDTO dto) {
|
||
AirlineFile model = AirlineFileServiceConvert.toModel(dto);
|
||
Long id = iAirlineFileDomain.save(model);
|
||
dto.setId(id);
|
||
return dto;
|
||
}
|
||
|
||
|
||
/**
|
||
* 上传航线文件，如果是KMZ的需要转换成 waypoint 的，并且需要将 原始航线文件保存。
|
||
* 并且限制航线修改，因为暂时没做到 从waypoingt 转 KMZ
|
||
*
|
||
* @param file
|
||
* @return
|
||
*/
|
||
@Override
|
||
public AirlineFileDTO parseAndUplload(MultipartFile file) {
|
||
|
||
KmlInfo kmlInfo = new KmlInfo();
|
||
try (ArchiveInputStream archiveInputStream = new ZipArchiveInputStream(file.getInputStream());) {
|
||
ArchiveEntry entry;
|
||
while (!Objects.isNull(entry = archiveInputStream.getNextEntry())) {
|
||
String name = entry.getName();
|
||
if (name.toLowerCase().endsWith(".kml")) {
|
||
// 解KML文件
|
||
kmlInfo = KmlFileUtils.parseKml(archiveInputStream);
|
||
}
|
||
}
|
||
|
||
if (Objects.isNull(kmlInfo)) {
|
||
throw new BaseException("kmz文件内容缺失");
|
||
}
|
||
String globalHeight = kmlInfo.getDocument().getFolder().getGlobalHeight();
|
||
R<String> fileUrl = remoteFileService.uploadFileByData(UUID.randomUUID().toString(), "waypoints", WayPointUitls.kmz2waypoint(kmlInfo));
|
||
AirlineFileDTO dto = new AirlineFileDTO();
|
||
dto.setFileUrl(fileUrl.getData());
|
||
dto.setAirVendor("");
|
||
dto.setAirType("");
|
||
return dto;
|
||
|
||
} catch (IOException e) {
|
||
throw new BaseException("Waypoints文件生成失败");
|
||
} catch (Exception e) {
|
||
log.error("kmz航线转换失败", e);
|
||
throw new BaseException("kmz航线转换失败");
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 航点编辑，直接将大疆KMZ转换成waypoints操作，就不需要区分大疆还是拓恒了
|
||
*
|
||
* @param airlineFile 航线文件dto
|
||
* @return
|
||
*/
|
||
@Override
|
||
public AirlineFileDTO createOrupdate(AirlineFileDTO airlineFile) {
|
||
|
||
StringBuilder waypointBuilder = new StringBuilder("QGC WPL 110\n");
|
||
List<AirLinePointDTO> LineDto = airlineFile.getLinePointDtoList();
|
||
// 新建字节输出流,Freemarker操作此输出流写入生成的业务文件.
|
||
if (LineDto != null && !LineDto.isEmpty()) {
|
||
for (int i = 0; i < LineDto.size(); i++) {
|
||
AirLinePointDTO point = LineDto.get(i);
|
||
waypointBuilder.append(WayPointUitls.formatWaypointLine(point, i)).append("\n");
|
||
}
|
||
|
||
// 上传文件
|
||
R<String> fileUrl = remoteFileService.uploadFileByData(UUID.randomUUID().toString(), "waypoints", waypointBuilder.toString());
|
||
|
||
//保存航线文件数据
|
||
airlineFile.setFileUrl(fileUrl.getData());
|
||
airlineFile.setSource("airport");
|
||
airlineFile.setStatus(airlineFile.getStatus() == null ? 1 : airlineFile.getStatus());
|
||
AirlineFile model = AirlineFileServiceConvert.toModel(airlineFile);
|
||
Long id = iAirlineFileDomain.save(model);
|
||
model.setId(id);
|
||
return AirlineFileServiceConvert.toDTO(model);
|
||
}
|
||
return null;
|
||
|
||
}
|
||
|
||
@Override
|
||
public AirlineFile selectById(Long airlineId) {
|
||
if (airlineId == null) {
|
||
return null;
|
||
}
|
||
List<Long> ids = List.of(airlineId);
|
||
List<AirlineFile> airlineFiles = iAirlineFileDomain.selectFileListByIds(ids);
|
||
return airlineFiles != null && !airlineFiles.isEmpty() ? airlineFiles.get(0) : null;
|
||
}
|
||
} |