rain
2024-07-31 587e31924f7bae43f8f31cb4fc9bf5f33593b50f
src/main/java/com/dji/sample/media/service/impl/FileServiceImpl.java
@@ -312,6 +312,47 @@
    }
    @Override
    public PaginationData<MediaFileEntity> mediaQuerys(Integer page, Integer pageSize, String workspaceId) {
        // 创建查询条件对象
        LambdaQueryWrapper<MediaFileEntity> queryWrapper = new LambdaQueryWrapper<>();
        // 添加查询条件
        queryWrapper.eq(MediaFileEntity::getWorkspaceId, workspaceId);
        queryWrapper.last("ORDER BY JSON_EXTRACT(metadata, '$.createdTime') DESC");
        // 执行查询获取所有结果
        List<MediaFileEntity> allResults = mapper.selectList(queryWrapper);
        // 处理结果去重并设置任务名称
        Map<String, MediaFileEntity> uniqueFileMap = new LinkedHashMap<>();
        List<MediaFileEntity> uniqueResults = allResults.stream()
                .peek(mediaFile -> {
                    String taskNameResult = waylineJobService.getName(mediaFile.getJobId());
                    mediaFile.setJobName(taskNameResult);
                })
                .filter(mediaFile -> uniqueFileMap.putIfAbsent(mediaFile.getJobId(), mediaFile) == null)
                .collect(Collectors.toList());
        // 计算分页信息
        int total = uniqueResults.size();
        int start = (page - 1) * pageSize;
        int end = Math.min(start + pageSize, total);
        // 获取当前页的结果
        List<MediaFileEntity> pageResults = uniqueResults.subList(start, end);
        // 创建临时的 Page 对象
        Page<MediaFileEntity> resultPage = new Page<>(page, pageSize);
        resultPage.setRecords(pageResults);
        resultPage.setTotal(total);
        // 返回分页数据
        return new PaginationData<>(pageResults, new Pagination(resultPage));
    }
    @Override
    public PaginationData<MediaFileNailEntity> mediaNailQuery(Integer page, Integer pageSize, Long updateStart, Long updateEnd, Long photoStart, Long photoEnd, String jobName, String workspaceId, String type) {
        // 创建查询条件对象
        LambdaQueryWrapper<MediaFileNailEntity> queryWrapper = new LambdaQueryWrapper<>();
@@ -940,31 +981,48 @@
        // 分页查询
        Page<MediaFileEntity> page = new Page<>(pageNum, pageSize);
        // 使用 QueryWrapper 来执行 DISTINCT 查询,并且选择 payload 和 create_time 字段
        // 使用 QueryWrapper 执行查询
        QueryWrapper<MediaFileEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper
                .select("DISTINCT job_id", "payload", "create_time")  // 选择 job_id, payload 和 create_time 字段
                .eq("workspace_id", workspaceId)
                .orderByDesc("create_time");
                .orderByDesc("create_time");  // 根据 create_time 降序排序
        // 执行分页查询
        Page<MediaFileEntity> resultPage = mapper.selectPage(page, queryWrapper);
        List<MediaFileEntity> result = resultPage.getRecords();
        // 根据 job_id 字段进行去重
        Map<String, MediaFileEntity> uniqueFilesMap = result.stream()
                .collect(Collectors.toMap(
                        MediaFileEntity::getJobId,
                        entity -> entity,
                        (existing, replacement) -> existing // 如果有重复的,保留已有的
                ));
        List<MediaFileEntity> uniqueFiles = new ArrayList<>(uniqueFilesMap.values());
        // 为每个 MediaFileEntity 设置 jobName
        for (MediaFileEntity mediaFile : result) {
        for (MediaFileEntity mediaFile : uniqueFiles) {
            String taskNameResult = waylineJobService.getName(mediaFile.getJobId());
            mediaFile.setJobName(taskNameResult);
        }
        // 使用分页对象构造 Pagination
        Pagination pagination = new Pagination(resultPage);
        return new PaginationData<>(result, pagination);
        Page<MediaFileEntity> newPage = new Page<>(pageNum, pageSize);
        newPage.setRecords(uniqueFiles);
        newPage.setTotal(resultPage.getTotal()); // 设置总记录数为原始查询的总记录数
        Pagination pagination = new Pagination(newPage);
        return new PaginationData<>(uniqueFiles, pagination);
    }
    @Override
    public PaginationData<MediaFileEntity> getPhotoByJobId(int pageNum, int pageSize, String workspaceId, String jobId) {
        // 分页查询