| | |
|---|
| | | package com.dji.sample.patches.service.impl; |
|---|
| | | |
|---|
| | | import com.dji.sample.common.error.CommonErrorEnum; |
|---|
| | | import com.dji.sample.component.mqtt.model.RequestsReply; |
|---|
| | | import com.dji.sample.patches.dao.ShpToDataSourceMapper; |
|---|
| | | import com.dji.sample.patches.model.entity.LotInfo; |
|---|
| | | import com.dji.sample.patches.service.ShpToDataSourceService; |
|---|
| | | import com.dji.sample.patches.utils.FormatConversionUtil; |
|---|
| | | import com.dji.sample.patches.utils.GeoToolsUtil; |
|---|
| | | import com.dji.sample.patches.utils.MultipartFileTOFileUtil; |
|---|
| | | import com.dji.sample.patches.utils.ShapeFileUtil; |
|---|
| | | import com.dji.sample.patches.xml.mode.XMLTemplateModel; |
|---|
| | | import com.dji.sample.patches.xml.utils.CreateWaylineFileUtils; |
|---|
| | | import org.locationtech.jts.geom.Coordinate; |
|---|
| | | import org.springframework.beans.factory.annotation.Autowired; |
|---|
| | | import org.springframework.core.io.FileSystemResource; |
|---|
| | | import org.springframework.core.io.Resource; |
|---|
| | | import org.springframework.http.HttpHeaders; |
|---|
| | | import org.springframework.http.HttpStatus; |
|---|
| | | import org.springframework.http.MediaType; |
|---|
| | | import org.springframework.http.ResponseEntity; |
|---|
| | | import org.springframework.stereotype.Service; |
|---|
| | | import org.springframework.web.multipart.MultipartFile; |
|---|
| | | |
|---|
| | | import java.io.File; |
|---|
| | | import java.io.IOException; |
|---|
| | | import java.io.UnsupportedEncodingException; |
|---|
| | | import java.net.URLEncoder; |
|---|
| | | import java.nio.charset.StandardCharsets; |
|---|
| | | import java.util.ArrayList; |
|---|
| | | import java.util.List; |
|---|
| | | import java.util.UUID; |
|---|
| | | |
|---|
| | | import static com.dji.sample.common.error.CommonErrorEnum.ILLEGAL_ARGUMENT; |
|---|
| | | import static com.dji.sample.patches.utils.ZipUtil.compressToKMZ; |
|---|
| | | import static org.locationtech.jts.io.WKTConstants.POLYGON; |
|---|
| | | |
|---|
| | | @Service |
|---|
| | | public class ShpToDataSourceServiceImpl implements ShpToDataSourceService { |
|---|
| | | @Autowired |
|---|
| | | private ShpToDataSourceMapper mapper; |
|---|
| | | |
|---|
| | | private int createTime = 0; |
|---|
| | | private int updateTime = 0; |
|---|
| | | String head = POLYGON; |
|---|
| | | double airportLat = 28.624514734; // 机场纬度 |
|---|
| | | double airportLon = 115.856725497; // 机场经度 |
|---|
| | | |
|---|
| | | String kmlFilePath = "src/main/resources/template/wpmz/template.kml"; // KML文件路径 |
|---|
| | | String wpmlFilePath = "src/main/resources/template/wpmz/waylines.wpml"; // WPML文件路径 |
|---|
| | | String end = ".kmz"; |
|---|
| | | String json; |
|---|
| | | |
|---|
| | | public void insertGeo(MultipartFile file) throws IOException { |
|---|
| | | LotInfo patches = new LotInfo(); |
|---|
| | | public List<LotInfo> insertGeo(MultipartFile file, String workspaceId, String waylineName) throws IOException { |
|---|
| | | List<LotInfo> list = new ArrayList<>(); |
|---|
| | | MultipartFileTOFileUtil multipartFileTOFileUtil = new MultipartFileTOFileUtil(); |
|---|
| | | File file1 = multipartFileTOFileUtil.multipartFile2File(file); |
|---|
| | | List<String> s = ShapeFileUtil.shpToGeoJson(file1); |
|---|
| | | String[] arr1 = FormatConversionUtil.formatConversion(s); |
|---|
| | | for (int i = 0; i < arr1.length; i++) { |
|---|
| | | LotInfo lotInfo = new LotInfo(); |
|---|
| | | String temp = arr1[i].trim(); |
|---|
| | | temp = FormatConversionUtil.modifySpacesAndCommas(temp); |
|---|
| | | json = head + temp; |
|---|
| | | patches.setBsm(UUID.randomUUID().toString()); |
|---|
| | | patches.setCreateTime(createTime); |
|---|
| | | patches.setUpdateTime(updateTime); |
|---|
| | | patches.setDkfw(json); |
|---|
| | | mapper.insert(patches); |
|---|
| | | String uuid = UUID.randomUUID().toString(); |
|---|
| | | String bsm = uuid.replaceAll("-", ""); |
|---|
| | | lotInfo.setBsm(bsm); |
|---|
| | | lotInfo.setWorkspaceId(workspaceId); |
|---|
| | | lotInfo.setDkfw(json); |
|---|
| | | lotInfo.setDkbh("dkbh" + i); |
|---|
| | | mapper.insert(lotInfo); |
|---|
| | | list.add(lotInfo); |
|---|
| | | System.out.println(lotInfo); |
|---|
| | | } |
|---|
| | | Coordinate[] coordinates = GeoToolsUtil.getRoutePointOrder(list, airportLat, airportLon); |
|---|
| | | XMLTemplateModel xmlModel = XMLTemplateModel.init(coordinates, list); |
|---|
| | | CreateWaylineFileUtils.createWaylineFile(xmlModel); |
|---|
| | | String destinationKMZFilePath = "src/main/resources/template/kmz/" + waylineName + ".kmz"; |
|---|
| | | try { |
|---|
| | | compressToKMZ(kmlFilePath, wpmlFilePath, destinationKMZFilePath); |
|---|
| | | System.out.println("Files compressed to KMZ successfully."); |
|---|
| | | } catch (IOException e) { |
|---|
| | | e.printStackTrace(); |
|---|
| | | System.out.println("Error compressing files to KMZ: " + e.getMessage()); |
|---|
| | | } |
|---|
| | | return list; |
|---|
| | | } |
|---|
| | | |
|---|
| | | public ResponseEntity<Resource> backWayline(String fileName) { |
|---|
| | | String name = fileName + end; |
|---|
| | | try { |
|---|
| | | File kmzFile = new File("src/main/resources/template/kmz/" + name); |
|---|
| | | boolean kzm=(kmzFile.exists()); |
|---|
| | | System.out.println(kzm); |
|---|
| | | System.out.println(kmzFile); |
|---|
| | | if (!kzm) { |
|---|
| | | System.out.println(kzm); |
|---|
| | | // 如果文件不存在，返回404 Not Found响应 |
|---|
| | | return ResponseEntity.status(HttpStatus.NOT_FOUND).body(null); |
|---|
| | | } |
|---|
| | | HttpHeaders headers = new HttpHeaders(); |
|---|
| | | headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + name + "\""); |
|---|
| | | headers.add(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_OCTET_STREAM_VALUE); |
|---|
| | | return ResponseEntity.ok().headers(headers).contentLength(kmzFile.length()). |
|---|
| | | contentType(MediaType.APPLICATION_OCTET_STREAM).body(new FileSystemResource(kmzFile)); |
|---|
| | | } catch (Exception e) { |
|---|
| | | e.printStackTrace(); |
|---|
| | | return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(null); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
