package com.ruoyi.business.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.basic.entity.StorageAttachment;
|
import com.ruoyi.basic.entity.StorageBlob;
|
import com.ruoyi.basic.entity.dto.StorageBlobDTO;
|
import com.ruoyi.basic.mapper.StorageAttachmentMapper;
|
import com.ruoyi.basic.mapper.StorageBlobMapper;
|
import com.ruoyi.basic.service.StorageAttachmentService;
|
import com.ruoyi.business.dto.ArchiveDto;
|
import com.ruoyi.business.entity.Archive;
|
import com.ruoyi.business.entity.Tree;
|
import com.ruoyi.business.mapper.ArchiveMapper;
|
import com.ruoyi.business.mapper.TreeMapper;
|
import com.ruoyi.business.service.ArchiveService;
|
import com.ruoyi.common.utils.bean.BeanUtils;
|
import com.ruoyi.common.utils.file.MinioUtils;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.stereotype.Service;
|
|
import java.util.*;
|
import java.util.function.Function;
|
import java.util.stream.Collectors;
|
|
import static com.ruoyi.common.constant.StorageAttachmentConstants.StorageAttachmentFile;
|
import static com.ruoyi.common.enums.StorageAttachmentRecordType.Archives;
|
|
|
/**
|
* <p>
|
* 档案信息表，记录系统中各类档案的基本信息 服务实现类
|
* </p>
|
*
|
* @author ruoyi
|
* @since 2025-06-10
|
*/
|
@Service
|
@RequiredArgsConstructor
|
public class ArchiveServiceImpl extends ServiceImpl<ArchiveMapper, Archive> implements ArchiveService {
|
|
private final ArchiveMapper archiveMapper;
|
|
private final TreeMapper treeMapper;
|
|
private final StorageAttachmentService storageAttachmentService;
|
|
private final StorageBlobMapper storageBlobMapper;
|
|
private final StorageAttachmentMapper storageAttachmentMapper;
|
|
private final MinioUtils minioUtils;
|
|
|
@Override
|
public IPage<ArchiveDto> selectArchiveList(Page<Archive> page, ArchiveDto archiveDto) {
|
// 1. 分页查询主数据
|
LambdaQueryWrapper<Archive> queryWrapper = new LambdaQueryWrapper<>();
|
queryWrapper.orderByDesc(Archive::getCreateTime);
|
if (archiveDto.getTreeId() != null) {
|
Long treeId = archiveDto.getTreeId();
|
// 判断是否为主ID
|
Tree tree = treeMapper.selectById(archiveDto.getTreeId());
|
boolean isMainId = tree != null && tree.getParentId() == null;
|
|
if (isMainId) {
|
// 获取主ID的所有递归子节点ID
|
List<Long> recursiveSubIds = treeMapper.listRecursiveSubNodeIds(treeId);
|
|
// 将主ID本身和所有子节点ID加入查询条件
|
recursiveSubIds.add(treeId);
|
queryWrapper.in(Archive::getTreeId, recursiveSubIds);
|
} else {
|
// 非主ID，直接按原条件查询
|
queryWrapper.eq(Archive::getTreeId, treeId);
|
}
|
}
|
if (archiveDto.getSearchAll() != null) {
|
queryWrapper.like(Archive::getName, archiveDto.getSearchAll());
|
}
|
IPage<Archive> archivePage = archiveMapper.selectPage(page, queryWrapper);
|
|
// 2. 无数据提前返回
|
if (CollectionUtils.isEmpty(archivePage.getRecords())) {
|
return new Page<>(archivePage.getCurrent(), archivePage.getSize(), archivePage.getTotal());
|
}
|
|
// 3. 批量获取所有档案ID
|
List<Long> archiveIds = archivePage.getRecords()
|
.stream()
|
.map(Archive::getId)
|
.collect(Collectors.toList());
|
|
// 4. 批量查询附件映射关系（避免N+1）
|
|
List<StorageAttachment> storageAttachments = storageAttachmentMapper
|
.selectList(new LambdaQueryWrapper<StorageAttachment>()
|
.in(StorageAttachment::getRecordId, archiveIds)
|
.eq(StorageAttachment::getRecordType, Archives.ordinal())
|
);
|
|
Map<Long, List<StorageAttachment>> attachmentsMap = storageAttachmentMapper
|
.selectList(new LambdaQueryWrapper<StorageAttachment>()
|
.in(StorageAttachment::getRecordId, archiveIds)
|
.eq(StorageAttachment::getRecordType, Archives.ordinal())
|
).stream()
|
.collect(Collectors.groupingBy(StorageAttachment::getRecordId));
|
|
// 5. 批量查询所有需要的文件数据
|
Set<Long> blobIds = attachmentsMap.values()
|
.stream()
|
.flatMap(List::stream)
|
.map(StorageAttachment::getStorageBlobId)
|
.collect(Collectors.toSet());
|
Map<Long, StorageBlob> blobMap = blobIds.isEmpty()
|
? Collections.emptyMap()
|
: storageBlobMapper.selectList(new LambdaQueryWrapper<StorageBlob>().in(StorageBlob::getId, blobIds))
|
.stream()
|
.collect(Collectors.toMap(StorageBlob::getId, Function.identity()));
|
|
// 6. 组装DTO（单次循环完成数据装配）
|
List<ArchiveDto> dtoList = archivePage.getRecords().stream().map(archive -> {
|
ArchiveDto dto = new ArchiveDto();
|
BeanUtils.copyProperties(archive, dto); // 复制主对象属性
|
|
// 按需添加附件信息
|
List<StorageBlobDTO> blobDTOs = Optional.ofNullable(attachmentsMap.get(archive.getId()))
|
.orElse(Collections.emptyList())
|
.stream()
|
.map(att -> blobMap.get(att.getStorageBlobId()))
|
.filter(Objects::nonNull)
|
.map(blob -> {
|
StorageBlobDTO blobDTO = new StorageBlobDTO();
|
BeanUtils.copyProperties(blob, blobDTO);
|
|
// 动态生成预览地址和下载地址
|
// 设置预览URL
|
blobDTO.setUrl(minioUtils.getPreviewUrls(
|
blob.getBucketFilename(),
|
blob.getBucketName(),
|
true
|
));
|
|
// 设置下载URL
|
blobDTO.setDownloadUrl(minioUtils.getDownloadUrls(
|
blob.getBucketFilename(),
|
blob.getBucketName(),
|
blob.getOriginalFilename(), // 原始文件名
|
true
|
));
|
return blobDTO;
|
})
|
.collect(Collectors.toList());
|
|
dto.setStorageBlobDTO(blobDTOs);
|
return dto;
|
}).collect(Collectors.toList());
|
|
// 7. 构建返回分页对象
|
IPage<ArchiveDto> resultPage = new Page<>();
|
BeanUtils.copyProperties(archivePage, resultPage);
|
resultPage.setRecords(dtoList);
|
return resultPage;
|
}
|
|
@Override
|
public int addOrEditArchive(ArchiveDto archiveDto) {
|
Archive archive = new Archive();
|
BeanUtils.copyProperties(archiveDto, archive);
|
int i;
|
if (Objects.isNull(archiveDto.getId())) {
|
i = archiveMapper.insert(archive);
|
} else {
|
i = archiveMapper.updateById(archive);
|
}
|
if (archiveDto.getStorageBlobDTO() != null && !archiveDto.getStorageBlobDTO().isEmpty()) {
|
List<StorageAttachment> attachments = new ArrayList<>();
|
|
for (StorageBlobDTO storageBlobDTO : archiveDto.getStorageBlobDTO()) {
|
StorageAttachment storageAttachment = new StorageAttachment(
|
StorageAttachmentFile,
|
(long) Archives.ordinal(),
|
archive.getId()
|
);
|
storageAttachment.setStorageBlobDTO(storageBlobDTO);
|
attachments.add(storageAttachment);
|
}
|
storageAttachmentService.saveStorageAttachment(attachments, archive.getId(), Archives, StorageAttachmentFile);
|
}
|
return i;
|
}
|
|
@Override
|
public int delByIds(Long[] ids) {
|
// 检查参数
|
if (ids == null || ids.length == 0) {
|
return 0;
|
}
|
// 构造更新条件
|
UpdateWrapper<Archive> updateWrapper = new UpdateWrapper<>();
|
updateWrapper.in("id", ids)
|
.set("deleted", 1); // 设置 deleted 为 1 表示已删除
|
// 执行批量逻辑删除
|
return archiveMapper.update(null, updateWrapper);
|
}
|
|
@Override
|
public List<StorageAttachment> fileList(ArchiveDto archiveDto) {
|
List<StorageAttachment> storageAttachments = storageAttachmentService.selectStorageAttachments(archiveDto.getId(), Archives, StorageAttachmentFile);
|
return storageAttachments;
|
}
|
}
|
