无人机java后台代码
blame | history | patch | commit | commitdiff | ignore whitespace
rain
2024-08-21 2db1aa88e8ab53096a936163d686b90d8e056a99 
 src/main/java/com/dji/sample/territory/service/impl/TbDkjbxxServiceImpl.java


@@ -8,12 +8,15 @@


import com.dji.sample.patches.model.entity.LotInfo;


import com.dji.sample.patches.utils.GeoToolsUtil;


import com.dji.sample.patches.utils.MultipartFileTOFileUtil;


import com.dji.sample.patches.utils.PointPO;


import com.dji.sample.patches.utils.TimerUtil;


import com.dji.sample.patches.xml.mode.XMLTemplateModel;


import com.dji.sample.patches.xml.utils.CreateWaylineFileUtils;


import com.dji.sample.territory.dao.ITbDkjbxxMapper;


import com.dji.sample.territory.dao.ITbFjMapper;


import com.dji.sample.territory.model.entity.TbDkjbxxEntity;


import com.dji.sample.territory.model.entity.TbFjEntity;


import com.dji.sample.territory.model.entity.param.UploadUrlParam;


import com.dji.sample.territory.pojo.TerritoryConfigPojo;


import com.dji.sample.territory.service.ITbDkjbxxService;


import org.locationtech.jts.geom.Coordinate;


@@ -27,12 +30,16 @@


import java.io.*;


import java.net.HttpURLConnection;


import java.net.URL;


import java.net.URLConnection;


import java.util.ArrayList;


import java.util.List;


import java.util.Objects;


import java.util.UUID;




import static com.dji.sample.patches.utils.MultipartFileTOFileUtil.convert;


import static com.dji.sample.patches.utils.ZipUtil.zipFolder;


import static com.dji.sample.territory.utils.CoordinateSystemUtil.*;






/**


 * @PROJECT_NAME: drone


@@ -45,6 +52,8 @@


public class TbDkjbxxServiceImpl extends ServiceImpl<ITbDkjbxxMapper, TbDkjbxxEntity> implements ITbDkjbxxService {


    @Autowired


    private ITbDkjbxxMapper iTbDkjbxxMapper;


    @Autowired


    private TbFjServiceImpl tbFjService;


    @Autowired


    private PatchesConfigPojo patchesConfigPojo;


    @Autowired


@@ -68,7 +77,7 @@


            // 获取上传的文件输入流


            InputStream inputStream = file.getInputStream();


            // 创建输出流，将文件内容写入资源文件


            OutputStream outputStream = new FileOutputStream(territoryConfigPojo.getPath());


            OutputStream outputStream = new FileOutputStream(territoryConfigPojo.getResource());


            byte[] buffer = new byte[1024];


            int bytesRead;


            while ((bytesRead = inputStream.read(buffer)) != -1) {


@@ -83,22 +92,11 @@


        }


    }






    /**


     * 根据给定的工作空间ID、航线名称、机场纬度和经度，生成并返回一个包含航线文件的MultipartFile对象。


     *


     * @param workspaceId 工作空间ID，用于查询相关数据。


     * @param waylineName 航线名称，用于命名生成的文件。


     * @param airportLat  机场纬度，用于地理坐标转换和航迹点排序。


     * @param airportLon  机场经度，用于地理坐标转换和航迹点排序。


     * @return MultipartFile 对象，包含压缩后的航迹文件。


     * @throws IOException 如果文件操作失败，则抛出IOException。


     */


    @Transactional


    public MultipartFile listFile(String workspaceId, String waylineName, double airportLat, double airportLon) throws IOException {


        List<TbDkjbxxEntity> list = iTbDkjbxxMapper.selectList(null);


        List<LotInfo> info = dbConvertToEntity(list);


        Coordinate[] coordinates = GeoToolsUtil.getRoutePointOrder(info, airportLat, airportLon);


        List<PointPO> coordinates = GeoToolsUtil.getRoutePointOrder(info, airportLat, airportLon);


        // 创建XML模板模型


        XMLTemplateModel xmlModel = XMLTemplateModel.init(coordinates, info);


        CreateWaylineFileUtils.createWaylineFile(xmlModel, patchesConfigPojo.getTemplate(), patchesConfigPojo.getTargetTemplate(), patchesConfigPojo.getWaylines(), patchesConfigPojo.getTargetWaylines());


@@ -106,8 +104,45 @@


        String destKMZFile = patchesConfigPojo.getDestKMZFile() + waylineName + ".kmz"; // 输出的KMZ文件路径


        zipFolder(patchesConfigPojo.getSourceDir(), destKMZFile);


        // 将压缩文件转换为MultipartFile对象


        MultipartFile multipartFile = convert(new File(destKMZFile));


        return multipartFile;


        return convert(new File(destKMZFile));


    }




    @Override


    public void uploadUrl(String url, String taskName) {


        if (url.isEmpty()) {


            throw new IllegalArgumentException("上传文件为空");


        }


        String saveFilePath = territoryConfigPojo.getResource(); // 要保存文件的本地路径


        try {


            downloadFile(url, saveFilePath);


        } catch (IOException e) {


            throw new IllegalArgumentException("下载db文件发生错误" + e);


        }


        TimerUtil util = new TimerUtil();


        util.dbSave(territoryConfigPojo.getResource(), territoryConfigPojo.getResourcesave(), taskName);




    }




    /**


     * 根据地址下载文件


     *


     * @param fileUrl


     * @param saveFilePath


     * @throws IOException


     */


    public static void downloadFile(String fileUrl, String saveFilePath) throws IOException {


        // 创建URL对象


        URL url = new URL(fileUrl);




        // 打开连接


        try (BufferedInputStream in = new BufferedInputStream(url.openStream());


             FileOutputStream fileOutputStream = new FileOutputStream(saveFilePath)) {


            byte[] dataBuffer = new byte[1024];


            int bytesRead;


            while ((bytesRead = in.read(dataBuffer, 0, 1024)) != -1) {


                fileOutputStream.write(dataBuffer, 0, bytesRead);


            }


        }


    }




    /**


@@ -123,20 +158,10 @@


            LotInfo.LotInfoBuilder builder = LotInfo.builder();


            if (file != null) {


                // 从TbDkjbxxEntity实体中提取信息，构建LotInfo对象


                builder.bsm(file.getBsm())


                        .bz(file.getBz())


                        .dkmj(file.getDkmj())


                        .dkmc(file.getDkmc())


                        .dkbh(file.getDkbh())


                        .xmc(file.getXmc())


                        .xzb(file.getXzb())


                        .sfbhzdk(file.getSfbhzdk())


                        .sjlx(file.getSjlx())


                        .dkfw(file.getDkfw())


                        .xzqdm(file.getXzqdm())


                        .yzb(file.getYzb())


                        .kzxx(file.getKzxx())


                        .dklx(file.getDklx())


                builder.bsm(file.getFId())


                        .dkbh(file.getFTbbh())


                        .dkfw(file.getFShape())


                        .xzqdm(file.getFXzqdm())


                        .build();


                // 将构建好的LotInfo对象添加到infos列表中


                infos.add(builder.build());