|
|
|
@ -1,123 +1,139 @@
|
|
|
|
package com.unilife.utils;
|
|
|
|
package com.unilife.utils;
|
|
|
|
|
|
|
|
|
|
|
|
import com.aliyun.oss.OSS;
|
|
|
|
import com.aliyun.oss.OSS;
|
|
|
|
import com.aliyun.oss.model.DeleteObjectRequest;
|
|
|
|
import com.aliyun.oss.model.OSSObject;
|
|
|
|
import com.aliyun.oss.model.GeneratePresignedUrlRequest;
|
|
|
|
import com.aliyun.oss.model.GeneratePresignedUrlRequest;
|
|
|
|
import com.aliyun.oss.model.PutObjectRequest;
|
|
|
|
|
|
|
|
import com.unilife.config.OssConfig;
|
|
|
|
import com.unilife.config.OssConfig;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
import org.springframework.stereotype.Component;
|
|
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
|
|
|
|
|
|
|
|
import java.io.IOException;
|
|
|
|
import java.io.IOException;
|
|
|
|
import java.io.InputStream;
|
|
|
|
import java.io.InputStream;
|
|
|
|
import java.net.URL;
|
|
|
|
import java.net.URL;
|
|
|
|
import java.time.LocalDateTime;
|
|
|
|
import java.text.SimpleDateFormat;
|
|
|
|
import java.time.format.DateTimeFormatter;
|
|
|
|
|
|
|
|
import java.util.Date;
|
|
|
|
import java.util.Date;
|
|
|
|
import java.util.UUID;
|
|
|
|
import java.util.UUID;
|
|
|
|
|
|
|
|
|
|
|
|
@Service
|
|
|
|
|
|
|
|
@Slf4j
|
|
|
|
@Slf4j
|
|
|
|
|
|
|
|
@Component
|
|
|
|
public class OssService {
|
|
|
|
public class OssService {
|
|
|
|
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
@Autowired
|
|
|
|
private OSS ossClient;
|
|
|
|
private OSS ossClient;
|
|
|
|
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
@Autowired
|
|
|
|
private OssConfig ossConfig;
|
|
|
|
private OssConfig ossConfig;
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
/**
|
|
|
|
* 上传文件到OSS
|
|
|
|
* 上传文件到OSS
|
|
|
|
|
|
|
|
* @param file 文件
|
|
|
|
|
|
|
|
* @param dir 存储目录
|
|
|
|
|
|
|
|
* @return 返回文件访问URL
|
|
|
|
*/
|
|
|
|
*/
|
|
|
|
public String uploadFile(MultipartFile file) throws IOException {
|
|
|
|
public String uploadFile(MultipartFile file, String dir) {
|
|
|
|
// 生成文件名
|
|
|
|
String bucketName = ossConfig.getBucketName();
|
|
|
|
String originalFilename = file.getOriginalFilename();
|
|
|
|
String urlPrefix = ossConfig.getUrlPrefix();
|
|
|
|
String extension = "";
|
|
|
|
|
|
|
|
if (originalFilename != null && originalFilename.contains(".")) {
|
|
|
|
|
|
|
|
extension = originalFilename.substring(originalFilename.lastIndexOf("."));
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 按日期组织文件夹
|
|
|
|
try {
|
|
|
|
String dateFolder = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy/MM/dd"));
|
|
|
|
// 获取文件原始名称
|
|
|
|
String fileName = dateFolder + "/" + UUID.randomUUID().toString() + extension;
|
|
|
|
String originalFilename = file.getOriginalFilename();
|
|
|
|
|
|
|
|
// 获取文件后缀
|
|
|
|
try (InputStream inputStream = file.getInputStream()) {
|
|
|
|
String suffix = "";
|
|
|
|
// 创建上传请求
|
|
|
|
if (originalFilename != null && originalFilename.contains(".")) {
|
|
|
|
PutObjectRequest putObjectRequest = new PutObjectRequest(
|
|
|
|
suffix = originalFilename.substring(originalFilename.lastIndexOf("."));
|
|
|
|
ossConfig.getBucketName(),
|
|
|
|
}
|
|
|
|
fileName,
|
|
|
|
|
|
|
|
inputStream
|
|
|
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 执行上传
|
|
|
|
// 构建OSS存储路径：目录/日期/随机UUID.后缀
|
|
|
|
ossClient.putObject(putObjectRequest);
|
|
|
|
String dateDir = new SimpleDateFormat("yyyy/MM/dd").format(new Date());
|
|
|
|
|
|
|
|
String filename = UUID.randomUUID().toString().replaceAll("-", "") + suffix;
|
|
|
|
|
|
|
|
String objectName = dir + "/" + dateDir + "/" + filename;
|
|
|
|
|
|
|
|
|
|
|
|
// 返回文件URL
|
|
|
|
// 获取文件输入流
|
|
|
|
String fileUrl = ossConfig.getUrlPrefix() + fileName;
|
|
|
|
InputStream inputStream = file.getInputStream();
|
|
|
|
log.info("文件上传成功: {}", fileUrl);
|
|
|
|
|
|
|
|
return fileUrl;
|
|
|
|
// 上传文件到OSS
|
|
|
|
} catch (Exception e) {
|
|
|
|
ossClient.putObject(bucketName, objectName, inputStream);
|
|
|
|
log.error("文件上传失败: {}", e.getMessage(), e);
|
|
|
|
|
|
|
|
throw new RuntimeException("文件上传失败: " + e.getMessage());
|
|
|
|
// 关闭输入流
|
|
|
|
|
|
|
|
inputStream.close();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 返回文件访问URL
|
|
|
|
|
|
|
|
return urlPrefix + objectName;
|
|
|
|
|
|
|
|
} catch (IOException e) {
|
|
|
|
|
|
|
|
log.error("上传文件到OSS失败: ", e);
|
|
|
|
|
|
|
|
throw new RuntimeException("上传文件失败");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
/**
|
|
|
|
* 删除OSS文件
|
|
|
|
* 从OSS删除文件
|
|
|
|
|
|
|
|
* @param fileUrl 文件URL
|
|
|
|
*/
|
|
|
|
*/
|
|
|
|
public void deleteFile(String fileUrl) {
|
|
|
|
public void deleteFile(String fileUrl) {
|
|
|
|
try {
|
|
|
|
String bucketName = ossConfig.getBucketName();
|
|
|
|
// 从URL中提取对象名称
|
|
|
|
String urlPrefix = ossConfig.getUrlPrefix();
|
|
|
|
String objectName = getObjectNameFromUrl(fileUrl);
|
|
|
|
|
|
|
|
if (objectName != null) {
|
|
|
|
if (fileUrl.startsWith(urlPrefix)) {
|
|
|
|
DeleteObjectRequest deleteObjectRequest = new DeleteObjectRequest(
|
|
|
|
String objectName = fileUrl.substring(urlPrefix.length());
|
|
|
|
ossConfig.getBucketName(),
|
|
|
|
ossClient.deleteObject(bucketName, objectName);
|
|
|
|
objectName
|
|
|
|
|
|
|
|
);
|
|
|
|
|
|
|
|
ossClient.deleteObject(deleteObjectRequest);
|
|
|
|
|
|
|
|
log.info("文件删除成功: {}", fileUrl);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
} catch (Exception e) {
|
|
|
|
|
|
|
|
log.error("文件删除失败: {}", e.getMessage(), e);
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
/**
|
|
|
|
* 生成预签名URL（临时访问链接）
|
|
|
|
* 获取OSS文件输入流
|
|
|
|
|
|
|
|
* @param fileUrl 文件URL
|
|
|
|
|
|
|
|
* @return 文件输入流
|
|
|
|
*/
|
|
|
|
*/
|
|
|
|
public String generatePresignedUrl(String fileUrl) {
|
|
|
|
public InputStream getFileStream(String fileUrl) {
|
|
|
|
|
|
|
|
String bucketName = ossConfig.getBucketName();
|
|
|
|
|
|
|
|
String urlPrefix = ossConfig.getUrlPrefix();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (fileUrl.startsWith(urlPrefix)) {
|
|
|
|
|
|
|
|
String objectName = fileUrl.substring(urlPrefix.length());
|
|
|
|
|
|
|
|
OSSObject ossObject = ossClient.getObject(bucketName, objectName);
|
|
|
|
|
|
|
|
return ossObject.getObjectContent();
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
|
|
|
* 获取文件的临时访问URL（适用于私有bucket）
|
|
|
|
|
|
|
|
* @param objectName 对象名称
|
|
|
|
|
|
|
|
* @param expiration 过期时间（毫秒）
|
|
|
|
|
|
|
|
* @return 带签名的临时URL
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
public String generatePresignedUrl(String objectName, long expiration) {
|
|
|
|
|
|
|
|
String bucketName = ossConfig.getBucketName();
|
|
|
|
|
|
|
|
|
|
|
|
try {
|
|
|
|
try {
|
|
|
|
String objectName = getObjectNameFromUrl(fileUrl);
|
|
|
|
// 计算过期时间
|
|
|
|
if (objectName == null) {
|
|
|
|
Date expirationDate = new Date(System.currentTimeMillis() + expiration);
|
|
|
|
return fileUrl;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 设置URL过期时间为1小时
|
|
|
|
|
|
|
|
Date expiration = new Date(System.currentTimeMillis() + 3600 * 1000);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GeneratePresignedUrlRequest generatePresignedUrlRequest = new GeneratePresignedUrlRequest(
|
|
|
|
// 创建请求
|
|
|
|
ossConfig.getBucketName(),
|
|
|
|
GeneratePresignedUrlRequest request = new GeneratePresignedUrlRequest(bucketName, objectName);
|
|
|
|
objectName
|
|
|
|
request.setExpiration(expirationDate);
|
|
|
|
);
|
|
|
|
|
|
|
|
generatePresignedUrlRequest.setExpiration(expiration);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
URL url = ossClient.generatePresignedUrl(generatePresignedUrlRequest);
|
|
|
|
// 生成URL
|
|
|
|
|
|
|
|
URL url = ossClient.generatePresignedUrl(request);
|
|
|
|
return url.toString();
|
|
|
|
return url.toString();
|
|
|
|
} catch (Exception e) {
|
|
|
|
} catch (Exception e) {
|
|
|
|
log.error("生成预签名URL失败: {}", e.getMessage(), e);
|
|
|
|
log.error("生成临时访问URL失败: ", e);
|
|
|
|
return fileUrl;
|
|
|
|
throw new RuntimeException("生成临时访问URL失败");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
/**
|
|
|
|
* 从URL中提取对象名称
|
|
|
|
* 从完整URL中提取对象名称
|
|
|
|
|
|
|
|
* @param fileUrl 完整的文件URL
|
|
|
|
|
|
|
|
* @return 对象名称
|
|
|
|
*/
|
|
|
|
*/
|
|
|
|
public String getObjectNameFromUrl(String fileUrl) {
|
|
|
|
public String getObjectNameFromUrl(String fileUrl) {
|
|
|
|
if (fileUrl == null || !fileUrl.startsWith(ossConfig.getUrlPrefix())) {
|
|
|
|
String urlPrefix = ossConfig.getUrlPrefix();
|
|
|
|
return null;
|
|
|
|
if (fileUrl != null && fileUrl.startsWith(urlPrefix)) {
|
|
|
|
|
|
|
|
return fileUrl.substring(urlPrefix.length());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return fileUrl.substring(ossConfig.getUrlPrefix().length());
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
