package com.monkeylessey.file.serivce.impl;
|
|
import com.aliyun.oss.model.AppendObjectRequest;
|
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
|
import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
|
import com.monkeylessey.constant.RedisKeyPrefixConstants;
|
|
import com.monkeylessey.enums.general.FileUploadTypeEnum;
|
import com.monkeylessey.exception.ServiceException;
|
import com.monkeylessey.file.config.OssTemplate;
|
import com.monkeylessey.file.domain.entity.FileInfo;
|
import com.monkeylessey.file.domain.vo.FileVo;
|
import com.monkeylessey.file.domain.vo.STSVO;
|
import com.monkeylessey.file.mapper.FileMapper;
|
import com.monkeylessey.file.properties.OssProperties;
|
import com.monkeylessey.file.serivce.OssFileService;
|
import com.monkeylessey.file.utils.FileUtil;
|
import com.monkeylessey.response.Result;
|
import com.monkeylessey.framework.utils.RedisUtil;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.context.annotation.Primary;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.util.StringUtils;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.ByteArrayInputStream;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Objects;
|
|
|
/**
|
* @author 29443
|
* @version 1.0
|
* @date 2022/4/25
|
*/
|
@Primary
|
@Service
|
@RequiredArgsConstructor
|
public class OssFileServiceImpl implements OssFileService {
|
|
private final OssTemplate ossTemplate;
|
private final OssProperties ossProperties;
|
private final FileMapper fileMapper;
|
private final RedisUtil redisUtil;
|
|
/**
|
* 单文件上传，返回有过期时间的链接
|
*
|
* @param file
|
* @return
|
*/
|
@Override
|
public Result putObject(MultipartFile file) {
|
FileInfo fileInfo = FileUtil.getFileInfo(file);
|
try {
|
ossTemplate.putObject(file.getInputStream(), fileInfo);
|
fileMapper.insert(fileInfo);
|
} catch (IOException e) {
|
e.printStackTrace();
|
throw new ServiceException("上传异常，请重试");
|
}
|
String url = ossTemplate.getHasExpireUrl(fileInfo.getFileKey());
|
FileVo fileVo = new FileVo().setUrl(url).setFileKey(fileInfo.getFileKey());
|
return Result.ok("上传成功").data(fileVo);
|
}
|
|
/**
|
* 单文件上传，返回没有过期时间的链接
|
*
|
* @param file
|
* @return
|
*/
|
@Override
|
public Result putObjectNoExpire(MultipartFile file) {
|
FileInfo fileInfo = FileUtil.getFileInfo(file);
|
try {
|
ossTemplate.putObject(file.getInputStream(), fileInfo);
|
fileMapper.insert(fileInfo);
|
} catch (IOException e) {
|
e.printStackTrace();
|
throw new ServiceException("上传异常，请重试");
|
}
|
FileVo fileVo = new FileVo().setUrl(ossTemplate.getNoExpireUrl(fileInfo.getFileKey())).setFileKey(fileInfo.getFileKey()).setOriginalName(fileInfo.getOriginalFilename());
|
return Result.ok("上传成功").data(fileVo);
|
}
|
|
/**
|
* 多文件上传，返回有过期时间的链接
|
*
|
* @param fileList
|
* @return
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public Result putObjects(List<MultipartFile> fileList) {
|
int successCount = 0;
|
int fileNumber = fileList.size();
|
List<FileVo> files = new ArrayList();
|
for (MultipartFile file : fileList) {
|
FileInfo fileInfo = FileUtil.getFileInfo(file);
|
try {
|
ossTemplate.putObject(file.getInputStream(), fileInfo);
|
fileMapper.insert(fileInfo);
|
files.add(new FileVo().setUrl(ossTemplate.getHasExpireUrl(fileInfo.getFileKey())).setFileKey(fileInfo.getFileKey()).setOriginalName(fileInfo.getOriginalFilename()));
|
successCount++;
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
String msg = String.format("上传文件总数：'{}'，成功'{}'个，失败'{}'个",
|
fileNumber,
|
successCount,
|
fileNumber - successCount);
|
return Result.ok(msg).data(files);
|
}
|
|
/**
|
* 多文件上传，返回没有过期时间的链接
|
*
|
* @param fileList
|
* @return
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public Result putObjectsNoExpire(List<MultipartFile> fileList) {
|
int successCount = 0;
|
int fileNumber = fileList.size();
|
List<FileVo> files = new ArrayList();
|
for (MultipartFile file : fileList) {
|
FileInfo fileInfo = (FileInfo) FileUtil.getFileInfo(file);
|
try {
|
ossTemplate.putObject(file.getInputStream(), fileInfo);
|
fileMapper.insert(fileInfo);
|
files.add(new FileVo().setFileKey(fileInfo.getFileKey()).setUrl(ossTemplate.getNoExpireUrl(fileInfo.getFileKey())).setOriginalName(fileInfo.getOriginalFilename()));
|
successCount++;
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
String msg = String.format("上传文件总数：'{}'，成功'{}'个，失败'{}'个",
|
fileNumber,
|
successCount,
|
fileNumber - successCount);
|
return Result.ok(msg).data(files);
|
}
|
|
/**
|
* 删除文件
|
*
|
* @param fileKey
|
* @return
|
*/
|
@Override
|
public Result deleteObject(String fileKey) {
|
ossTemplate.removeObject(fileKey);
|
new LambdaUpdateChainWrapper<>(fileMapper).eq(FileInfo::getFileKey, fileKey).remove();
|
return Result.ok("删除成功");
|
}
|
|
/**
|
* 批量删除文件
|
*
|
* @param fileKeys
|
* @return
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public Result deleteObjects(List<String> fileKeys) {
|
fileKeys.forEach(fileKey -> {
|
ossTemplate.removeObject(fileKey);
|
new LambdaUpdateChainWrapper<>(fileMapper).eq(FileInfo::getFileKey, fileKey).remove();
|
});
|
return Result.ok("批量删除成功");
|
}
|
|
/**
|
* 下载文件
|
*
|
* @param fileKey
|
* @return
|
*/
|
@Override
|
public Result getObject(String fileKey) {
|
InputStream object = ossTemplate.getObject(fileKey);
|
return Result.ok("下载成功").data(object);
|
}
|
|
/**
|
* 批量下载文件
|
*
|
* @param fileKeys
|
* @return
|
*/
|
@Override
|
public Result getObjects(List<String> fileKeys) {
|
List<InputStream> files = new ArrayList<>();
|
fileKeys.forEach(fileKey -> {
|
InputStream object = ossTemplate.getObject(fileKey);
|
files.add(object);
|
});
|
return Result.ok("下载成功").data(files);
|
}
|
|
|
/**
|
* 获取STS凭证
|
*
|
* @return
|
*/
|
@Override
|
public Result getSTSToken() {
|
String stsToken = redisUtil.getValue(RedisKeyPrefixConstants.OSSTOKEN, String.class);
|
String accessId = redisUtil.getValue(RedisKeyPrefixConstants.OSSACCESSKEYID, String.class);
|
String accessSecret = redisUtil.getValue(RedisKeyPrefixConstants.OSSACCESSKEYSECRET, String.class);
|
if (!StringUtils.hasText(stsToken) || !StringUtils.hasText(accessId) || !StringUtils.hasText(accessSecret)) {
|
ossTemplate.getSTSToken();
|
}
|
stsToken = redisUtil.getValue(RedisKeyPrefixConstants.OSSTOKEN, String.class);
|
accessId = redisUtil.getValue(RedisKeyPrefixConstants.OSSACCESSKEYID, String.class);
|
accessSecret = redisUtil.getValue(RedisKeyPrefixConstants.OSSACCESSKEYSECRET, String.class);
|
STSVO stsvo = new STSVO()
|
.setStsToken(stsToken)
|
.setAccessKeyId(accessId)
|
.setAccessKeySecret(accessSecret)
|
.setRegion(handleRegion(ossProperties.getEndpoint()))
|
.setBucket(ossProperties.getBucketName());
|
return Result.ok().data(stsvo);
|
}
|
|
/**
|
* 追加文件演示，追加上传适用于文件固定，即filekey固定的文件
|
* 如果是追加到动态（即用户上传一个文件，系统进行追加这种）
|
* 因为单从文件的属性上不能获取到一个不变固定的filekey，文件也就失去了唯一标识，不能根据filekey去数据库查询到该文件的追加位置
|
*/
|
@Override
|
public void uploadAppendDemo() {
|
// 追加的内容
|
String content0 = "Hello OSS -1 \n";
|
String content1 = "Hello OSS A \n";
|
String content2 = "Hello OSS B \n";
|
String content3 = "Hello OSS C \n";
|
String content4 = "Hello Append \n";
|
|
// 对于要追加的文件，先去数据库查找。
|
FileInfo fileInfo = new LambdaQueryChainWrapper<>(fileMapper)
|
.eq(FileInfo::getFileKey, "text/demo.txt")
|
.one();
|
// 数据库没有，说明是第一次上传，需要构建文件内容对象然后保存。第一次上传的逻辑也包含在内
|
if (Objects.isNull(fileInfo)) {
|
// 文件信息
|
fileInfo = new FileInfo();
|
fileInfo.setFileKey("text/demo.txt");
|
fileInfo.setOriginalFilename("abc.txt");
|
fileInfo.setUploadType(FileUploadTypeEnum.APPEND);
|
|
// 第一次上传
|
AppendObjectRequest appendObjectRequest = ossTemplate.firstAppend(new ByteArrayInputStream(content0.getBytes()), fileInfo, null);
|
// 后续追加
|
// 第一次追加（也可以把第一次上传理解为一次追加）
|
appendObjectRequest.setInputStream(new ByteArrayInputStream(content1.getBytes()));
|
appendObjectRequest = ossTemplate.nextAppend(appendObjectRequest);
|
|
// 第二次追加
|
appendObjectRequest.setInputStream(new ByteArrayInputStream(content2.getBytes()));
|
appendObjectRequest = ossTemplate.nextAppend(appendObjectRequest);
|
|
// 第三次追加
|
appendObjectRequest.setInputStream(new ByteArrayInputStream(content3.getBytes()));
|
appendObjectRequest = ossTemplate.nextAppend(appendObjectRequest);
|
// 追加完需要在数据库保存下次追加位置（必须持久化该值）
|
fileInfo.setAppendPosition(appendObjectRequest.getPosition());
|
}
|
// 数据库能找到说明，以前追加过，现在继续追加，追加位置拿数据库里查到的
|
else {
|
AppendObjectRequest appendObjectRequest = ossTemplate.getAppendObjectRequest(new ByteArrayInputStream(content4.getBytes()), fileInfo, ossTemplate.getDefaultMetadata(fileInfo));
|
appendObjectRequest.setPosition(fileInfo.getAppendPosition());
|
appendObjectRequest = ossTemplate.nextAppend(appendObjectRequest);
|
fileInfo.setAppendPosition(appendObjectRequest.getPosition());
|
}
|
this.saveOrUpdate(fileInfo);
|
}
|
|
|
/**
|
* 去掉region后面的.aliyuncs.com
|
*
|
* @param region
|
* @return
|
*/
|
public String handleRegion(String region) {
|
if (StringUtils.hasText(region)) {
|
int i = region.indexOf(".", 1);
|
return region.substring(0, i);
|
}
|
return null;
|
}
|
|
/**
|
* 添加或保存文件信息
|
* @param fileInfo
|
*/
|
private void saveOrUpdate(FileInfo fileInfo) {
|
if (StringUtils.hasText(fileInfo.getId())) {
|
fileMapper.updateById(fileInfo);
|
} else {
|
fileMapper.insert(fileInfo);
|
}
|
}
|
|
}
|
