2210088963/AiJiNanProject
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1.增加了对文件名后缀的解析，若为图片则判断大小是否在范围内，不在则返回提示信息。

Browse Source
This commit is contained in:
2210088963 2025-02-08 18:03:17 +08:00
parent d1a0d75733
commit fefc6e54a4
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/main/java/com/platform/modules/live/controller/MediaUploadController.java
View File

@ -43,8 +43,7 @@ import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Calendar;
import java.util.UUID;
import java.util.*;
@RestController
@Api(value = "资源上传", tags = "资源上传")
@ -115,11 +114,25 @@ public class MediaUploadController {
@PostMapping("/common/upload_media/chunk")
@RequiresPermissions(value = {"live:multimedia:save", "live:multimedia:update"}, logical = Logical.OR)
public R uploadChunk(PlmChunkEntity chunk) {
public R uploadChunk(PlmChunkEntity chunk,@RequestParam(value = "limitSize", required = false, defaultValue = "500")Long limitSize) {
if (null == chunk || null == chunk.getFile() || StringUtils.isBlank(chunk.getIdentifier())
|| null == chunk.getChunkNumber()) {
return R.error("缺少必要参数");
}
// 支持的图片格式白名单
Set<String> ALLOWED_IMAGE_TYPES = new HashSet<>(
Arrays.asList("png", "jpg", "jpeg", "bmp", "gif", "webp")
);
int lastDotIndex = chunk.getFilename().lastIndexOf('.');
if (lastDotIndex == -1 || lastDotIndex == chunk.getFilename().length() - 1) {
return R.error("路径错误");
}
String extension = chunk.getFilename().substring(lastDotIndex + 1);
if(ALLOWED_IMAGE_TYPES.contains(extension)&&(limitSize * 1024) <= chunk.getTotalSize()) {
return R.error("图片大小必须控制在" + limitSize + "之内！");
}
String key = RedisKeys.getUploadMediaKey() + chunk.getIdentifier();
String field = chunk.getIdentifier() + "#" + chunk.getChunkNumber();
// 续传判断