package com.tcm.training.admin.upload;
|
|
import com.tcm.training.admin.common.ApiResponse;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.RestController;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.IOException;
|
import java.util.Arrays;
|
import java.util.HashSet;
|
import java.util.Set;
|
|
@RestController
|
@RequestMapping({"/api/v1", "/api"})
|
public class UploadController {
|
|
private static final long MAX_SIZE = 5 * 1024 * 1024L; // 5MB
|
private static final long VIDEO_MAX_SIZE = 200 * 1024 * 1024L; // 200MB
|
private static final Set<String> ALLOWED_TYPES = new HashSet<>(
|
Arrays.asList("image/jpeg", "image/png", "image/gif", "image/webp")
|
);
|
private static final Set<String> ALLOWED_VIDEO_TYPES = new HashSet<>(
|
Arrays.asList("video/mp4", "video/quicktime", "video/x-msvideo", "video/x-matroska", "video/webm")
|
);
|
|
private final ObsUploadService obsUploadService;
|
private final UploadProgressService uploadProgressService;
|
|
public UploadController(ObsUploadService obsUploadService, UploadProgressService uploadProgressService) {
|
this.obsUploadService = obsUploadService;
|
this.uploadProgressService = uploadProgressService;
|
}
|
|
/**
|
* 图片上传
|
* POST /api/v1/upload/image
|
* @return { url: "访问地址" }
|
*/
|
@PostMapping("/upload/image")
|
public ApiResponse<UploadResult> uploadImage(
|
@RequestParam("file") MultipartFile file) throws IOException {
|
return ApiResponse.ok(storeFile(file, MAX_SIZE, ALLOWED_TYPES, "仅支持 JPG、PNG、GIF、WebP 格式", "image", null));
|
}
|
|
@PostMapping("/upload/video")
|
public ApiResponse<UploadResult> uploadVideo(
|
@RequestParam("file") MultipartFile file,
|
@RequestParam(value = "uploadTaskId", required = false) String uploadTaskId) throws IOException {
|
return ApiResponse.ok(storeFile(file, VIDEO_MAX_SIZE, ALLOWED_VIDEO_TYPES, "仅支持 MP4、MOV、AVI、MKV、WebM 格式", "video", uploadTaskId));
|
}
|
|
@GetMapping("/upload/video/progress")
|
public ApiResponse<UploadProgressInfo> uploadVideoProgress(
|
@RequestParam("taskId") String taskId) {
|
return ApiResponse.ok(uploadProgressService.get(taskId));
|
}
|
|
private UploadResult storeFile(MultipartFile file,
|
long maxSize,
|
Set<String> allowedTypes,
|
String typeMessage,
|
String bizPath,
|
String uploadTaskId) throws IOException {
|
if (file == null || file.isEmpty()) {
|
throw new IllegalArgumentException("请选择要上传的文件");
|
}
|
if (file.getSize() > maxSize) {
|
throw new IllegalArgumentException("文件大小超出限制");
|
}
|
String contentType = file.getContentType();
|
if (contentType == null || !allowedTypes.contains(contentType)) {
|
throw new IllegalArgumentException(typeMessage);
|
}
|
UploadFileResult uploadResult = obsUploadService.upload(file, bizPath, uploadTaskId);
|
return new UploadResult(uploadResult.getUrl(), uploadResult.getFileName());
|
}
|
|
public static class UploadResult {
|
private final String url;
|
private final String fileName;
|
|
public UploadResult(String url) {
|
this(url, null);
|
}
|
|
public UploadResult(String url, String fileName) {
|
this.url = url;
|
this.fileName = fileName;
|
}
|
|
public String getUrl() {
|
return url;
|
}
|
|
public String getFileName() {
|
return fileName;
|
}
|
}
|
}
|
