From de821f0a99e234ce8e336b1775e90169e875cdd5 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Tue, 09 Apr 2024 09:44:38 +0800
Subject: [PATCH] 图斑生成kmz航线文件,入库并返回前端,优化图斑模块

---
 src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java |   29 ++++++++++++++++++++++++++++-
 1 files changed, 28 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java b/src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java
index a90846f..449316f 100644
--- a/src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java
+++ b/src/main/java/com/dji/sample/wayline/service/impl/WaylineFileServiceImpl.java
@@ -9,6 +9,7 @@
 import com.dji.sample.component.oss.model.OssConfiguration;
 import com.dji.sample.component.oss.service.impl.OssServiceContext;
 import com.dji.sample.manage.model.enums.DeviceDomainEnum;
+import com.dji.sample.summarize.model.entity.FlyInfo;
 import com.dji.sample.wayline.dao.IWaylineFileMapper;
 import com.dji.sample.wayline.model.dto.KmzFileProperties;
 import com.dji.sample.wayline.model.dto.WaylineFileDTO;
@@ -28,7 +29,6 @@
 import org.springframework.util.StringUtils;
 import org.springframework.web.multipart.MultipartFile;
 
-import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.URL;
@@ -182,6 +182,33 @@
     }
 
     @Override
+    public WaylineFileDTO importKmzFileBack(MultipartFile file, String workspaceId, String creator) {
+        WaylineFileDTO waylineFile=null;
+        Optional<WaylineFileDTO> waylineFileOpt = validKmzFile(file);
+        if (waylineFileOpt.isEmpty()) {
+            throw new RuntimeException("文件格式错误");
+        }
+
+        try {
+            waylineFile = waylineFileOpt.get();
+            waylineFile.setWaylineId(workspaceId);
+            waylineFile.setUsername(creator);
+
+            ossService.putObject(OssConfiguration.bucket, waylineFile.getObjectKey(), file.getInputStream());
+            this.saveWaylineFile(workspaceId, waylineFile);
+
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        return waylineFile;
+    }
+
+    @Override
+    public WaylineFileEntity selectByName(String name) {
+        return mapper.selectOne(new LambdaQueryWrapper<WaylineFileEntity>().eq(WaylineFileEntity::getName,name));
+    }
+
+    @Override
     public List<WaylineListDTO> waylineList(String workspaceId,String droneName) {
         LambdaQueryChainWrapper<WaylineFileEntity> wrapper =
                 new LambdaQueryChainWrapper<>(mapper).eq(WaylineFileEntity::getWorkspaceId, workspaceId);

--
Gitblit v1.9.3