Compare commits
2 Commits
93d4a0d506
...
b0fa4279b0
| Author | SHA1 | Date | |
|---|---|---|---|
| b0fa4279b0 | |||
| fdc159de7e |
@@ -111,9 +111,10 @@ export function useUpload() {
|
||||
// 开发环境:使用 /oss 代理避免CORS问题
|
||||
// 生产环境:使用自定义域名(签名基于路径,域名可替换)
|
||||
const uploadUrl = isDev()
|
||||
? presignedData.presignedUrl.replace(OSS_ORIGINAL, '/oss')
|
||||
: presignedData.presignedUrl.replace(OSS_ORIGINAL, OSS_DOMAIN)
|
||||
? presignedData.presignedUrl.replace(OSS_DOMAIN, '/oss')
|
||||
: presignedData.presignedUrl
|
||||
xhr.open('PUT', uploadUrl)
|
||||
// 设置 Content-Type header(必须与签名时一致)
|
||||
if (presignedData.headers && presignedData.headers['Content-Type']) {
|
||||
xhr.setRequestHeader('Content-Type', presignedData.headers['Content-Type'])
|
||||
}
|
||||
@@ -169,12 +170,13 @@ export function useUpload() {
|
||||
duration = await getVideoDuration(file)
|
||||
}
|
||||
|
||||
// 第一步:获取预签名URL
|
||||
// 第一步:获取预签名URL(传递文件的 MIME 类型)
|
||||
const presignedData = await MaterialService.getPresignedUrl({
|
||||
fileName: file.name,
|
||||
fileCategory,
|
||||
groupId,
|
||||
fileSize: file.size
|
||||
fileSize: file.size,
|
||||
contentType: file.type || undefined // 传递文件的 MIME 类型
|
||||
})
|
||||
|
||||
// 第二步:直传文件到OSS
|
||||
|
||||
@@ -109,6 +109,12 @@
|
||||
<groupId>software.amazon.awssdk</groupId>
|
||||
<artifactId>s3</artifactId>
|
||||
</dependency>
|
||||
<!-- 阿里云 OSS SDK:用于生成阿里云签名 URL -->
|
||||
<dependency>
|
||||
<groupId>com.aliyun.oss</groupId>
|
||||
<artifactId>aliyun-sdk-oss</artifactId>
|
||||
<version>3.18.1</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.apache.tika</groupId>
|
||||
|
||||
@@ -52,6 +52,17 @@ public interface FileClient {
|
||||
throw new UnsupportedOperationException("不支持的操作");
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得文件预签名地址,用于上传(包含 Content-Type 签名)
|
||||
*
|
||||
* @param path 相对路径
|
||||
* @param contentType 文件的 Content-Type
|
||||
* @return 文件预签名地址
|
||||
*/
|
||||
default String presignPutUrl(String path, String contentType) {
|
||||
throw new UnsupportedOperationException("不支持的操作");
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成文件预签名地址,用于读取
|
||||
*
|
||||
|
||||
@@ -7,6 +7,11 @@ import cn.hutool.core.util.URLUtil;
|
||||
import cn.hutool.http.HttpUtil;
|
||||
import cn.iocoder.yudao.framework.common.util.http.HttpUtils;
|
||||
import cn.iocoder.yudao.module.infra.framework.file.core.client.AbstractFileClient;
|
||||
import com.aliyun.oss.OSS;
|
||||
import com.aliyun.oss.OSSClient;
|
||||
import com.aliyun.oss.HttpMethod;
|
||||
import com.aliyun.oss.model.OSSObject;
|
||||
import com.aliyun.oss.model.ObjectMetadata;
|
||||
import software.amazon.awssdk.auth.credentials.AwsBasicCredentials;
|
||||
import software.amazon.awssdk.auth.credentials.AwsCredentialsProvider;
|
||||
import software.amazon.awssdk.auth.credentials.StaticCredentialsProvider;
|
||||
@@ -16,15 +21,16 @@ import software.amazon.awssdk.services.s3.S3Client;
|
||||
import software.amazon.awssdk.services.s3.S3Configuration;
|
||||
import software.amazon.awssdk.services.s3.model.DeleteObjectRequest;
|
||||
import software.amazon.awssdk.services.s3.model.GetObjectRequest;
|
||||
import software.amazon.awssdk.services.s3.model.PutObjectRequest;
|
||||
import software.amazon.awssdk.services.s3.presigner.S3Presigner;
|
||||
import software.amazon.awssdk.services.s3.presigner.model.GetObjectPresignRequest;
|
||||
import software.amazon.awssdk.services.s3.presigner.model.PutObjectPresignRequest;
|
||||
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.net.URI;
|
||||
import java.net.URL;
|
||||
import java.time.Duration;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.time.Duration;
|
||||
import java.util.Date;
|
||||
|
||||
import org.springframework.web.util.UriUtils;
|
||||
|
||||
@@ -36,9 +42,15 @@ import org.springframework.web.util.UriUtils;
|
||||
public class S3FileClient extends AbstractFileClient<S3FileClientConfig> {
|
||||
|
||||
private static final Duration EXPIRATION_DEFAULT = Duration.ofHours(24);
|
||||
private static final String DEFAULT_REGION = "us-east-1";
|
||||
|
||||
private S3Client client;
|
||||
private S3Presigner presigner;
|
||||
/** S3 客户端(非阿里云使用) */
|
||||
private S3Client s3Client;
|
||||
private S3Presigner s3Presigner;
|
||||
/** 阿里云 OSS 客户端(阿里云专用) */
|
||||
private OSS aliyunOssClient;
|
||||
/** 是否为阿里云 */
|
||||
private boolean isAliyun;
|
||||
|
||||
public S3FileClient(Long id, S3FileClientConfig config) {
|
||||
super(id, config);
|
||||
@@ -50,22 +62,47 @@ public class S3FileClient extends AbstractFileClient<S3FileClientConfig> {
|
||||
if (StrUtil.isEmpty(config.getDomain())) {
|
||||
config.setDomain(buildDomain());
|
||||
}
|
||||
// 初始化 S3 客户端
|
||||
Region region = Region.of("us-east-1"); // 必须填,但填什么都行,常见的值有 "us-east-1",不填会报错
|
||||
|
||||
// 判断是否为阿里云
|
||||
isAliyun = StrUtil.contains(config.getEndpoint(), S3FileClientConfig.ENDPOINT_ALIYUN);
|
||||
|
||||
if (isAliyun) {
|
||||
// 阿里云:使用阿里云 OSS SDK
|
||||
initAliyunClient();
|
||||
} else {
|
||||
// 其他云厂商:使用 AWS S3 SDK
|
||||
initS3Client();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 初始化阿里云 OSS 客户端
|
||||
*/
|
||||
private void initAliyunClient() {
|
||||
String region = extractRegionFromEndpoint(config.getEndpoint());
|
||||
String endpoint = "https://oss-" + region + ".aliyuncs.com";
|
||||
aliyunOssClient = new OSSClient(endpoint, config.getAccessKey(), config.getAccessSecret());
|
||||
}
|
||||
|
||||
/**
|
||||
* 初始化 AWS S3 客户端
|
||||
*/
|
||||
private void initS3Client() {
|
||||
Region region = Region.of(extractRegionFromEndpoint(config.getEndpoint()));
|
||||
AwsCredentialsProvider credentialsProvider = StaticCredentialsProvider.create(
|
||||
AwsBasicCredentials.create(config.getAccessKey(), config.getAccessSecret()));
|
||||
URI endpoint = URI.create(buildEndpoint());
|
||||
S3Configuration serviceConfiguration = S3Configuration.builder() // Path-style 访问
|
||||
S3Configuration serviceConfiguration = S3Configuration.builder()
|
||||
.pathStyleAccessEnabled(Boolean.TRUE.equals(config.getEnablePathStyleAccess()))
|
||||
.chunkedEncodingEnabled(false) // 禁用分块编码,参见 https://t.zsxq.com/kBy57
|
||||
.chunkedEncodingEnabled(false)
|
||||
.build();
|
||||
client = S3Client.builder()
|
||||
s3Client = S3Client.builder()
|
||||
.credentialsProvider(credentialsProvider)
|
||||
.region(region)
|
||||
.endpointOverride(endpoint)
|
||||
.serviceConfiguration(serviceConfiguration)
|
||||
.build();
|
||||
presigner = S3Presigner.builder()
|
||||
s3Presigner = S3Presigner.builder()
|
||||
.credentialsProvider(credentialsProvider)
|
||||
.region(region)
|
||||
.endpointOverride(endpoint)
|
||||
@@ -75,125 +112,196 @@ public class S3FileClient extends AbstractFileClient<S3FileClientConfig> {
|
||||
|
||||
@Override
|
||||
public String upload(byte[] content, String path, String type) {
|
||||
// 构造 PutObjectRequest
|
||||
PutObjectRequest putRequest = PutObjectRequest.builder()
|
||||
if (isAliyun) {
|
||||
ObjectMetadata metadata = new ObjectMetadata();
|
||||
metadata.setContentType(type);
|
||||
metadata.setContentLength(content.length);
|
||||
com.aliyun.oss.model.PutObjectRequest putRequest = new com.aliyun.oss.model.PutObjectRequest(
|
||||
config.getBucket(), path, new ByteArrayInputStream(content), metadata);
|
||||
aliyunOssClient.putObject(putRequest);
|
||||
} else {
|
||||
software.amazon.awssdk.services.s3.model.PutObjectRequest putRequest = software.amazon.awssdk.services.s3.model.PutObjectRequest.builder()
|
||||
.bucket(config.getBucket())
|
||||
.key(path)
|
||||
.contentType(type)
|
||||
.contentLength((long) content.length)
|
||||
.build();
|
||||
// 上传文件
|
||||
client.putObject(putRequest, RequestBody.fromBytes(content));
|
||||
// 拼接返回路径
|
||||
s3Client.putObject(putRequest, RequestBody.fromBytes(content));
|
||||
}
|
||||
return presignGetUrl(path, null);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void delete(String path) {
|
||||
if (isAliyun) {
|
||||
aliyunOssClient.deleteObject(config.getBucket(), path);
|
||||
} else {
|
||||
DeleteObjectRequest deleteRequest = DeleteObjectRequest.builder()
|
||||
.bucket(config.getBucket())
|
||||
.key(path)
|
||||
.build();
|
||||
client.deleteObject(deleteRequest);
|
||||
s3Client.deleteObject(deleteRequest);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public byte[] getContent(String path) {
|
||||
if (isAliyun) {
|
||||
OSSObject ossObject = aliyunOssClient.getObject(config.getBucket(), path);
|
||||
return IoUtil.readBytes(ossObject.getObjectContent());
|
||||
} else {
|
||||
GetObjectRequest getRequest = GetObjectRequest.builder()
|
||||
.bucket(config.getBucket())
|
||||
.key(path)
|
||||
.build();
|
||||
return IoUtil.readBytes(client.getObject(getRequest));
|
||||
return IoUtil.readBytes(s3Client.getObject(getRequest));
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String presignPutUrl(String path) {
|
||||
return presigner.presignPutObject(PutObjectPresignRequest.builder()
|
||||
return presignPutUrl(path, null);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String presignPutUrl(String path, String contentType) {
|
||||
String signedUrl;
|
||||
if (isAliyun) {
|
||||
Date expirationDate = new Date(System.currentTimeMillis() + EXPIRATION_DEFAULT.toMillis());
|
||||
// 使用 GeneratePresignedUrlRequest 以支持设置 Content-Type
|
||||
com.aliyun.oss.model.GeneratePresignedUrlRequest request =
|
||||
new com.aliyun.oss.model.GeneratePresignedUrlRequest(config.getBucket(), path, HttpMethod.PUT);
|
||||
request.setExpiration(expirationDate);
|
||||
// 如果指定了 Content-Type,则包含在签名中
|
||||
if (StrUtil.isNotBlank(contentType)) {
|
||||
request.setContentType(contentType);
|
||||
}
|
||||
URL url = aliyunOssClient.generatePresignedUrl(request);
|
||||
signedUrl = url.toString();
|
||||
} else {
|
||||
software.amazon.awssdk.services.s3.model.PutObjectRequest.Builder putRequestBuilder =
|
||||
software.amazon.awssdk.services.s3.model.PutObjectRequest.builder()
|
||||
.bucket(config.getBucket())
|
||||
.key(path);
|
||||
// 如果指定了 Content-Type,则包含在签名中
|
||||
if (StrUtil.isNotBlank(contentType)) {
|
||||
putRequestBuilder.contentType(contentType);
|
||||
}
|
||||
signedUrl = s3Presigner.presignPutObject(PutObjectPresignRequest.builder()
|
||||
.signatureDuration(EXPIRATION_DEFAULT)
|
||||
.putObjectRequest(b -> b.bucket(config.getBucket()).key(path)).build())
|
||||
.putObjectRequest(putRequestBuilder.build()).build())
|
||||
.url().toString();
|
||||
}
|
||||
// 直传使用 OSS 原始域名,不替换为 CDN(CDN 可能不支持签名直传)
|
||||
return signedUrl;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String presignGetUrl(String url, Integer expirationSeconds) {
|
||||
// 1. 将 url 转换为 path(支持多种URL格式)
|
||||
String path = extractPathFromUrl(url);
|
||||
// 1. 将 url 转换为 path
|
||||
String path = StrUtil.removePrefix(url, config.getDomain() + "/");
|
||||
path = HttpUtils.removeUrlQuery(path);
|
||||
String decodedPath = URLUtil.decode(path, StandardCharsets.UTF_8);
|
||||
|
||||
// 2.1 情况一:公开访问:无需签名
|
||||
// 考虑到老版本的兼容,所以必须是 config.getEnablePublicAccess() 为 false 时,才进行签名
|
||||
// 2. 公开访问:无需签名
|
||||
if (!BooleanUtil.isFalse(config.getEnablePublicAccess())) {
|
||||
String encodedPath = UriUtils.encodePath(decodedPath, StandardCharsets.UTF_8);
|
||||
return config.getDomain() + "/" + encodedPath;
|
||||
}
|
||||
|
||||
// 2.2 情况二:私有访问:生成 GET 预签名 URL
|
||||
String finalPath = decodedPath;
|
||||
Duration expiration = expirationSeconds != null ? Duration.ofSeconds(expirationSeconds) : EXPIRATION_DEFAULT;
|
||||
URL signedUrl = presigner.presignGetObject(GetObjectPresignRequest.builder()
|
||||
.signatureDuration(expiration)
|
||||
.getObjectRequest(b -> b.bucket(config.getBucket()).key(finalPath)).build())
|
||||
.url();
|
||||
return signedUrl.toString();
|
||||
// 3. 私有访问:生成预签名 URL
|
||||
int expiration = expirationSeconds != null ? expirationSeconds : (int) EXPIRATION_DEFAULT.getSeconds();
|
||||
|
||||
String signedUrl;
|
||||
if (isAliyun) {
|
||||
Date expirationDate = new Date(System.currentTimeMillis() + expiration * 1000L);
|
||||
signedUrl = aliyunOssClient.generatePresignedUrl(config.getBucket(), decodedPath, expirationDate).toString();
|
||||
} else {
|
||||
Duration duration = Duration.ofSeconds(expiration);
|
||||
signedUrl = s3Presigner.presignGetObject(GetObjectPresignRequest.builder()
|
||||
.signatureDuration(duration)
|
||||
.getObjectRequest(b -> b.bucket(config.getBucket()).key(decodedPath)).build())
|
||||
.url().toString();
|
||||
}
|
||||
|
||||
// 4. 替换为 CDN 域名
|
||||
if (StrUtil.isNotEmpty(config.getDomain())) {
|
||||
return replaceUrlHost(signedUrl, config.getDomain());
|
||||
}
|
||||
return signedUrl;
|
||||
}
|
||||
|
||||
/**
|
||||
* 从URL中提取相对路径
|
||||
* 支持多种URL格式:
|
||||
* 1. CDN域名:https://oss.muyetools.cn/path/to/file
|
||||
* 2. OSS原始域名:https://bucket.oss-region.aliyuncs.com/path/to/file
|
||||
* 3. 相对路径:path/to/file
|
||||
* 替换 URL 的 host 部分为目标域名
|
||||
*
|
||||
* @param url 原始 URL
|
||||
* @param targetDomain 目标域名(含协议)
|
||||
* @return 替换后的 URL
|
||||
*/
|
||||
private String extractPathFromUrl(String url) {
|
||||
if (StrUtil.isEmpty(url)) {
|
||||
return url;
|
||||
private String replaceUrlHost(String url, String targetDomain) {
|
||||
try {
|
||||
URI uri = new URI(url);
|
||||
String originalHost = uri.getScheme() + "://" + uri.getHost();
|
||||
if (uri.getPort() > 0) {
|
||||
originalHost += ":" + uri.getPort();
|
||||
}
|
||||
|
||||
// 1. 尝试移除配置的域名前缀
|
||||
String path = StrUtil.removePrefix(url, config.getDomain() + "/");
|
||||
if (!StrUtil.equals(url, path)) {
|
||||
return path;
|
||||
return StrUtil.replace(url, originalHost, targetDomain);
|
||||
} catch (Exception e) {
|
||||
// 降级处理:使用 buildDomain()
|
||||
return StrUtil.replace(url, buildDomain(), targetDomain);
|
||||
}
|
||||
|
||||
// 2. 如果不是以配置域名开头,检查是否是完整URL
|
||||
if (url.contains("://")) {
|
||||
// 提取域名后的路径
|
||||
int pathStart = url.indexOf("/", url.indexOf("://") + 3);
|
||||
if (pathStart > 0) {
|
||||
return url.substring(pathStart + 1);
|
||||
}
|
||||
}
|
||||
|
||||
// 3. 已经是相对路径,直接返回
|
||||
return url;
|
||||
}
|
||||
|
||||
/**
|
||||
* 基于 bucket + endpoint 构建访问的 Domain 地址
|
||||
*
|
||||
* @return Domain 地址
|
||||
*/
|
||||
private String buildDomain() {
|
||||
// 如果已经是 http 或者 https,则不进行拼接.主要适配 MinIO
|
||||
if (HttpUtil.isHttp(config.getEndpoint()) || HttpUtil.isHttps(config.getEndpoint())) {
|
||||
return StrUtil.format("{}/{}", config.getEndpoint(), config.getBucket());
|
||||
}
|
||||
// 阿里云、腾讯云、华为云都适合。七牛云比较特殊,必须有自定义域名
|
||||
return StrUtil.format("https://{}.{}", config.getBucket(), config.getEndpoint());
|
||||
}
|
||||
|
||||
/**
|
||||
* 节点地址补全协议头
|
||||
*
|
||||
* @return 节点地址
|
||||
*/
|
||||
private String buildEndpoint() {
|
||||
// 如果已经是 http 或者 https,则不进行拼接
|
||||
if (HttpUtil.isHttp(config.getEndpoint()) || HttpUtil.isHttps(config.getEndpoint())) {
|
||||
return config.getEndpoint();
|
||||
}
|
||||
return StrUtil.format("https://{}", config.getEndpoint());
|
||||
}
|
||||
|
||||
/**
|
||||
* 从 endpoint 中提取 region
|
||||
*/
|
||||
private String extractRegionFromEndpoint(String endpoint) {
|
||||
if (StrUtil.isEmpty(endpoint)) {
|
||||
return DEFAULT_REGION;
|
||||
}
|
||||
|
||||
// 移除协议前缀和 bucket 前缀
|
||||
String host = StrUtil.removePrefixIgnoreCase(endpoint, "https://");
|
||||
host = StrUtil.removePrefixIgnoreCase(host, "http://");
|
||||
host = StrUtil.removePrefix(host, config.getBucket() + ".");
|
||||
|
||||
// 阿里云 OSS
|
||||
if (host.contains(S3FileClientConfig.ENDPOINT_ALIYUN)) {
|
||||
String prefix = StrUtil.subBefore(host, "." + S3FileClientConfig.ENDPOINT_ALIYUN, false);
|
||||
if (StrUtil.isNotEmpty(prefix)) {
|
||||
return StrUtil.removePrefix(prefix, "oss-");
|
||||
}
|
||||
}
|
||||
|
||||
// 腾讯云 COS
|
||||
if (host.contains(S3FileClientConfig.ENDPOINT_TENCENT)) {
|
||||
String region = StrUtil.subBefore(host, "." + S3FileClientConfig.ENDPOINT_TENCENT, false);
|
||||
if (StrUtil.isNotEmpty(region) && region.startsWith("cos.")) {
|
||||
return region.substring(4);
|
||||
}
|
||||
}
|
||||
|
||||
return DEFAULT_REGION;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -100,12 +100,14 @@ public class AppTikUserFileController {
|
||||
@Parameter(name = "fileCategory", description = "文件分类(video/audio/image)", required = true, example = "video")
|
||||
@Parameter(name = "groupId", description = "分组编号(可选)")
|
||||
@Parameter(name = "fileSize", description = "文件大小(字节)")
|
||||
@Parameter(name = "contentType", description = "文件 MIME 类型(可选,如 video/mp4)")
|
||||
public CommonResult<Object> getPresignedUrl(
|
||||
@RequestParam("fileName") String fileName,
|
||||
@RequestParam("fileCategory") String fileCategory,
|
||||
@RequestParam(value = "groupId", required = false) Long groupId,
|
||||
@RequestParam(value = "fileSize", required = false) Long fileSize) {
|
||||
return success(userFileService.getPresignedUrl(fileName, fileCategory, groupId, fileSize));
|
||||
@RequestParam(value = "fileSize", required = false) Long fileSize,
|
||||
@RequestParam(value = "contentType", required = false) String contentType) {
|
||||
return success(userFileService.getPresignedUrl(fileName, fileCategory, groupId, fileSize, contentType));
|
||||
}
|
||||
|
||||
@PostMapping("/complete-upload")
|
||||
|
||||
@@ -82,9 +82,10 @@ public interface TikUserFileService {
|
||||
* @param fileCategory 文件分类
|
||||
* @param groupId 分组编号(可选)
|
||||
* @param fileSize 文件大小(可选)
|
||||
* @param contentType 文件 MIME 类型(可选)
|
||||
* @return 预签名URL信息
|
||||
*/
|
||||
Object getPresignedUrl(String fileName, String fileCategory, Long groupId, Long fileSize);
|
||||
Object getPresignedUrl(String fileName, String fileCategory, Long groupId, Long fileSize, String contentType);
|
||||
|
||||
/**
|
||||
* 确认上传(直传模式)
|
||||
|
||||
@@ -53,24 +53,8 @@ import static cn.iocoder.yudao.module.tik.enums.ErrorCodeConstants.*;
|
||||
@Slf4j
|
||||
public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
|
||||
/** 预签名URL过期时间(24小时,单位:秒) */
|
||||
private static final int PRESIGN_URL_EXPIRATION_SECONDS = 24 * 60 * 60;
|
||||
|
||||
/**
|
||||
* 上传文件的前缀,是否包含日期
|
||||
*
|
||||
* 目的:按照日期,进行分目录
|
||||
*/
|
||||
private static boolean PATH_PREFIX_DATE_ENABLE = true;
|
||||
|
||||
/**
|
||||
* 上传文件的后缀,是否包含时间戳
|
||||
*
|
||||
* 目的:保证文件的唯一性,避免覆盖
|
||||
* 定制:可按需调整成 UUID、或者其他方式
|
||||
*/
|
||||
private static boolean PATH_SUFFIX_TIMESTAMP_ENABLE = true;
|
||||
|
||||
@Resource
|
||||
private TikUserFileMapper userFileMapper;
|
||||
|
||||
@@ -93,25 +77,18 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
public Long uploadFile(MultipartFile file, String fileCategory, String coverBase64, Integer duration, Long groupId) {
|
||||
Long userId = SecurityFrameworkUtils.getLoginUserId();
|
||||
|
||||
// ========== 第一阶段:校验和准备 ==========
|
||||
// 1. 校验文件分类和配额
|
||||
// 校验文件分类和配额
|
||||
validateUploadRequest(fileCategory, file.getSize());
|
||||
|
||||
// 2. 获取OSS基础目录路径
|
||||
// 获取OSS基础目录路径
|
||||
String baseDirectory = ossInitService.getOssDirectoryByCategory(userId, fileCategory);
|
||||
|
||||
// ========== 第二阶段:处理文件和上传 ==========
|
||||
// 处理文件和上传
|
||||
UploadFileContext context = processFileAndUpload(file, baseDirectory);
|
||||
|
||||
// ========== 第三阶段:保存数据库 ==========
|
||||
try {
|
||||
// 保存数据库
|
||||
return saveFileRecord(userId, file, fileCategory, context.fileUrl, context.filePath,
|
||||
coverBase64, baseDirectory, context.infraFileId, duration, groupId);
|
||||
} catch (Exception e) {
|
||||
log.error("[uploadFile][保存数据库失败]", e);
|
||||
deleteOssFile(context.infraFileId, context.filePath, context.fileUrl);
|
||||
throw e;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -143,12 +120,11 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
}
|
||||
|
||||
/**
|
||||
* 统一处理文件和上传逻辑
|
||||
* 处理文件和上传逻辑
|
||||
*/
|
||||
private UploadFileContext processFileAndUpload(MultipartFile file, String baseDirectory) {
|
||||
UploadFileContext context = new UploadFileContext();
|
||||
|
||||
// 1. 读取文件内容
|
||||
byte[] fileContent;
|
||||
try {
|
||||
fileContent = IoUtil.readBytes(file.getInputStream());
|
||||
@@ -158,12 +134,10 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
}
|
||||
|
||||
try {
|
||||
// 2. 处理文件名和类型
|
||||
String fileName = file.getOriginalFilename();
|
||||
String fileType = file.getContentType();
|
||||
if (StrUtil.isEmpty(fileType)) {
|
||||
fileType = FileTypeUtils.getMineType(fileContent, fileName);
|
||||
}
|
||||
String fileType = StrUtil.isEmpty(file.getContentType())
|
||||
? FileTypeUtils.getMineType(fileContent, fileName)
|
||||
: file.getContentType();
|
||||
if (StrUtil.isEmpty(fileName)) {
|
||||
fileName = DigestUtil.sha256Hex(fileContent);
|
||||
}
|
||||
@@ -174,15 +148,12 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
}
|
||||
}
|
||||
|
||||
// 3. 生成上传路径
|
||||
String filePath = generateUploadPath(fileName, baseDirectory);
|
||||
|
||||
// 4. 上传到OSS
|
||||
FileClient client = fileConfigService.getMasterFileClient();
|
||||
Assert.notNull(client, "客户端(master) 不能为空");
|
||||
String presignedUrl = client.upload(fileContent, filePath, fileType);
|
||||
|
||||
// 5. 保存到 infra_file 表
|
||||
String fileUrl = HttpUtils.removeUrlQuery(presignedUrl);
|
||||
FileDO infraFile = new FileDO()
|
||||
.setConfigId(client.getId())
|
||||
@@ -193,7 +164,6 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
.setSize((int) file.getSize());
|
||||
fileMapper.insert(infraFile);
|
||||
|
||||
// 6. 设置上下文
|
||||
context.fileUrl = fileUrl;
|
||||
context.filePath = filePath;
|
||||
context.infraFileId = infraFile.getId();
|
||||
@@ -210,13 +180,12 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
}
|
||||
|
||||
/**
|
||||
* 统一处理视频封面上传
|
||||
* 处理视频封面上传
|
||||
*/
|
||||
private String handleCoverUpload(String coverBase64, String fileName, String fileType, String baseDirectory) {
|
||||
if (StrUtil.isBlank(coverBase64) || !StrUtil.containsIgnoreCase(fileType, "video")) {
|
||||
return null;
|
||||
}
|
||||
|
||||
try {
|
||||
String base64Data = coverBase64.contains(",")
|
||||
? coverBase64.substring(coverBase64.indexOf(",") + 1)
|
||||
@@ -234,102 +203,62 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存文件记录到数据库(在事务中执行)
|
||||
* 保存文件记录到数据库
|
||||
*/
|
||||
@Transactional(rollbackFor = Exception.class)
|
||||
public Long saveFileRecord(Long userId, MultipartFile file, String fileCategory,
|
||||
String fileUrl, String filePath, String coverBase64, String baseDirectory, Long infraFileId, Integer duration, Long groupId) {
|
||||
// 验证 infraFileId 不为空
|
||||
String fileUrl, String filePath, String coverBase64,
|
||||
String baseDirectory, Long infraFileId, Integer duration, Long groupId) {
|
||||
if (infraFileId == null) {
|
||||
log.error("[saveFileRecord][infra_file.id 为空,无法保存文件记录,用户({}),URL({})]", userId, fileUrl);
|
||||
log.error("[saveFileRecord][infra_file.id 为空,用户({}),URL({})]", userId, fileUrl);
|
||||
throw exception(FILE_NOT_EXISTS, "文件记录保存失败:无法获取文件ID");
|
||||
}
|
||||
|
||||
// 处理视频封面和文件名
|
||||
String fileName = file.getOriginalFilename();
|
||||
String fileType = file.getContentType();
|
||||
String coverUrl = handleCoverUpload(coverBase64, fileName, fileType, baseDirectory);
|
||||
// 处理显示名称(去除文件后缀)
|
||||
String displayName = FileUtil.mainName(fileName);
|
||||
|
||||
// 9. 创建文件记录(保存完整路径,包含封面URL和Base64)
|
||||
TikUserFileDO userFile = new TikUserFileDO()
|
||||
.setUserId(userId)
|
||||
.setFileId(infraFileId) // 关联infra_file表,用于后续通过FileService管理文件
|
||||
.setFileName(fileName) // 保留原始文件名(系统标识)
|
||||
.setDisplayName(displayName) // 设置显示名称(无后缀,用户可编辑)
|
||||
.setFileId(infraFileId)
|
||||
.setFileName(fileName)
|
||||
.setDisplayName(displayName)
|
||||
.setFileType(fileType)
|
||||
.setFileCategory(fileCategory)
|
||||
.setFileSize(file.getSize())
|
||||
.setFileUrl(fileUrl)
|
||||
.setFilePath(filePath) // 保存完整的OSS路径(由FileService生成)
|
||||
.setCoverUrl(coverUrl) // 设置封面URL(如果有)
|
||||
.setCoverBase64(StrUtil.isNotBlank(coverBase64) ? coverBase64 : null) // 保存原始base64数据(如果有)
|
||||
.setDuration(duration) // 设置视频时长(如果有)
|
||||
.setGroupId(groupId); // 设置分组编号(如果有)
|
||||
.setFilePath(filePath)
|
||||
.setCoverUrl(coverUrl)
|
||||
.setCoverBase64(StrUtil.isNotBlank(coverBase64) ? coverBase64 : null)
|
||||
.setDuration(duration)
|
||||
.setGroupId(groupId);
|
||||
|
||||
userFileMapper.insert(userFile);
|
||||
|
||||
// 10. 更新配额
|
||||
memberUserProfileService.increaseUsedStorage(String.valueOf(userId), file.getSize());
|
||||
|
||||
log.info("[saveFileRecord][用户({})保存文件记录成功,文件编号({}),infra文件编号({})]",
|
||||
log.info("[saveFileRecord][用户({})保存文件成功,userFileId({}),infraFileId({})]",
|
||||
userId, userFile.getId(), infraFileId);
|
||||
// 返回 infra_file.id,保持与现有配音功能的兼容性
|
||||
return infraFileId;
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除OSS文件(当数据库保存失败时调用)
|
||||
*/
|
||||
private void deleteOssFile(Long infraFileId, String filePath, String fileUrl) {
|
||||
try {
|
||||
if (infraFileId != null) {
|
||||
// 优先通过 infra_file.id 删除(更准确)
|
||||
// 注意:这里需要调用 FileService.deleteFile,但 FileApi 没有提供删除方法
|
||||
// 暂时通过 filePath 删除
|
||||
log.warn("[deleteOssFile][通过 infra_file.id 删除文件,但 FileApi 未提供删除方法,id({})]", infraFileId);
|
||||
}
|
||||
|
||||
// 通过 filePath 删除(需要获取 FileClient)
|
||||
// TODO: 实现通过 filePath 删除OSS文件的逻辑
|
||||
// 可以通过 FileService 或 FileClient 删除
|
||||
log.warn("[deleteOssFile][准备删除OSS文件,路径({}),URL({})]", filePath, fileUrl);
|
||||
} catch (Exception e) {
|
||||
// 删除OSS文件失败不影响主流程,只记录日志
|
||||
log.error("[deleteOssFile][删除OSS文件失败,路径({}),URL({})]", filePath, fileUrl, e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成上传路径(与 FileService 保持一致)
|
||||
* 生成上传路径
|
||||
* 格式:{directory}/{yyyyMMdd}/{filename}_{timestamp}.ext
|
||||
*/
|
||||
private String generateUploadPath(String name, String directory) {
|
||||
// 1. 生成前缀、后缀
|
||||
String prefix = null;
|
||||
if (PATH_PREFIX_DATE_ENABLE) {
|
||||
prefix = LocalDateTimeUtil.format(LocalDateTimeUtil.now(), PURE_DATE_PATTERN);
|
||||
}
|
||||
String suffix = null;
|
||||
if (PATH_SUFFIX_TIMESTAMP_ENABLE) {
|
||||
suffix = String.valueOf(System.currentTimeMillis());
|
||||
}
|
||||
String prefix = LocalDateTimeUtil.format(LocalDateTimeUtil.now(), PURE_DATE_PATTERN);
|
||||
String suffix = String.valueOf(System.currentTimeMillis());
|
||||
|
||||
// 2.1 先拼接 suffix 后缀
|
||||
if (StrUtil.isNotEmpty(suffix)) {
|
||||
// 拼接后缀
|
||||
String ext = FileUtil.extName(name);
|
||||
if (StrUtil.isNotEmpty(ext)) {
|
||||
name = FileUtil.mainName(name) + "_" + suffix + "." + ext;
|
||||
} else {
|
||||
name = name + "_" + suffix;
|
||||
}
|
||||
}
|
||||
// 2.2 再拼接 prefix 前缀
|
||||
if (StrUtil.isNotEmpty(prefix)) {
|
||||
// 拼接前缀和目录
|
||||
name = prefix + "/" + name;
|
||||
}
|
||||
// 2.3 最后拼接 directory 目录
|
||||
if (StrUtil.isNotEmpty(directory)) {
|
||||
name = directory + "/" + name;
|
||||
}
|
||||
@@ -338,51 +267,36 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
|
||||
@Override
|
||||
public PageResult<AppTikUserFileRespVO> getFilePage(AppTikUserFilePageReqVO pageReqVO) {
|
||||
// 自动填充当前登录用户ID
|
||||
Long userId = SecurityFrameworkUtils.getLoginUserId();
|
||||
pageReqVO.setUserId(userId);
|
||||
pageReqVO.setUserId(SecurityFrameworkUtils.getLoginUserId());
|
||||
|
||||
// 查询文件列表
|
||||
PageResult<TikUserFileDO> pageResult = userFileMapper.selectPage(pageReqVO);
|
||||
|
||||
// 转换为VO并生成预览URL
|
||||
return CollectionUtils.convertPage(pageResult, file -> {
|
||||
AppTikUserFileRespVO vo = BeanUtils.toBean(file, AppTikUserFileRespVO.class);
|
||||
|
||||
// 判断文件类型
|
||||
boolean isVideo = StrUtil.containsIgnoreCase(file.getFileType(), "video");
|
||||
boolean isImage = FileTypeUtils.isImage(file.getFileType());
|
||||
vo.setIsVideo(isVideo);
|
||||
vo.setIsImage(isImage);
|
||||
|
||||
// 视频文件不使用任何OSS预签名URL(包括coverUrl和previewUrl)
|
||||
// 只返回coverBase64(如果有),前端从本地缓存获取
|
||||
String coverUrlPresigned = null;
|
||||
String thumbnailUrlPresigned = null;
|
||||
String previewUrl = null;
|
||||
// 视频文件不生成OSS预签名URL(前端使用coverBase64缓存)
|
||||
if (isVideo) {
|
||||
vo.setCoverUrl(null);
|
||||
vo.setThumbnailUrl(null);
|
||||
vo.setPreviewUrl(null);
|
||||
return vo;
|
||||
}
|
||||
|
||||
// 非视频文件才生成预签名URL
|
||||
if (!isVideo) {
|
||||
// 图片文件:生成缩略图URL
|
||||
thumbnailUrlPresigned = StrUtil.isNotBlank(file.getThumbnailUrl())
|
||||
// 非视频文件生成预签名URL
|
||||
String thumbnailUrl = StrUtil.isNotBlank(file.getThumbnailUrl())
|
||||
? getCachedPresignUrl(file.getThumbnailUrl(), PRESIGN_URL_EXPIRATION_SECONDS)
|
||||
: null;
|
||||
vo.setCoverUrl(null);
|
||||
vo.setThumbnailUrl(thumbnailUrl);
|
||||
|
||||
// 生成预览URL
|
||||
if (isImage) {
|
||||
// 图片:优先使用缩略图,没有缩略图时使用原图
|
||||
previewUrl = thumbnailUrlPresigned != null
|
||||
? thumbnailUrlPresigned
|
||||
: getCachedPresignUrl(file.getFileUrl(), PRESIGN_URL_EXPIRATION_SECONDS);
|
||||
}
|
||||
}
|
||||
|
||||
// 设置封面和缩略图URL
|
||||
vo.setCoverUrl(coverUrlPresigned);
|
||||
vo.setThumbnailUrl(thumbnailUrlPresigned);
|
||||
|
||||
// 生成预览URL
|
||||
vo.setPreviewUrl(previewUrl);
|
||||
// 图片预览URL:优先缩略图,否则原图
|
||||
vo.setPreviewUrl(isImage
|
||||
? (thumbnailUrl != null ? thumbnailUrl : getCachedPresignUrl(file.getFileUrl(), PRESIGN_URL_EXPIRATION_SECONDS))
|
||||
: null);
|
||||
|
||||
return vo;
|
||||
});
|
||||
@@ -396,44 +310,16 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
}
|
||||
Long userId = SecurityFrameworkUtils.getLoginUserId();
|
||||
|
||||
// 查询文件列表
|
||||
List<TikUserFileDO> files = userFileMapper.selectBatchIds(fileIds);
|
||||
if (files.isEmpty()) {
|
||||
return;
|
||||
}
|
||||
|
||||
// 校验文件属于当前用户
|
||||
for (TikUserFileDO file : files) {
|
||||
if (!file.getUserId().equals(userId)) {
|
||||
boolean allOwnedByUser = files.stream().allMatch(file -> file.getUserId().equals(userId));
|
||||
if (!allOwnedByUser) {
|
||||
throw exception(FILE_NOT_EXISTS, "文件不属于当前用户");
|
||||
}
|
||||
}
|
||||
|
||||
// 物理删除OSS文件(使用 file_path,如果为空则使用 file_url)
|
||||
for (TikUserFileDO file : files) {
|
||||
try {
|
||||
// 优先使用 file_path,如果没有则从 file_url 提取路径
|
||||
String pathToDelete = file.getFilePath();
|
||||
if (StrUtil.isBlank(pathToDelete)) {
|
||||
// 如果 file_path 为空,尝试从 file_url 提取路径
|
||||
// file_url 格式可能是:https://domain.com/path/to/file 或 /path/to/file
|
||||
pathToDelete = extractPathFromUrl(file.getFileUrl());
|
||||
}
|
||||
|
||||
if (StrUtil.isNotBlank(pathToDelete)) {
|
||||
// 通过 FileApi 删除文件(需要先获取 FileDO,然后删除)
|
||||
// 注意:这里需要关联 infra_file 表获取 configId 和 path
|
||||
// 如果 file_id 存在,可以通过 FileService 删除
|
||||
// 否则需要直接调用 FileClient 删除
|
||||
// TODO: 实现物理删除OSS文件的逻辑
|
||||
log.debug("[deleteFiles][准备删除OSS文件,路径({})]", pathToDelete);
|
||||
}
|
||||
} catch (Exception e) {
|
||||
// 删除OSS文件失败不影响逻辑删除,只记录日志
|
||||
log.warn("[deleteFiles][删除OSS文件失败,文件编号({}),路径({})]",
|
||||
file.getId(), file.getFilePath(), e);
|
||||
}
|
||||
}
|
||||
|
||||
// 逻辑删除文件
|
||||
userFileMapper.deleteBatchIds(fileIds);
|
||||
@@ -442,7 +328,7 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
long totalSize = files.stream().mapToLong(TikUserFileDO::getFileSize).sum();
|
||||
memberUserProfileService.decreaseUsedStorage(String.valueOf(userId), totalSize);
|
||||
|
||||
log.info("[deleteFiles][用户({})删除文件成功,文件数量({})]", userId, fileIds.size());
|
||||
log.info("[deleteFiles][用户({})删除文件成功,数量({})]", userId, fileIds.size());
|
||||
}
|
||||
|
||||
@Override
|
||||
@@ -514,37 +400,25 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
|
||||
/**
|
||||
* 确定预览URL
|
||||
*
|
||||
* @param file 文件对象
|
||||
* @param type 预览类型(cover/thumbnail,可选)
|
||||
* @return 预览URL
|
||||
*/
|
||||
private String determinePreviewUrl(TikUserFileDO file, String type) {
|
||||
// 视频文件:不返回任何OSS URL(前端使用coverBase64缓存)
|
||||
boolean isVideo = StrUtil.containsIgnoreCase(file.getFileType(), "video");
|
||||
if (isVideo) {
|
||||
if (StrUtil.containsIgnoreCase(file.getFileType(), "video")) {
|
||||
return null;
|
||||
}
|
||||
|
||||
// 明确指定封面类型(非视频文件)
|
||||
// 明确指定类型
|
||||
if (StrUtil.equals(type, "cover") && StrUtil.isNotBlank(file.getCoverUrl())) {
|
||||
return file.getCoverUrl();
|
||||
}
|
||||
// 明确指定缩略图类型
|
||||
if (StrUtil.equals(type, "thumbnail") && StrUtil.isNotBlank(file.getThumbnailUrl())) {
|
||||
return file.getThumbnailUrl();
|
||||
}
|
||||
|
||||
// 根据文件类型自动选择
|
||||
boolean isImage = FileTypeUtils.isImage(file.getFileType());
|
||||
|
||||
if (isImage) {
|
||||
return StrUtil.isNotBlank(file.getThumbnailUrl())
|
||||
? file.getThumbnailUrl()
|
||||
: file.getFileUrl();
|
||||
if (FileTypeUtils.isImage(file.getFileType())) {
|
||||
return StrUtil.isNotBlank(file.getThumbnailUrl()) ? file.getThumbnailUrl() : file.getFileUrl();
|
||||
}
|
||||
|
||||
// 默认返回原文件URL
|
||||
return file.getFileUrl();
|
||||
}
|
||||
|
||||
@@ -558,92 +432,41 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
return fileApi.presignGetUrl(url, expirationSeconds != null ? expirationSeconds : 86400);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 从URL中提取文件路径
|
||||
*
|
||||
* @param url 文件URL
|
||||
* @return 文件路径
|
||||
*/
|
||||
private String extractPathFromUrl(String url) {
|
||||
if (StrUtil.isBlank(url)) {
|
||||
return null;
|
||||
}
|
||||
try {
|
||||
// 移除URL中的查询参数(签名参数等)
|
||||
String cleanUrl = url;
|
||||
if (url.contains("?")) {
|
||||
cleanUrl = url.substring(0, url.indexOf("?"));
|
||||
}
|
||||
|
||||
// 如果URL包含域名,提取路径部分
|
||||
if (cleanUrl.contains("://")) {
|
||||
int pathStart = cleanUrl.indexOf("/", cleanUrl.indexOf("://") + 3);
|
||||
if (pathStart > 0) {
|
||||
String fullPath = cleanUrl.substring(pathStart);
|
||||
// 路径可能包含 bucket 名称,需要提取实际的文件路径
|
||||
// 例如:/bucket-name/user-id/tenant-id/voice/20251117/file.wav
|
||||
// 实际 path 可能是:user-id/tenant-id/voice/20251117/file.wav
|
||||
// 但数据库中的 path 格式是:voice/20251117/file_timestamp.wav
|
||||
// 所以我们需要找到包含日期格式的部分(yyyyMMdd)
|
||||
return fullPath;
|
||||
}
|
||||
}
|
||||
// 如果已经是路径格式,直接返回(去除查询参数)
|
||||
if (cleanUrl.startsWith("/")) {
|
||||
return cleanUrl;
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.warn("[extractPathFromUrl][从URL提取路径失败,URL({})]", url, e);
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object getPresignedUrl(String fileName, String fileCategory, Long groupId, Long fileSize) {
|
||||
public Object getPresignedUrl(String fileName, String fileCategory, Long groupId, Long fileSize, String contentType) {
|
||||
Long userId = SecurityFrameworkUtils.getLoginUserId();
|
||||
|
||||
// 1. 校验请求
|
||||
validateUploadRequest(fileCategory, fileSize);
|
||||
|
||||
// 2. 获取OSS基础目录路径
|
||||
String baseDirectory = ossInitService.getOssDirectoryByCategory(userId, fileCategory);
|
||||
|
||||
// 3. 生成文件路径
|
||||
String filePath = generateUploadPath(fileName, baseDirectory);
|
||||
|
||||
// 4. 生成预签名URL
|
||||
FileClient client = fileConfigService.getMasterFileClient();
|
||||
Assert.notNull(client, "客户端(master) 不能为空");
|
||||
|
||||
String presignedUrl = client.presignPutUrl(filePath);
|
||||
String effectiveContentType = StrUtil.isNotBlank(contentType)
|
||||
? contentType
|
||||
: FileTypeUtils.getMineType(fileName);
|
||||
|
||||
String presignedUrl = client.presignPutUrl(filePath, effectiveContentType);
|
||||
String visitUrl = client.presignGetUrl(filePath, null);
|
||||
|
||||
// 5. 替换为自定义域名
|
||||
String domain = getFileClientDomain(client.getId());
|
||||
if (StrUtil.isNotBlank(domain)) {
|
||||
presignedUrl = replaceUrlDomain(presignedUrl, domain);
|
||||
visitUrl = replaceUrlDomain(visitUrl, domain);
|
||||
}
|
||||
|
||||
// 6. 构建返回结果
|
||||
return Map.of(
|
||||
"presignedUrl", presignedUrl,
|
||||
"uploadUrl", HttpUtils.removeUrlQuery(visitUrl),
|
||||
"fileKey", filePath,
|
||||
"expiresIn", PRESIGN_URL_EXPIRATION_SECONDS,
|
||||
"method", "PUT",
|
||||
"headers", Map.of(
|
||||
"Content-Type", FileTypeUtils.getMineType(fileName)
|
||||
)
|
||||
"headers", Map.of("Content-Type", effectiveContentType)
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取文件客户端的自定义域名
|
||||
* @param configId 配置编号
|
||||
* @return 自定义域名,如果没有配置则返回 null
|
||||
*/
|
||||
private String getFileClientDomain(Long configId) {
|
||||
if (configId == null) {
|
||||
return null;
|
||||
@@ -661,27 +484,19 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
|
||||
/**
|
||||
* 替换URL的域名部分
|
||||
* @param url 原始URL
|
||||
* @param newDomain 新域名
|
||||
* @return 替换后的URL
|
||||
*/
|
||||
private String replaceUrlDomain(String url, String newDomain) {
|
||||
if (StrUtil.isBlank(url) || StrUtil.isBlank(newDomain)) {
|
||||
return url;
|
||||
}
|
||||
try {
|
||||
// 提取原始URL的协议和主机名
|
||||
int schemeEnd = url.indexOf("://");
|
||||
if (schemeEnd > 0) {
|
||||
if (schemeEnd <= 0) {
|
||||
return url;
|
||||
}
|
||||
int pathStart = url.indexOf("/", schemeEnd + 3);
|
||||
String pathAndQuery = pathStart > 0 ? url.substring(pathStart) : "";
|
||||
return newDomain + pathAndQuery;
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.warn("[replaceUrlDomain][替换域名失败,url({})]", url, e);
|
||||
}
|
||||
return url;
|
||||
}
|
||||
|
||||
@Override
|
||||
@Transactional(rollbackFor = Exception.class)
|
||||
@@ -689,7 +504,6 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
public Object completeUpload(Object request) {
|
||||
Long userId = SecurityFrameworkUtils.getLoginUserId();
|
||||
|
||||
// 解析请求参数
|
||||
Map<String, Object> params = (Map<String, Object>) request;
|
||||
String fileKey = (String) params.get("fileKey");
|
||||
String fileName = (String) params.get("fileName");
|
||||
@@ -700,16 +514,12 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
String coverBase64 = (String) params.get("coverBase64");
|
||||
Integer duration = params.get("duration") != null ? ((Number) params.get("duration")).intValue() : null;
|
||||
|
||||
// 1. 校验请求
|
||||
validateUploadRequest(fileCategory, fileSize);
|
||||
|
||||
// 2. 获取文件访问URL
|
||||
FileClient client = fileConfigService.getMasterFileClient();
|
||||
Assert.notNull(client, "客户端(master) 不能为空");
|
||||
String fileUrl = client.presignGetUrl(fileKey, null);
|
||||
fileUrl = HttpUtils.removeUrlQuery(fileUrl);
|
||||
String fileUrl = HttpUtils.removeUrlQuery(client.presignGetUrl(fileKey, null));
|
||||
|
||||
// 3. 保存到 infra_file 表
|
||||
FileDO infraFile = new FileDO()
|
||||
.setConfigId(client.getId())
|
||||
.setName(fileName)
|
||||
@@ -720,18 +530,15 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
fileMapper.insert(infraFile);
|
||||
Long infraFileId = infraFile.getId();
|
||||
|
||||
// 4. 处理视频封面
|
||||
String baseDirectory = ossInitService.getOssDirectoryByCategory(userId, fileCategory);
|
||||
String coverUrl = handleCoverUpload(coverBase64, fileName, fileType, baseDirectory);
|
||||
// 处理显示名称(去除文件后缀)
|
||||
String displayName = FileUtil.mainName(fileName);
|
||||
|
||||
// 5. 保存到 tik_user_file 表
|
||||
TikUserFileDO userFile = new TikUserFileDO()
|
||||
.setUserId(userId)
|
||||
.setFileId(infraFileId)
|
||||
.setFileName(fileName) // 保留原始文件名(系统标识)
|
||||
.setDisplayName(displayName) // 设置显示名称(无后缀,用户可编辑)
|
||||
.setFileName(fileName)
|
||||
.setDisplayName(displayName)
|
||||
.setFileType(fileType)
|
||||
.setFileCategory(fileCategory)
|
||||
.setFileSize(fileSize)
|
||||
@@ -743,13 +550,11 @@ public class TikUserFileServiceImpl implements TikUserFileService {
|
||||
.setGroupId(groupId);
|
||||
userFileMapper.insert(userFile);
|
||||
|
||||
// 6. 更新配额
|
||||
memberUserProfileService.increaseUsedStorage(String.valueOf(userId), fileSize);
|
||||
|
||||
log.info("[completeUpload][用户({})直传上传完成,infraFileId({}),userFileId({})]",
|
||||
userId, infraFileId, userFile.getId());
|
||||
|
||||
// 7. 返回结果
|
||||
return Map.of(
|
||||
"infraFileId", infraFileId,
|
||||
"userFileId", userFile.getId(),
|
||||
|
||||
Reference in New Issue
Block a user