图斑图片绑定

rain

2024-07-26 d125bf54fd48661cf03a3e6083bdb764fde3c347

 src/main/java/com/dji/sample/media/service/impl/FileServiceImpl.java

@@ -112,27 +112,14 @@
    }

    @Override
    public Integer saveFile(String workspaceId, FileUploadDTO file) throws IOException, ImageProcessingException {
    public Integer saveFile(String workspaceId, FileUploadDTO file) {
        MediaFileEntity fileEntity = this.fileUploadConvertToEntity(file);
        fileEntity.setWorkspaceId(workspaceId);
        fileEntity.setFileId(UUID.randomUUID().toString());
        int count = mapper.insert(fileEntity);
            String url = pojo.getEndpoint() + "/" + pojo.getBucket() + file.getObjectKey();
            if (file.getObjectKey().endsWith(".jpeg")) {
                File file1 = TbFjServiceImpl.downloadFile(url);
                Object data = ImgUtil.getInfo(file1);
                fileEntity.setDroneData(data);
                saveNailFile(workspaceId, file, file1,data);
                count = mapper.insert(fileEntity);
                file1.delete();
            }
            if (file.getObjectKey().endsWith(".mp4")) {
                saveNailFile(workspaceId, file, null,null);
            }
            return count;
        return mapper.insert(fileEntity);
    }

    public void saveNailFile(String workspaceId, FileUploadDTO file, File file1,Object data) throws IOException {
    public void saveNailFile(String workspaceId, FileUploadDTO file) throws IOException, ImageProcessingException {
        updateStatue(file.getName());
        boolean endsWith = file.getObjectKey().endsWith(".mp4");
        if (endsWith) {
@@ -141,8 +128,12 @@
            nailEntity.setFileId(UUID.randomUUID().toString());
            nailMapper.insert(nailEntity);
        } else {
            String url = pojo.getEndpoint() + "/" + pojo.getBucket() + file.getObjectKey();
            File file1 = TbFjServiceImpl.downloadFile(url);
            File nailFile = new File(ImgZipUtil.compressImage(file1, 50).toURI());
            MediaFileNailEntity nailEntity = this.fileUploadConvertToNailEntity(file);
            Object data = ImgUtil.getInfo(file1);
            nailEntity.setIsadd(0);
            nailEntity.setIsOriginal(false);
            nailEntity.setDronedata(data);
            nailEntity.setWorkspaceId(workspaceId);
@@ -195,11 +186,20 @@
                    return markEntity.getDronedata();
                }
            }
        } else {
           String names = "nail" + filename;
            List<MediaFileNailEntity> entities = nailMapper.selectList(new LambdaQueryWrapper<MediaFileNailEntity>()
                    .eq(MediaFileNailEntity::getFileName, names));
            if (entities != null && !entities.isEmpty()) {
                for (MediaFileNailEntity NailEntity : entities) {
                    if (NailEntity.getDronedata() != null) {
                        return NailEntity.getDronedata();
                    }
                }
            }
        }

        return "null";
    }


    @Override
    public List<MediaFileDTO> getAllFilesByWorkspaceId(String workspaceId) {
@@ -711,8 +711,8 @@
    public void updateMarkMediaFileNames(String jobId) {
        try {
            // 查询符合条件的数据
            List<MediaFileMarkEntity> mediaFiles = markMapper.selectList(new LambdaQueryWrapper<MediaFileMarkEntity>()
                    .eq(MediaFileMarkEntity::getJobId, jobId));
            List<MediaFileNailEntity> mediaFiles = nailMapper.selectList(new LambdaQueryWrapper<MediaFileNailEntity>()
                    .eq(MediaFileNailEntity::getJobId, jobId));

            // 判断所有 fileName 是否都包含 '~' 或者都不包含 '~'
            boolean allContainTilde = mediaFiles.stream().allMatch(file -> file.getFileName().contains("~"));
@@ -728,17 +728,17 @@
            }

            // 筛选出name字段不包含'~'的数据
            List<MediaFileMarkEntity> filteredFiles = mediaFiles.stream()
            List<MediaFileNailEntity> filteredFiles = mediaFiles.stream()
                    .filter(file -> !file.getFileName().contains("~"))
                    .collect(Collectors.toList());

            for (MediaFileMarkEntity currentFile : filteredFiles) {
            for (MediaFileNailEntity currentFile : filteredFiles) {
                String currentName = currentFile.getFileName();
                Map<String, Object> currentMetadata = JSON.parseObject(JSON.toJSONString(currentFile.getMetadata()), Map.class);
                Long currentCreatedTime = (Long) currentMetadata.get("createdTime");

                // 找到metadata中的createdTime小于当前数据的createdTime且最接近的那条数据
                Optional<MediaFileMarkEntity> closestFileOpt = mediaFiles.stream()
                Optional<MediaFileNailEntity> closestFileOpt = mediaFiles.stream()
                        .filter(file -> {
                            Map<String, Object> metadata = JSON.parseObject(JSON.toJSONString(file.getMetadata()), Map.class);
                            Long createdTime = (Long) metadata.get("createdTime");
@@ -909,11 +909,11 @@
                .groupBy(MediaFileEntity::getFileName));  // 使用groupBy去重
    }

    public void updateById(Integer id, MediaFileMarkEntity entity) {
    public void updateById(Integer id, MediaFileNailEntity entity) {
        entity.setIsadd(1);
        UpdateWrapper<MediaFileMarkEntity> updateWrapper = new UpdateWrapper<>();
        UpdateWrapper<MediaFileNailEntity> updateWrapper = new UpdateWrapper<>();
        updateWrapper.eq("id", id);
        markMapper.update(entity, updateWrapper);
        nailMapper.update(entity, updateWrapper);
    }

    public void updateMediaById(Integer id, MediaFileEntity entity) {
@@ -923,8 +923,8 @@
    }

    public void getNoaddFile() {
        List<MediaFileMarkEntity> markEntities = markMapper.selectList(new LambdaQueryWrapper<MediaFileMarkEntity>().eq(MediaFileMarkEntity::getIsadd, 0));
        for (MediaFileMarkEntity mark : markEntities) {
        List<MediaFileNailEntity> markEntities = nailMapper.selectList(new LambdaQueryWrapper<MediaFileNailEntity>().eq(MediaFileNailEntity::getIsadd, 0));
        for (MediaFileNailEntity mark : markEntities) {
            updateMarkMediaFileNames(mark.getJobId());
            updateMediaFileNames(mark.getJobId());
        }