| | |
| | | package com.ruoyi.basic.utils; |
| | | |
| | | import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
| | | import com.baomidou.mybatisplus.core.metadata.IPage; |
| | | import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; |
| | | import com.ruoyi.basic.constant.ApplicationType; |
| | | import com.ruoyi.basic.constant.RecordType; |
| | | import com.baomidou.mybatisplus.extension.plugins.pagination.Page; |
| | | import com.ruoyi.basic.dto.StorageAttachmentDTO; |
| | | import com.ruoyi.basic.dto.StorageAttachmentVO; |
| | | import com.ruoyi.basic.dto.StorageBlobDTO; |
| | | import com.ruoyi.basic.dto.StorageBlobVO; |
| | | import com.ruoyi.basic.enums.ApplicationTypeEnum; |
| | | import com.ruoyi.basic.enums.RecordTypeEnum; |
| | | import com.ruoyi.basic.mapper.StorageAttachmentMapper; |
| | | import com.ruoyi.basic.mapper.StorageBlobMapper; |
| | | import com.ruoyi.basic.pojo.StorageAttachment; |
| | | import com.ruoyi.basic.pojo.StorageBlob; |
| | | import com.ruoyi.common.config.FileProperties; |
| | | import io.jsonwebtoken.Jwts; |
| | | import io.jsonwebtoken.SignatureAlgorithm; |
| | | import io.jsonwebtoken.security.Keys; |
| | | import lombok.RequiredArgsConstructor; |
| | | import net.coobird.thumbnailator.Thumbnails; |
| | | import org.springframework.beans.BeanUtils; |
| | | import org.springframework.data.redis.core.StringRedisTemplate; |
| | | import org.springframework.stereotype.Component; |
| | | import org.springframework.util.StringUtils; |
| | | |
| | | import javax.crypto.SecretKey; |
| | | import java.io.File; |
| | | import java.math.BigDecimal; |
| | | import java.nio.charset.StandardCharsets; |
| | | import java.time.LocalDate; |
| | | import java.time.format.DateTimeFormatter; |
| | | import java.util.*; |
| | |
| | | public class FileUtil { |
| | | private final FileProperties properties; |
| | | private final StorageAttachmentMapper storageAttachmentMapper; |
| | | private final StorageBlobMapper storageBlobMapper; |
| | | private final StringRedisTemplate stringRedisTemplate; |
| | | private final StorageBlobMapper storageBlobMapper; |
| | | |
| | | private static final String TOKEN_USAGE_KEY_PREFIX = "file:token:usage:"; |
| | | private static final DateTimeFormatter YEAR_PATH_FORMATTER = DateTimeFormatter.ofPattern("yyyy"); |
| | |
| | | * @param application 文件用途 |
| | | * @param recordType 关联记录类型 |
| | | * @param recordId 关联记录id |
| | | * @param storageBlobVOS 文件信息 |
| | | * @param storageBlobDTOS 文件信息 |
| | | */ |
| | | public void saveStorageAttachment(ApplicationType application, RecordType recordType, Long recordId, List<StorageBlobVO> storageBlobVOS) { |
| | | if (CollectionUtils.isEmpty(storageBlobVOS)) { |
| | | throw new RuntimeException("文件信息不能为空"); |
| | | } |
| | | if (!application.isValid()) { |
| | | public void saveStorageAttachment(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId, List<StorageBlobDTO> storageBlobDTOS) { |
| | | if (application == null) { |
| | | throw new RuntimeException("文件用途不能为空"); |
| | | } |
| | | if (!recordType.isValid()) { |
| | | if (recordType == null) { |
| | | throw new RuntimeException("关联记录类型不能为空"); |
| | | } |
| | | if (recordId == null || recordId <= 0) { |
| | |
| | | } |
| | | // 删除旧附件信息 |
| | | deleteStorageAttachmentsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (CollectionUtils.isEmpty(storageBlobDTOS)) { |
| | | return; |
| | | } |
| | | List<StorageAttachment> storageAttachments = new ArrayList<>(); |
| | | for (StorageBlobVO storageBlobVO : storageBlobVOS) { |
| | | for (StorageBlobDTO storageBlobDTO : storageBlobDTOS) { |
| | | StorageAttachment storageAttachment = new StorageAttachment(); |
| | | storageAttachment.setApplication(application.getType()); |
| | | storageAttachment.setRecordType(recordType.getType()); |
| | | storageAttachment.setRecordId(recordId); |
| | | storageAttachment.setStorageBlobId(storageBlobVO.getId()); |
| | | storageAttachment.setStorageBlobId(storageBlobDTO.getId()); |
| | | storageAttachment.setDeleted(0L); |
| | | storageAttachments.add(storageAttachment); |
| | | } |
| | | // todo fileChange |
| | | // storageAttachmentMapper.insert(storageAttachments); |
| | | storageAttachmentMapper.insert(storageAttachments); |
| | | } |
| | | |
| | | /** |
| | |
| | | * @param storageBlobIds 文件id |
| | | */ |
| | | public void deleteStorageBlobs(List<Long> storageBlobIds) { |
| | | // todo fileChange |
| | | // storageBlobMapper.deleteByIds(storageBlobIds); |
| | | storageBlobMapper.deleteByIds(storageBlobIds); |
| | | } |
| | | |
| | | /** |
| | |
| | | * @param storageAttachmentIds 文件id |
| | | */ |
| | | public void deleteStorageBlobsByStorageAttachmentIds(List<Long> storageAttachmentIds) { |
| | | // todo fileChange |
| | | // List<StorageAttachment> storageAttachments = storageAttachmentMapper.selectByIds(storageAttachmentIds); |
| | | // List<Long> storageBlobIds = storageAttachments.stream().map(StorageAttachment::getStorageBlobId).collect(Collectors.toList()); |
| | | // deleteStorageBlobs(storageBlobIds); |
| | | List<StorageAttachment> storageAttachments = storageAttachmentMapper.selectByIds(storageAttachmentIds); |
| | | List<Long> storageBlobIds = storageAttachments.stream().map(StorageAttachment::getStorageBlobId).collect(Collectors.toList()); |
| | | deleteStorageBlobs(storageBlobIds); |
| | | } |
| | | |
| | | /** |
| | |
| | | * |
| | | * @param application 文件用途 |
| | | * @param recordType 关联记录类型 |
| | | * @param recordId 关联记录id |
| | | * @param recordIds 关联记录id |
| | | */ |
| | | public void deleteStorageBlobsByApplicationAndRecordTypeAndRecordId(ApplicationType application, RecordType recordType, Long recordId) { |
| | | if (!application.isValid()) { |
| | | public void deleteStorageBlobsByApplicationAndRecordTypeAndRecordIds(ApplicationTypeEnum application, RecordTypeEnum recordType, List<Long> recordIds) { |
| | | if (recordIds == null || recordIds.isEmpty()) { |
| | | throw new RuntimeException("关联记录id不能为空"); |
| | | } |
| | | if (application == null) { |
| | | throw new RuntimeException("文件用途不能为空"); |
| | | } |
| | | if (!recordType.isValid()) { |
| | | if (recordType == null) { |
| | | throw new RuntimeException("关联记录类型不能为空"); |
| | | } |
| | | if (recordId == null || recordId <= 0) { |
| | | throw new RuntimeException("关联记录id不能为空"); |
| | | } |
| | | List<StorageAttachment> storageAttachments = storageAttachmentMapper.selectList(new LambdaQueryWrapper<StorageAttachment>() |
| | | .eq(StorageAttachment::getRecordType, recordType.getType()) |
| | | .eq(StorageAttachment::getRecordId, recordId) |
| | | .in(StorageAttachment::getRecordId, recordIds) |
| | | .eq(StorageAttachment::getApplication, application.getType())); |
| | | if (CollectionUtils.isNotEmpty(storageAttachments)) { |
| | | List<Long> storageAttachmentIds = storageAttachments.stream().map(StorageAttachment::getStorageBlobId) |
| | |
| | | */ |
| | | public void deleteStorageAttachmentsByStorageAttachmentIds(List<Long> storageAttachmentIds) { |
| | | deleteStorageBlobsByStorageAttachmentIds(storageAttachmentIds); |
| | | // todo fileChange |
| | | // storageAttachmentMapper.deleteByIds(storageAttachmentIds); |
| | | storageAttachmentMapper.deleteByIds(storageAttachmentIds); |
| | | } |
| | | |
| | | public void deleteStorageAttachmentsByApplicationAndRecordTypeAndRecordId(ApplicationType application, RecordType recordType, Long recordId) { |
| | | if (!application.isValid()) { |
| | | throw new RuntimeException("文件用途不能为空"); |
| | | } |
| | | if (!recordType.isValid()) { |
| | | throw new RuntimeException("关联记录类型不能为空"); |
| | | } |
| | | /** |
| | | * 删除文件关联信息 |
| | | * |
| | | * @param application 文件用途 |
| | | * @param recordType 关联记录类型 |
| | | * @param recordId 关联记录id |
| | | */ |
| | | public void deleteStorageAttachmentsByApplicationAndRecordTypeAndRecordId(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId) { |
| | | if (recordId == null || recordId <= 0) { |
| | | throw new RuntimeException("关联记录id不能为空"); |
| | | } |
| | | deleteStorageBlobsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (application == null) { |
| | | throw new RuntimeException("文件用途不能为空"); |
| | | } |
| | | if (recordType == null) { |
| | | throw new RuntimeException("关联记录类型不能为空"); |
| | | } |
| | | deleteStorageBlobsByApplicationAndRecordTypeAndRecordIds(application, recordType, Arrays.asList(recordId)); |
| | | storageAttachmentMapper.delete(new LambdaQueryWrapper<StorageAttachment>() |
| | | .eq(StorageAttachment::getRecordType, recordType.getType()) |
| | | .eq(StorageAttachment::getRecordId, recordId) |
| | |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件信息 |
| | | * 批量删除文件关联信息 attachment |
| | | * |
| | | * @param application 文件用途 |
| | | * @param recordType 关联记录类型 |
| | | * @param recordIds 关联记录id |
| | | */ |
| | | public void deleteStorageAttachmentsByApplicationAndRecordTypeAndRecordIds(ApplicationTypeEnum application, RecordTypeEnum recordType, List<Long> recordIds) { |
| | | if (recordIds == null || recordIds.isEmpty()) { |
| | | throw new RuntimeException("关联记录id不能为空"); |
| | | } |
| | | if (application == null) { |
| | | throw new RuntimeException("文件用途不能为空"); |
| | | } |
| | | if (recordType == null) { |
| | | throw new RuntimeException("关联记录类型不能为空"); |
| | | } |
| | | deleteStorageBlobsByApplicationAndRecordTypeAndRecordIds(application, recordType, recordIds); |
| | | storageAttachmentMapper.delete(new LambdaQueryWrapper<StorageAttachment>() |
| | | .eq(StorageAttachment::getRecordType, recordType.getType()) |
| | | .in(StorageAttachment::getRecordId, recordIds) |
| | | .eq(StorageAttachment::getApplication, application.getType())); |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件信息 attachment |
| | | * |
| | | * @param storageAttachmentIds 文件id |
| | | */ |
| | |
| | | if (CollectionUtils.isEmpty(storageAttachmentIds)) { |
| | | throw new RuntimeException("文件id不能为空"); |
| | | } |
| | | // todo fileChange |
| | | // return storageAttachmentMapper.selectByIds(storageAttachmentIds); |
| | | return new ArrayList<>(); |
| | | return storageAttachmentMapper.selectByIds(storageAttachmentIds); |
| | | } |
| | | |
| | | /** |
| | | * 通过文件用途、关联记录类型、关联记录id获取文件关联信息 |
| | | * 通过记录类型获取文件信息 attachment(分页) |
| | | * |
| | | * @param page 分页参数 |
| | | * @param storageAttachmentDTO 关联记录信息 |
| | | */ |
| | | public IPage<StorageAttachmentVO> getStorageAttachmentVosPageListByApplicationAndRecordTypeAndRecordId(Page page, StorageAttachmentDTO storageAttachmentDTO) { |
| | | // 分页查询符合条件的 StorageAttachment 记录 |
| | | LambdaQueryWrapper<StorageAttachment> queryWrapper = new LambdaQueryWrapper<StorageAttachment>() |
| | | .eq(StorageAttachment::getRecordType, storageAttachmentDTO.getRecordType()) |
| | | .eq(StorageAttachment::getRecordId, storageAttachmentDTO.getRecordId()); |
| | | if (storageAttachmentDTO.getApplication() != null) { |
| | | queryWrapper.eq(StorageAttachment::getApplication, storageAttachmentDTO.getApplication()); |
| | | } |
| | | IPage<StorageAttachmentVO> storageAttachmentIPage = storageAttachmentMapper.selectPage(page, queryWrapper); |
| | | |
| | | // 转换为 StorageAttachmentVO 并获取对应的 StorageBlobVO |
| | | List<StorageAttachmentVO> storageAttachmentVOS = new ArrayList<>(); |
| | | if (CollectionUtils.isNotEmpty(storageAttachmentIPage.getRecords())) { |
| | | for (StorageAttachment storageAttachment : storageAttachmentIPage.getRecords()) { |
| | | StorageAttachmentVO storageAttachmentVO = new StorageAttachmentVO(); |
| | | BeanUtils.copyProperties(storageAttachment, storageAttachmentVO); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobVOsByStorageAttachmentIds(Collections.singletonList(storageAttachment.getId())); |
| | | if (CollectionUtils.isEmpty(storageBlobVOS)) { |
| | | storageAttachmentVO.setStorageBlobVOS(new ArrayList<>()); |
| | | } else { |
| | | storageAttachmentVO.setStorageBlobVOS(storageBlobVOS); |
| | | } |
| | | storageAttachmentVOS.add(storageAttachmentVO); |
| | | } |
| | | } |
| | | |
| | | // 构建分页结果 |
| | | IPage<StorageAttachmentVO> resultPage = new Page<>(); |
| | | BeanUtils.copyProperties(storageAttachmentIPage, resultPage); |
| | | resultPage.setRecords(storageAttachmentVOS); |
| | | |
| | | return resultPage; |
| | | } |
| | | |
| | | /** |
| | | * 通过文件用途、关联记录类型、关联记录id获取文件关联信息 attachment |
| | | * |
| | | * @param application 文件用途 |
| | | * @param recordType 关联记录类型 |
| | | * @param recordId 关联记录id |
| | | */ |
| | | public List<StorageAttachment> getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(ApplicationType application, RecordType recordType, Long recordId) { |
| | | if (!application.isValid()) { |
| | | throw new RuntimeException("文件用途不能为空"); |
| | | } |
| | | if (!recordType.isValid()) { |
| | | throw new RuntimeException("关联记录类型不能为空"); |
| | | } |
| | | public List<StorageAttachment> getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId) { |
| | | if (recordId == null || recordId <= 0) { |
| | | throw new RuntimeException("关联记录id不能为空"); |
| | | } if (application == null) { |
| | | throw new RuntimeException("文件用途不能为空"); |
| | | } |
| | | if (recordType == null) { |
| | | throw new RuntimeException("关联记录类型不能为空"); |
| | | } |
| | | return storageAttachmentMapper.selectList(new LambdaQueryWrapper<StorageAttachment>() |
| | | .eq(StorageAttachment::getRecordType, recordType.getType()) |
| | |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件信息 |
| | | * 通过文件关联id获取文件信息 blob |
| | | * |
| | | * @param storageAttachmentIds 文件id |
| | | */ |
| | | public List<StorageBlobVO> getStorageBlobDTOsByStorageAttachmentIds(List<Long> storageAttachmentIds) { |
| | | public List<StorageBlobVO> getStorageBlobVOsByStorageAttachmentIds(List<Long> storageAttachmentIds) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByStorageAttachmentIds(storageAttachmentIds); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return null; |
| | | } |
| | | List<Long> storageBlobIds = storageAttachments.stream().map(StorageAttachment::getStorageBlobId).collect(Collectors.toList()); |
| | | // todo fileChange |
| | | // List<StorageBlob> storageBlobs = storageBlobMapper.selectByIds(storageBlobIds); |
| | | // List<StorageBlobDTO> storageBlobDTOS = new ArrayList<>(); |
| | | // for (StorageBlob storageBlob : storageBlobs) { |
| | | // StorageBlobDTO storageBlobDTO = new StorageBlobDTO(); |
| | | // BeanUtils.copyProperties(storageBlob, storageBlobDTO); |
| | | // storageBlobDTO.setPreviewURL(buildSignedPreviewUrl(storageBlobDTO)); |
| | | // storageBlobDTO.setDownloadURL(buildSignedDownloadUrl(storageBlobDTO)); |
| | | // storageBlobDTOS.add(storageBlobDTO); |
| | | // } |
| | | return new ArrayList<>(); |
| | | List<StorageBlob> storageBlobs = storageBlobMapper.selectByIds(storageBlobIds); |
| | | List<StorageBlobVO> storageBlobDTOS = new ArrayList<>(); |
| | | for (StorageBlob storageBlob : storageBlobs) { |
| | | StorageBlobVO storageBlobVO = new StorageBlobVO(); |
| | | BeanUtils.copyProperties(storageBlob, storageBlobVO); |
| | | storageBlobVO.setPreviewURL(buildSignedPreviewUrl(storageBlobVO)); |
| | | storageBlobVO.setDownloadURL(buildSignedDownloadUrl(storageBlobVO)); |
| | | storageBlobDTOS.add(storageBlobVO); |
| | | } |
| | | return storageBlobDTOS; |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件信息存在过期时间 |
| | | * 通过文件用途、关联记录类型、关联记录id获取文件信息 blob |
| | | * |
| | | * @param application 文件用途 |
| | | * @param recordType 关联记录类型 |
| | | * @param recordId 关联记录id |
| | | */ |
| | | public List<StorageBlobVO> getStorageBlobVOsByApplicationAndRecordTypeAndRecordId(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return null; |
| | | } |
| | | List<Long> storageBlobIds = storageAttachments.stream().map(StorageAttachment::getStorageBlobId).collect(Collectors.toList()); |
| | | List<StorageBlob> storageBlobs = storageBlobMapper.selectByIds(storageBlobIds); |
| | | List<StorageBlobVO> storageBlobDTOS = new ArrayList<>(); |
| | | for (StorageBlob storageBlob : storageBlobs) { |
| | | StorageBlobVO storageBlobVO = new StorageBlobVO(); |
| | | BeanUtils.copyProperties(storageBlob, storageBlobVO); |
| | | storageBlobVO.setPreviewURL(buildSignedPreviewUrl(storageBlobVO)); |
| | | storageBlobVO.setDownloadURL(buildSignedDownloadUrl(storageBlobVO)); |
| | | storageBlobDTOS.add(storageBlobVO); |
| | | } |
| | | return storageBlobDTOS; |
| | | } |
| | | |
| | | /** |
| | | * 通过文件用途、关联记录类型、关联记录id获取文件信息 自定义过期时间(分钟) blob |
| | | * |
| | | * @param application 文件用途 |
| | | * @param recordType 关联记录类型 |
| | | * @param recordId 关联记录id |
| | | * @param expired 过期时间 |
| | | */ |
| | | public List<StorageBlobVO> getStorageBlobVOsByApplicationAndRecordTypeAndRecordId(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId, BigDecimal expired) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return null; |
| | | } |
| | | List<Long> storageBlobIds = storageAttachments.stream().map(StorageAttachment::getStorageBlobId).collect(Collectors.toList()); |
| | | List<StorageBlob> storageBlobs = storageBlobMapper.selectByIds(storageBlobIds); |
| | | List<StorageBlobVO> storageBlobDTOS = new ArrayList<>(); |
| | | for (StorageBlob storageBlob : storageBlobs) { |
| | | StorageBlobVO storageBlobVO = new StorageBlobVO(); |
| | | BeanUtils.copyProperties(storageBlob, storageBlobVO); |
| | | storageBlobVO.setPreviewURL(buildSignedUrl(storageBlobVO, "/preview/", expired)); |
| | | storageBlobVO.setDownloadURL(buildSignedUrl(storageBlobVO, "/download/", expired)); |
| | | storageBlobDTOS.add(storageBlobVO); |
| | | } |
| | | return storageBlobDTOS; |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件信息存在过期时间 自定义过期时间(分钟) blob |
| | | * |
| | | * @param storageAttachmentIds 文件id |
| | | * @param expired 过期时间 |
| | | */ |
| | | public List<StorageBlobVO> getStorageBlobDTOsByStorageAttachmentIds(List<Long> storageAttachmentIds, BigDecimal expired) { |
| | | public List<StorageBlobVO> getStorageBlobVOsByStorageAttachmentIds(List<Long> storageAttachmentIds, BigDecimal expired) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByStorageAttachmentIds(storageAttachmentIds); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return null; |
| | | } |
| | | List<Long> storageBlobIds = storageAttachments.stream().map(StorageAttachment::getStorageBlobId).collect(Collectors.toList()); |
| | | // todo fileChange |
| | | // List<StorageBlob> storageBlobs = storageBlobMapper.selectByIds(storageBlobIds); |
| | | // List<StorageBlobDTO> storageBlobDTOS = new ArrayList<>(); |
| | | // for (StorageBlob storageBlob : storageBlobs) { |
| | | // StorageBlobDTO storageBlobDTO = new StorageBlobDTO(); |
| | | // BeanUtils.copyProperties(storageBlob, storageBlobDTO); |
| | | // storageBlobDTO.setPreviewURL(buildSignedUrl(storageBlobDTO, "/preview/", expired)); |
| | | // storageBlobDTO.setDownloadURL(buildSignedUrl(storageBlobDTO, "/download/", expired)); |
| | | // storageBlobDTOS.add(storageBlobDTO); |
| | | // } |
| | | // return storageBlobDTOS; |
| | | return new ArrayList<>(); |
| | | List<StorageBlob> storageBlobs = storageBlobMapper.selectByIds(storageBlobIds); |
| | | List<StorageBlobVO> storageBlobDTOS = new ArrayList<>(); |
| | | for (StorageBlob storageBlob : storageBlobs) { |
| | | StorageBlobVO storageBlobVO = new StorageBlobVO(); |
| | | BeanUtils.copyProperties(storageBlob, storageBlobVO); |
| | | storageBlobVO.setPreviewURL(buildSignedUrl(storageBlobVO, "/preview/", expired)); |
| | | storageBlobVO.setDownloadURL(buildSignedUrl(storageBlobVO, "/download/", expired)); |
| | | storageBlobDTOS.add(storageBlobVO); |
| | | } |
| | | return storageBlobDTOS; |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件信息 |
| | | * 通过文件关联id获取文件信息 attachment |
| | | * |
| | | * @param storageAttachmentIds 文件id |
| | | */ |
| | | public List<StorageAttachmentDTO> getStorageAttachmentDTOsByStorageAttachmentIds(List<Long> storageAttachmentIds) { |
| | | public List<StorageAttachmentVO> getStorageAttachmentVOSByStorageAttachmentIds(List<Long> storageAttachmentIds) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByStorageAttachmentIds(storageAttachmentIds); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return new ArrayList<>(); |
| | | } |
| | | List<StorageAttachmentDTO> storageAttachmentDTOS = new ArrayList<>(); |
| | | List<StorageAttachmentVO> storageAttachmentVOS = new ArrayList<>(); |
| | | for (StorageAttachment storageAttachment : storageAttachments) { |
| | | StorageAttachmentDTO storageAttachmentDTO = new StorageAttachmentDTO(); |
| | | BeanUtils.copyProperties(storageAttachment, storageAttachmentDTO); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobDTOsByStorageAttachmentIds(Collections.singletonList(storageAttachment.getId())); |
| | | StorageAttachmentVO storageAttachmentVO = new StorageAttachmentVO(); |
| | | BeanUtils.copyProperties(storageAttachment, storageAttachmentVO); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobVOsByStorageAttachmentIds(Collections.singletonList(storageAttachment.getId())); |
| | | if (CollectionUtils.isEmpty(storageBlobVOS)) { |
| | | storageAttachmentDTO.setStorageBlobVOS(new ArrayList<>()); |
| | | storageAttachmentVO.setStorageBlobVOS(new ArrayList<>()); |
| | | } else { |
| | | storageAttachmentDTO.setStorageBlobVOS(storageBlobVOS); |
| | | storageAttachmentVO.setStorageBlobVOS(storageBlobVOS); |
| | | } |
| | | storageAttachmentDTOS.add(storageAttachmentDTO); |
| | | storageAttachmentVOS.add(storageAttachmentVO); |
| | | } |
| | | return storageAttachmentDTOS; |
| | | return storageAttachmentVOS; |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件信息存在过期时间 |
| | | * 通过文件关联id获取文件信息存在过期时间 自定义过期时间(分钟) attachment |
| | | * |
| | | * @param storageAttachmentIds 文件id |
| | | * @param expired 过期时间 |
| | | */ |
| | | public List<StorageAttachmentDTO> getStorageAttachmentDTOsByStorageAttachmentIds(List<Long> storageAttachmentIds, BigDecimal expired) { |
| | | public List<StorageAttachmentVO> getStorageAttachmentVOSByStorageAttachmentIds(List<Long> storageAttachmentIds, BigDecimal expired) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByStorageAttachmentIds(storageAttachmentIds); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return new ArrayList<>(); |
| | | } |
| | | List<StorageAttachmentDTO> storageAttachmentDTOS = new ArrayList<>(); |
| | | List<StorageAttachmentVO> storageAttachmentVOS = new ArrayList<>(); |
| | | for (StorageAttachment storageAttachment : storageAttachments) { |
| | | StorageAttachmentDTO storageAttachmentDTO = new StorageAttachmentDTO(); |
| | | BeanUtils.copyProperties(storageAttachment, storageAttachmentDTO); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobDTOsByStorageAttachmentIds(Collections.singletonList(storageAttachment.getId()), expired); |
| | | StorageAttachmentVO storageAttachmentVO = new StorageAttachmentVO(); |
| | | BeanUtils.copyProperties(storageAttachment, storageAttachmentVO); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobVOsByStorageAttachmentIds(Collections.singletonList(storageAttachment.getId()), expired); |
| | | if (CollectionUtils.isEmpty(storageBlobVOS)) { |
| | | storageAttachmentDTO.setStorageBlobVOS(new ArrayList<>()); |
| | | storageAttachmentVO.setStorageBlobVOS(new ArrayList<>()); |
| | | } else { |
| | | storageAttachmentDTO.setStorageBlobVOS(storageBlobVOS); |
| | | storageAttachmentVO.setStorageBlobVOS(storageBlobVOS); |
| | | } |
| | | storageAttachmentDTOS.add(storageAttachmentDTO); |
| | | storageAttachmentVOS.add(storageAttachmentVO); |
| | | } |
| | | return storageAttachmentDTOS; |
| | | return storageAttachmentVOS; |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件信息 |
| | | * 通过文件关联id获取文件信息 attachment |
| | | * |
| | | * @param application 应用 |
| | | * @param recordType 记录类型 |
| | | * @param recordId 记录id |
| | | */ |
| | | public List<StorageAttachmentDTO> getStorageAttachmentDTOsByApplicationAndRecordTypeAndRecordId(ApplicationType application, RecordType recordType, Long recordId) { |
| | | public List<StorageAttachmentVO> getStorageAttachmentVOSByApplicationAndRecordTypeAndRecordId(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return new ArrayList<>(); |
| | | } |
| | | List<StorageAttachmentDTO> storageAttachmentDTOS = new ArrayList<>(); |
| | | List<StorageAttachmentVO> storageAttachmentVOS = new ArrayList<>(); |
| | | for (StorageAttachment storageAttachment : storageAttachments) { |
| | | StorageAttachmentDTO storageAttachmentDTO = new StorageAttachmentDTO(); |
| | | BeanUtils.copyProperties(storageAttachment, storageAttachmentDTO); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobDTOsByStorageAttachmentIds(Collections.singletonList(storageAttachment.getId())); |
| | | StorageAttachmentVO storageAttachmentVO = new StorageAttachmentVO(); |
| | | BeanUtils.copyProperties(storageAttachment, storageAttachmentVO); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobVOsByStorageAttachmentIds(Collections.singletonList(storageAttachment.getId())); |
| | | if (CollectionUtils.isEmpty(storageBlobVOS)) { |
| | | storageAttachmentDTO.setStorageBlobVOS(new ArrayList<>()); |
| | | storageAttachmentVO.setStorageBlobVOS(new ArrayList<>()); |
| | | } else { |
| | | storageAttachmentDTO.setStorageBlobVOS(storageBlobVOS); |
| | | storageAttachmentVO.setStorageBlobVOS(storageBlobVOS); |
| | | } |
| | | storageAttachmentDTOS.add(storageAttachmentDTO); |
| | | storageAttachmentVOS.add(storageAttachmentVO); |
| | | } |
| | | return storageAttachmentDTOS; |
| | | return storageAttachmentVOS; |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件信息存在过期时间 |
| | | * 通过文件关联id获取文件信息存在过期时间 自定义过期时间(分钟) attachment |
| | | * |
| | | * @param application 应用 |
| | | * @param recordType 记录类型 |
| | | * @param recordId 记录id |
| | | * @param expired 过期时间 |
| | | */ |
| | | public List<StorageAttachmentDTO> getStorageAttachmentDTOsByApplicationAndRecordTypeAndRecordId(ApplicationType application, RecordType recordType, Long recordId, BigDecimal expired) { |
| | | public List<StorageAttachmentVO> getStorageAttachmentVOSByApplicationAndRecordTypeAndRecordId(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId, BigDecimal expired) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return new ArrayList<>(); |
| | | } |
| | | List<StorageAttachmentDTO> storageAttachmentDTOS = new ArrayList<>(); |
| | | List<StorageAttachmentVO> storageAttachmentVOS = new ArrayList<>(); |
| | | for (StorageAttachment storageAttachment : storageAttachments) { |
| | | StorageAttachmentDTO storageAttachmentDTO = new StorageAttachmentDTO(); |
| | | BeanUtils.copyProperties(storageAttachment, storageAttachmentDTO); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobDTOsByStorageAttachmentIds(Collections.singletonList(storageAttachment.getId()), expired); |
| | | StorageAttachmentVO storageAttachmentVO = new StorageAttachmentVO(); |
| | | BeanUtils.copyProperties(storageAttachment, storageAttachmentVO); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobVOsByStorageAttachmentIds(Collections.singletonList(storageAttachment.getId()), expired); |
| | | if (CollectionUtils.isEmpty(storageBlobVOS)) { |
| | | storageAttachmentDTO.setStorageBlobVOS(new ArrayList<>()); |
| | | storageAttachmentVO.setStorageBlobVOS(new ArrayList<>()); |
| | | } else { |
| | | storageAttachmentDTO.setStorageBlobVOS(storageBlobVOS); |
| | | storageAttachmentVO.setStorageBlobVOS(storageBlobVOS); |
| | | } |
| | | storageAttachmentDTOS.add(storageAttachmentDTO); |
| | | storageAttachmentVOS.add(storageAttachmentVO); |
| | | } |
| | | return storageAttachmentDTOS; |
| | | return storageAttachmentVOS; |
| | | } |
| | | |
| | | /** |
| | |
| | | */ |
| | | public List<String> getFilePreviewURLByStorageAttachmentIds(List<Long> storageAttachmentIds) { |
| | | List<String> res = new ArrayList<>(); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobDTOsByStorageAttachmentIds(storageAttachmentIds); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobVOsByStorageAttachmentIds(storageAttachmentIds); |
| | | for (StorageBlobVO storageBlobVO : storageBlobVOS) { |
| | | res.add(buildSignedPreviewUrl(storageBlobVO)); |
| | | } |
| | |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件预览地址存在过期时间 |
| | | * 通过文件关联id获取文件预览地址存在过期时间 自定义过期时间(分钟) |
| | | * |
| | | * @param storageAttachmentIds 文件关联id |
| | | * @param expired 过期时间 |
| | | */ |
| | | public List<String> getFilePreviewURLByStorageAttachmentIds(List<Long> storageAttachmentIds, BigDecimal expired) { |
| | | List<String> res = new ArrayList<>(); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobDTOsByStorageAttachmentIds(storageAttachmentIds); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobVOsByStorageAttachmentIds(storageAttachmentIds); |
| | | for (StorageBlobVO storageBlobVO : storageBlobVOS) { |
| | | res.add(buildSignedUrl(storageBlobVO, "/preview/", expired)); |
| | | } |
| | |
| | | */ |
| | | public List<String> getFileDownloadURLByStorageAttachmentIds(List<Long> storageAttachmentIds) { |
| | | List<String> res = new ArrayList<>(); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobDTOsByStorageAttachmentIds(storageAttachmentIds); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobVOsByStorageAttachmentIds(storageAttachmentIds); |
| | | for (StorageBlobVO storageBlobVO : storageBlobVOS) { |
| | | res.add(buildSignedDownloadUrl(storageBlobVO)); |
| | | } |
| | |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件下载地址存在过期时间 |
| | | * 通过文件关联id获取文件下载地址存在过期时间 自定义过期时间(分钟) |
| | | * |
| | | * @param storageAttachmentIds 文件关联id |
| | | * @param expired 过期时间 |
| | | */ |
| | | public List<String> getFileDownloadURLByStorageAttachmentIds(List<Long> storageAttachmentIds, BigDecimal expired) { |
| | | List<String> res = new ArrayList<>(); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobDTOsByStorageAttachmentIds(storageAttachmentIds); |
| | | List<StorageBlobVO> storageBlobVOS = getStorageBlobVOsByStorageAttachmentIds(storageAttachmentIds); |
| | | for (StorageBlobVO storageBlobVO : storageBlobVOS) { |
| | | res.add(buildSignedUrl(storageBlobVO, "/download/", expired)); |
| | | } |
| | | return res; |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件预览地址 |
| | | * |
| | | * @param application 应用 |
| | | * @param recordType 记录类型 |
| | | * @param recordId 记录id |
| | | */ |
| | | public List<String> getFilePreviewURLByApplicationAndRecordTypeAndRecordId(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return new ArrayList<>(); |
| | | } |
| | | return getFilePreviewURLByStorageAttachmentIds(storageAttachments.stream().map(StorageAttachment::getId).collect(Collectors.toList())); |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件预览地址存在过期时间 |
| | | * |
| | | * @param application 应用 |
| | | * @param recordType 记录类型 |
| | | * @param recordId 记录id |
| | | * @param expired 过期时间(分钟) |
| | | */ |
| | | public List<String> getFilePreviewURLByApplicationAndRecordTypeAndRecordIdAndExpired(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId, BigDecimal expired) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return new ArrayList<>(); |
| | | } |
| | | return getFilePreviewURLByStorageAttachmentIds(storageAttachments.stream().map(StorageAttachment::getId).collect(Collectors.toList()), expired); |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件下载地址 |
| | | * |
| | | * @param application 应用 |
| | | * @param recordType 记录类型 |
| | | * @param recordId 记录id |
| | | */ |
| | | public List<String> getFileDownloadURLByApplicationAndRecordTypeAndRecordId(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return new ArrayList<>(); |
| | | } |
| | | return getFileDownloadURLByStorageAttachmentIds(storageAttachments.stream().map(StorageAttachment::getId).collect(Collectors.toList())); |
| | | } |
| | | |
| | | /** |
| | | * 通过文件关联id获取文件下载地址存在过期时间 |
| | | * |
| | | * @param application 应用 |
| | | * @param recordType 记录类型 |
| | | * @param recordId 记录id |
| | | * @param expired 过期时间(分钟) |
| | | */ |
| | | public List<String> getFileDownloadURLByApplicationAndRecordTypeAndRecordIdAndExpired(ApplicationTypeEnum application, RecordTypeEnum recordType, Long recordId, BigDecimal expired) { |
| | | List<StorageAttachment> storageAttachments = getStorageAttachmentsByApplicationAndRecordTypeAndRecordId(application, recordType, recordId); |
| | | if (CollectionUtils.isEmpty(storageAttachments)) { |
| | | return new ArrayList<>(); |
| | | } |
| | | return getFileDownloadURLByStorageAttachmentIds(storageAttachments.stream().map(StorageAttachment::getId).collect(Collectors.toList()), expired); |
| | | } |
| | | |
| | | public String buildSignedPreviewUrl(StorageBlobVO storageBlob) { |
| | |
| | | if (storageBlob == null || !StringUtils.hasText(storageBlob.getUidFilename())) { |
| | | throw new IllegalArgumentException("文件信息不完整"); |
| | | } |
| | | long now = System.currentTimeMillis(); |
| | | long expiredMillis = expired.multiply(new BigDecimal("60000")).longValue(); |
| | | if (expiredMillis <= 0L) { |
| | | expiredMillis = 2L * 60L * 60L * 1000L; |
| | | } |
| | | Date issuedAt = new Date(now); |
| | | Date expiration = new Date(now + expiredMillis); |
| | | String token = Jwts.builder() |
| | | .setSubject(storageBlob.getUidFilename()) |
| | | .setIssuedAt(issuedAt) |
| | | .setExpiration(expiration) |
| | | .claim("path", storageBlob.getPath()) |
| | | .claim("resourceKey", storageBlob.getResourceKey()) |
| | | .signWith(SignatureAlgorithm.HS256, properties.getJwtSecret()) |
| | | .compact(); |
| | | cacheTokenUsage(token, expiredMillis); |
| | | String domain = StringUtils.trimTrailingCharacter(properties.getDomain(), '/'); |
| | | String prefix = properties.getUrlPrefix().startsWith("/") ? properties.getUrlPrefix() : "/" + properties.getUrlPrefix(); |
| | | String normalizedActionPath = StringUtils.hasText(actionPath) ? actionPath : "/preview/"; |
| | |
| | | if (!normalizedActionPath.endsWith("/")) { |
| | | normalizedActionPath = normalizedActionPath + "/"; |
| | | } |
| | | return domain + prefix + normalizedActionPath + storageBlob.getUidFilename() + "?token=" + token; |
| | | String baseUrl = domain + prefix + normalizedActionPath + storageBlob.getUidFilename(); |
| | | |
| | | // -1 表示永久有效,不生成 token,改为 publicKey 组合校验 |
| | | if (expired != null && BigDecimal.valueOf(-1L).compareTo(expired) == 0) { |
| | | if (!StringUtils.hasText(storageBlob.getResourceKey())) { |
| | | throw new IllegalArgumentException("公开链接缺少publicKey"); |
| | | } |
| | | return baseUrl + "?publicKey=" + storageBlob.getResourceKey(); |
| | | } |
| | | |
| | | long now = System.currentTimeMillis(); |
| | | BigDecimal expiredValue = expired == null ? new BigDecimal("120") : expired; |
| | | long expiredMillis = expiredValue.multiply(new BigDecimal("60000")).longValue(); |
| | | if (expiredMillis <= 0L) { |
| | | expiredMillis = 2L * 60L * 60L * 1000L; |
| | | } |
| | | Date issuedAt = new Date(now); |
| | | Date expiration = new Date(now + expiredMillis); |
| | | SecretKey key = Keys.hmacShaKeyFor(properties.getJwtSecret().getBytes(StandardCharsets.UTF_8)); |
| | | |
| | | String token = Jwts.builder() |
| | | .subject(storageBlob.getUidFilename()) |
| | | .issuedAt(issuedAt) // 新版建议直接调用 .issuedAt() |
| | | .expiration(expiration) // 新版建议直接调用 .expiration() |
| | | .claim("path", storageBlob.getPath()) |
| | | .claim("resourceKey", storageBlob.getResourceKey()) |
| | | .signWith(key) // 重点:传入上面生成的 key 对象,而不是 String |
| | | .compact(); |
| | | cacheTokenUsage(token, expiredMillis); |
| | | return baseUrl + "?token=" + token; |
| | | } |
| | | |
| | | private void cacheTokenUsage(String token, long expiredMillis) { |
| | |
| | | return properties.getUseLimit() == null || properties.getUseLimit() <= 0 ? 10 : properties.getUseLimit(); |
| | | } |
| | | |
| | | /** |
| | | * 压缩文件 图片 |
| | | * |
| | | * @param file 文件 |
| | | * @return 压缩后的文件 |
| | | */ |
| | | public File compressFile(File file) { |
| | | if (properties.getCompress() && isImage(file.getName()) && (file.length() > properties.getNeedCompressSize().toBytes())) { |
| | | try { |
| | | // 创建一个临时文件存放压缩后的图片,避免破坏原图 |
| | | File compressedFile = new File(file.getParent(), "thumb_" + file.getName()); |
| | | |
| | | // 1. 如果已经存在压缩过的文件,直接返回,不再消耗 CPU 压缩 |
| | | if (compressedFile.exists()) { |
| | | return compressedFile; |
| | | } |
| | | |
| | | // 使用 Thumbnailator 进行压缩 |
| | | Thumbnails.of(file) |
| | | .scale(1.0f) // 保持原尺寸 |
| | | .outputQuality(properties.getCompressQuality()) // 核心:设置画质 (0.0~1.0) |
| | | .toFile(compressedFile); |
| | | |
| | | return compressedFile; |
| | | } catch (Exception e) { |
| | | // 如果压缩失败,降级处理:返回原图 |
| | | return file; |
| | | } |
| | | } |
| | | return file; |
| | | } |
| | | // 简单的后缀判断 |
| | | private boolean isImage(String fileName) { |
| | | String ext = fileName.substring(fileName.lastIndexOf(".") + 1).toLowerCase(); |
| | | return "jpg".equals(ext) || "jpeg".equals(ext) || "png".equals(ext); |
| | | } |
| | | |
| | | |
| | | } |