大屏项目接口

liujie

2026-01-15 f94782b7302f761799c1a0fbc155bed2d4dc504f

 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/RecordServiceImpl.java

@@ -8,10 +8,12 @@
import com.ruoyi.system.mapper.RecordMapper;
import com.ruoyi.system.model.Record;
import com.ruoyi.system.model.RecordImage;
import com.ruoyi.system.model.UserFile;
import com.ruoyi.system.query.HiddenDangerListQuery;
import com.ruoyi.system.service.RecordImageService;
import com.ruoyi.system.service.RecordService;
import com.ruoyi.system.service.TaskService;
import com.ruoyi.system.service.UserFileService;
import com.ruoyi.system.vo.HiddenDangerListVo;
import com.ruoyi.system.vo.ProblemVo;
import java.util.Map;
@@ -37,6 +39,8 @@
    @Autowired
    private RecordImageService recordImageService;

    @Autowired
    private UserFileService userFileService;



@@ -52,6 +56,13 @@
            list.forEach(vo -> {
                List<RecordImage> collect = recordImages.stream().filter(recordImage -> recordImage.getRecordId().equals(vo.getId())).collect(Collectors.toList());
                vo.setFileNum(collect.size());
                if(!collect.isEmpty()){
                    List<Integer> fileIds = collect.stream().map(RecordImage::getUserFileId).collect(Collectors.toList());
                    List<UserFile> userFiles = userFileService.listByIds(fileIds);
                    vo.setUserFiles(userFiles);

                }

            });
        }