diff --git a/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionEquInfoServiceImpl.java b/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionEquInfoServiceImpl.java index f61533a1..d3fc0e89 100644 --- a/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionEquInfoServiceImpl.java +++ b/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionEquInfoServiceImpl.java @@ -63,21 +63,14 @@ public class InspectionEquInfoServiceImpl extends ServiceImpl fileList = inspectionFileService.list(Wrappers.lambdaQuery() - .eq(InspectionFile::getFatherId, inspectionEquInfo.getFolderId())); - - List copyFileList = new ArrayList<>(); - for (InspectionFile inspectionFile : fileList) { - FileDO fileDO = new FileDO(); - //把文件名后缀截取掉 - fileDO.setUrl(inspectionFile.getFilePath()); - fileDO.setName(inspectionFile.getFileName()); - copyFileList.add(fileDO); + InspectionFile folder = inspectionFileService.getOne(Wrappers.lambdaQuery().eq(InspectionFile::getId, inspectionEquInfo.getFolderId())); + if (ObjectUtil.isNull(folder)) { + inspectionEquInfo.setFolderId(null); } - inspectionEquInfo.setFileList(copyFileList); } + return inspectionEquInfo; }