From 36e5b1a2e89077e20c35764356a1f70798142a08 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Thu, 06 Jun 2024 14:47:11 +0800
Subject: [PATCH] 完善图斑调查取证模块信息

---
 src/main/java/com/dji/sample/media/service/impl/FileServiceImpl.java |   36 +++++++++++++++++++++++++-----------
 1 files changed, 25 insertions(+), 11 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..284894f 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
@@ -59,7 +59,6 @@
     private GetPatchesMapper patchesMapper;
     @Autowired
     private IDeviceDictionaryService deviceDictionaryService;
-
     @Autowired
     private OssServiceContext ossService;
 
@@ -90,17 +89,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,13 +126,15 @@
         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
-    public Object mediaInfo(String fileId) {
+    public Object mediaInfo(String filename) {
+        String name= "mark"+filename;
         MediaFileMarkEntity entity = markMapper.selectOne(new LambdaQueryWrapper<MediaFileMarkEntity>()
-                .eq(MediaFileMarkEntity::getFileId, fileId));
+                .eq(MediaFileMarkEntity::getFileName, name));
+
         return entity.getDronedata();
     }
 

--
Gitblit v1.9.3