From d4aecf9fb7d3d1dd504693f8d7175235087e4aeb Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Thu, 06 Jun 2024 09:34:06 +0800
Subject: [PATCH] 图斑图片调查状态

---
 src/main/java/com/dji/sample/media/service/impl/FileServiceImpl.java |   29 +++++++++++++++++++++--------
 1 files changed, 21 insertions(+), 8 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 9a54851..e5e90e0 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
@@ -90,17 +90,30 @@
         return mapper.insert(fileEntity);
     }
 
-    public Integer saveMarkFile(String workspaceId, FileUploadDTO file) throws IOException, FontFormatException, ImageProcessingException {
-        File file2 = null;
+    public void saveMarkFile(String workspaceId, FileUploadDTO file) throws IOException, FontFormatException, ImageProcessingException {
+        String name = TimerUtil.getDkbh(file.getName());
+        List<LotInfo> lotInfos = patchesMapper.selectList(new LambdaQueryWrapper<LotInfo>().eq(LotInfo::getDkbh, name));
+        if (!lotInfos.isEmpty()) {
+            LambdaUpdateWrapper<LotInfo> updateWrapper = new LambdaUpdateWrapper<>();
+            updateWrapper.eq(LotInfo::getDkbh, name)
+                    .eq(LotInfo::getInvestigate, 0)
+                    .set(LotInfo::getInvestigate, 1);
+            patchesMapper.update(null, updateWrapper);
+        }
+
+        boolean endsWith = file.getObjectKey().endsWith(".mp4");
+        if (endsWith) {
+            MediaFileMarkEntity mediaFileMarkEntity = this.fileUploadConvertToMarkEntity(file);
+            mediaFileMarkEntity.setWorkspaceId(workspaceId);
+            mediaFileMarkEntity.setFileId(UUID.randomUUID().toString());
+            markMapper.insert(mediaFileMarkEntity);
+        }
         MediaFileMarkEntity mediaFileMarkEntity = this.fileUploadConvertToMarkEntity(file);
         String url = "http://dev.jxpskj.com:9000/cloud-bucket" + file.getObjectKey();
         File file1 = TbFjServiceImpl.downloadFile(url);
         long timestamp = convertToTimestamp(file.getMetadata().getCreatedTime());
-        boolean endsWith = file.getObjectKey().endsWith(".mp4");
-        if (!endsWith) {
-            file2 = new File(WaterMarkUtil.addWatermark(file1, timestamp, file.getMetadata().getShootPosition().getLat(),
-                    file.getMetadata().getShootPosition().getLng(), file.getMetadata().getGimbalYawDegree()).toURI());
-        }
+        File file2 = new File(WaterMarkUtil.addWatermark(file1, timestamp, file.getMetadata().getShootPosition().getLat(),
+                file.getMetadata().getShootPosition().getLng(), file.getMetadata().getGimbalYawDegree()).toURI());
         Object data = ImgUtil.getInfo(file1);
         mediaFileMarkEntity.setDronedata(data);
         mediaFileMarkEntity.setWorkspaceId(workspaceId);
@@ -114,7 +127,7 @@
         String bucketName = "cloud-bucket";
         String objectName = mediaFileMarkEntity.getObjectKey(); // 例如 "folder/file.txt"
         uploadFile(endpoint, accessKey, secretKey, bucketName, objectName, file2);
-        return markMapper.insert(mediaFileMarkEntity);
+        markMapper.insert(mediaFileMarkEntity);
     }
 
     @Override

--
Gitblit v1.9.3