From 2927bb498574a0e30bcbb3a9f7ee9468636cf0d8 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Thu, 15 Aug 2024 11:31:33 +0800
Subject: [PATCH] 优化无人机状态更新,更新照片角度附属信息

---
 src/main/java/com/dji/sample/media/service/impl/FileServiceImpl.java |   32 ++++++++++++++++++++------------
 1 files changed, 20 insertions(+), 12 deletions(-)

diff --git a/src/main/java/com/dji/sample/media/service/impl/FileServiceImpl.java b/src/main/java/com/dji/sample/media/service/impl/FileServiceImpl.java
index c406384..40ea318 100644
--- a/src/main/java/com/dji/sample/media/service/impl/FileServiceImpl.java
+++ b/src/main/java/com/dji/sample/media/service/impl/FileServiceImpl.java
@@ -120,12 +120,15 @@
     public Integer saveFile(String workspaceId, FileUploadDTO file) throws ImageProcessingException, IOException {
         MediaFileEntity fileEntity = this.fileUploadConvertToEntity(file);
         fileEntity.setWorkspaceId(workspaceId);
-        fileEntity.setIsadd(0);
         fileEntity.setFileId(UUID.randomUUID().toString());
-        String url = pojo.getEndpoint() + "/" + pojo.getBucket() + file.getObjectKey();
-        File downloadedFile = TbFjServiceImpl.downloadFile(url);
-        Object data = ImgUtil.getInfo(downloadedFile);
-        fileEntity.setDroneData(data);
+        if (file.getObjectKey().endsWith("jpeg")){
+            fileEntity.setIsadd(0);
+            String url = pojo.getEndpoint() + "/" + pojo.getBucket() + file.getObjectKey();
+            File downloadedFile = TbFjServiceImpl.downloadFile(url);
+            Object data = ImgUtil.getInfo(downloadedFile);
+            fileEntity.setDroneData(data);
+            updateStatue(file.getName());
+        }
         return mapper.insert(fileEntity);
     }
     public void saveFiles(String workspaceId, FileUploadDTO file) {
@@ -715,6 +718,10 @@
 
         // 如果所有 fileName 都包含 '~' 或者都不包含 '~',将 is_add 字段改为 1
         if (allContainTilde || noneContainTilde) {
+            mediaFiles.forEach(file -> {
+                file.setIsadd(1);
+                updateMediaById(file.getId(), file);
+            });
             return;
         }
 
@@ -1081,7 +1088,7 @@
     }
 
     public void updateById(Integer id, MediaFileNailEntity entity) {
-        entity.setIsadd(1);
+//        entity.setIsadd(1);
         UpdateWrapper<MediaFileNailEntity> updateWrapper = new UpdateWrapper<>();
         updateWrapper.eq("id", id);
         nailMapper.update(entity, updateWrapper);
@@ -1095,18 +1102,19 @@
 
 
     public void updateMediaById(Integer id, MediaFileEntity entity) {
+        entity.setIsadd(1);
         UpdateWrapper<MediaFileEntity> updateWrapper = new UpdateWrapper<>();
         updateWrapper.eq("id", id);
         mapper.update(entity, updateWrapper);
     }
 
     public void getNoaddFile() {
-        List<MediaFileNailEntity> markEntities = nailMapper.selectList(
-                new LambdaQueryWrapper<MediaFileNailEntity>().eq(MediaFileNailEntity::getIsadd, 0));
-        for (MediaFileNailEntity mark : markEntities) {
-            updateNailMediaFileNames(mark.getJobId());
-            updateMediaFileNames(mark.getJobId());
-            updateMediaZipFileNames(mark.getJobId());
+        List<MediaFileEntity> mediaEntities = mapper.selectList(
+                new LambdaQueryWrapper<MediaFileEntity>().eq(MediaFileEntity::getIsadd, 0));
+        for (MediaFileEntity media : mediaEntities) {
+//            updateNailMediaFileNames(media.getJobId());
+            updateMediaFileNames(media.getJobId());
+//            updateMediaZipFileNames(media.getJobId());
         }
     }
 

--
Gitblit v1.9.3