Explorar o código

文件名长度验证

gaoxiong hai 1 ano
pai
achega
253949fd60

+ 8 - 0
soc-modules/soc-modules-file/src/main/java/com/xunmei/file/controller/SysFileController.java

@@ -20,6 +20,7 @@ import org.springframework.web.multipart.MultipartFile;
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 import java.util.Map;
+import java.util.Objects;
 
 /**
  * 文件请求处理
@@ -57,6 +58,10 @@ public class SysFileController {
     @PostMapping("/upload")
     public R<SysFile> upload(MultipartFile file) {
         try {
+            if(Objects.requireNonNull(file.getOriginalFilename()).length() > 54){
+               return R.fail("上传文件名长度不能大于50!");
+            }
+
             // 上传并返回访问地址
             String url = sysFileService.uploadFile(file);
             SysFile sysFile = new SysFile();
@@ -76,6 +81,9 @@ public class SysFileController {
     @PostMapping("/uploadFile")
     public R<SysFile> uploadFile(MultipartFile file, String busType) {
         try {
+            if(Objects.requireNonNull(file.getOriginalFilename()).length() > 54){
+                return R.fail("上传文件名长度不能大于50!");
+            }
             // 上传并返回访问地址
             String url = sysFileService.uploadFile(file, busType);
             SysFile sysFile = new SysFile();