fix: 增加上传校验

src/main/java/com/lqkj/link/config/ThreadPoolConfig.java

@@ -5,6 +5,7 @@ import org.springframework.context.annotation.Configuration;
 import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
 
 import java.util.concurrent.Executor;
+import java.util.concurrent.ThreadPoolExecutor;
 
 @Configuration
 public class ThreadPoolConfig {
@@ -15,6 +16,9 @@ public class ThreadPoolConfig {
         executor.setCorePoolSize(20); // 核心线程池大小
         executor.setMaxPoolSize(50); // 最大线程池大小
         executor.setQueueCapacity(500); // 队列容量
+        executor.setKeepAliveSeconds(300);
+        executor.setWaitForTasksToCompleteOnShutdown(true);
+        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
         executor.setThreadNamePrefix("Async-"); // 线程名称前缀
         executor.initialize();
         return executor;

src/main/java/com/lqkj/link/module/base/service/BaseService.java

@@ -90,6 +90,10 @@ public class BaseService {
     }
 
     public String ossAddModel(MultipartFile file, String s) {
+        String fileName = file.getOriginalFilename();
+        String suffix = fileName == null ? "" : fileName.substring(fileName.lastIndexOf(".") + 1);
+        if (!suffix.equals("obj") && !suffix.equals("fbx"))
+            throw new RuntimeException("上传文件类型必须是obj、fbx格式的压缩文件");
         if (aliOSSUtils.ossCheckCapacity(s) + file.getSize() >= 21474836480L)
             throw new RuntimeException("容量已满，无法上传文件");
         try {

src/main/java/com/lqkj/link/module/zone/controller/ResourceController.java

@@ -250,6 +250,11 @@ public class ResourceController {
         return MessageBean.ok(baseService.ossUpload(file), "oss上传");
     }
 
+    /**
+     * 检查本地容量
+     * @param request
+     * @return
+     */
     @GetMapping("/checkCapacity")
     public MessageBean<String> checkCapacity(HttpServletRequest request){
         String authHeader = request.getHeader("Authorization");
@@ -273,6 +278,12 @@ public class ResourceController {
         return MessageBean.ok(baseService.ossAddModel(file, "resource/model/" + userId + "/"), "oss上传");
     }
 
+
+    /**
+     * 阿里云检查容量
+     * @param request
+     * @return
+     */
     @GetMapping("/oss/checkCapacity")
     public MessageBean<String> ossCheckCapacity(HttpServletRequest request){
         String authHeader = request.getHeader("Authorization");

src/main/java/com/lqkj/link/module/zone/service/ResourceService.java

@@ -166,6 +166,11 @@ public class ResourceService {
         return userInfo.getRefreshResource() != null && userInfo.getRefreshResource();
     }
 
+    /**
+     * 检查本地个人库容量
+     * @param userCode
+     * @return
+     */
     public String checkCapacity(String userCode) {
         UserInfo userInfo = userInfoRepository.findByUserCode(userCode);
         String filePath = "./upload/resource/model/" + userInfo.getUserId();
@@ -174,6 +179,11 @@ public class ResourceService {
         return "个人库 已使用" + FileUtils.convertBytes(FileUtils.getFolderSize(new File(filePath))) + "/20G";
     }
 
+    /**
+     * 检查OSS个人库容量
+     * @param userCode
+     * @return
+     */
     public String ossCheckCapacity(String userCode) {
         UserInfo userInfo = userInfoRepository.findByUserCode(userCode);
         Long aLong = aliOSSUtils.ossCheckCapacity("resource/model/" + userInfo.getUserId() + "/");