fix: 个人库相关功能

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

@@ -14,7 +14,7 @@ public class ThreadPoolConfig {
         ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
         executor.setCorePoolSize(20); // 核心线程池大小
         executor.setMaxPoolSize(50); // 最大线程池大小
-        executor.setQueueCapacity(150); // 队列容量
+        executor.setQueueCapacity(500); // 队列容量
         executor.setThreadNamePrefix("Async-"); // 线程名称前缀
         executor.initialize();
         return executor;

src/main/java/com/lqkj/link/module/authority/service/UserInfoService.java

@@ -3,6 +3,7 @@ package com.lqkj.link.module.authority.service;
 import com.lqkj.link.module.authority.domain.UserInfo;
 import com.lqkj.link.module.authority.repository.RoleInfoRepository;
 import com.lqkj.link.module.authority.repository.UserInfoRepository;
+import com.lqkj.link.util.FileUtils;
 import jakarta.annotation.PostConstruct;
 import org.springframework.transaction.annotation.Transactional;
 import org.apache.commons.lang3.StringUtils;
@@ -12,6 +13,7 @@ import org.springframework.data.domain.Pageable;
 import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.stereotype.Service;
 
+import java.io.File;
 import java.util.*;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
@@ -176,4 +178,6 @@ public class UserInfoService {
     public void resourceRefreshed(String userCode) {
         userInfoRepository.resourceRefreshed();
     }
+
+
 }

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

@@ -88,4 +88,12 @@ public class BaseService {
             throw new RuntimeException(e);
         }
     }
+
+    public String ossAddModel(MultipartFile file, String s) {
+        try {
+            return aliOSSUtils.addModel(file, s);
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
 }

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

@@ -249,4 +249,34 @@ public class ResourceController {
     public MessageBean<String> ossUpload(MultipartFile file) {
         return MessageBean.ok(baseService.ossUpload(file), "oss上传");
     }
+
+    @GetMapping("/checkCapacity")
+    public MessageBean<String> checkCapacity(HttpServletRequest request){
+        String authHeader = request.getHeader("Authorization");
+        String userCode = jwtService.decryptUsernameWithHeader(authHeader);
+        return MessageBean.ok(resourceService.checkCapacity(userCode), "检查容量");
+    }
+
+    @Operation(
+            summary = "个人库模型上传接口",
+            description = "个人库模型上传接口",
+            parameters = {
+                    @Parameter(name = "file", description = "文件", required = true)
+            }
+    )
+    @PostMapping("/oss/addModel")
+    public MessageBean<String> ossAddModel(MultipartFile file,
+                                        HttpServletRequest request) {
+        String authHeader = request.getHeader("Authorization");
+        String userCode = jwtService.decryptUsernameWithHeader(authHeader);
+        Integer userId = userInfoService.detailByUserCode(userCode).getUserId();
+        return MessageBean.ok(baseService.ossAddModel(file, "resource/model/" + userId + "/"), "oss上传");
+    }
+
+    @GetMapping("/oss/checkCapacity")
+    public MessageBean<String> ossCheckCapacity(HttpServletRequest request){
+        String authHeader = request.getHeader("Authorization");
+        String userCode = jwtService.decryptUsernameWithHeader(authHeader);
+        return MessageBean.ok(resourceService.ossCheckCapacity(userCode), "检查容量");
+    }
 }

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

@@ -7,6 +7,7 @@ import com.lqkj.link.module.zone.domain.ModelInfo;
 import com.lqkj.link.module.zone.repository.ModelCategoryRepository;
 import com.lqkj.link.module.zone.repository.ModelInfoRepository;
 import com.lqkj.link.util.AliOSSUtils;
+import com.lqkj.link.util.FileUtils;
 import com.lqkj.link.util.Unzipper;
 import org.apache.commons.compress.archivers.ArchiveException;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -164,4 +165,18 @@ public class ResourceService {
         UserInfo userInfo = userInfoRepository.findByUserCode(userCode);
         return userInfo.getRefreshResource() != null && userInfo.getRefreshResource();
     }
+
+    public String checkCapacity(String userCode) {
+        UserInfo userInfo = userInfoRepository.findByUserCode(userCode);
+        String filePath = "./upload/resource/model/" + userInfo.getUserId();
+        File file = new File(filePath);
+        if (!file.exists()) return "0";
+        return "个人库 已使用" + FileUtils.convertBytes(FileUtils.getFolderSize(new File(filePath))) + "/20G";
+    }
+
+    public String ossCheckCapacity(String userCode) {
+        UserInfo userInfo = userInfoRepository.findByUserCode(userCode);
+        Long aLong = aliOSSUtils.ossCheckCapacity("resource/model/" + userInfo.getUserId() + "/");
+        return "个人库 已使用" + FileUtils.convertBytes(aLong) + "/20G";
+    }
 }

src/main/java/com/lqkj/link/util/AliOSSUtils.java

@@ -2,6 +2,9 @@ package com.lqkj.link.util;
 
 import com.aliyun.oss.OSS;
 import com.aliyun.oss.OSSClientBuilder;
+import com.aliyun.oss.model.ListObjectsV2Request;
+import com.aliyun.oss.model.ListObjectsV2Result;
+import com.aliyun.oss.model.OSSObjectSummary;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.springframework.web.multipart.MultipartFile;
@@ -13,6 +16,7 @@ import java.io.InputStream;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.LocalTime;
+import java.util.List;
 import java.util.UUID;
 import java.util.concurrent.Callable;
 
@@ -38,6 +42,11 @@ public class AliOSSUtils {
         String originalFilename = file.getName();
         String fileName = "file/" + LocalDate.now() + "/" + UUID.randomUUID() + originalFilename.substring(originalFilename.lastIndexOf("."));
         //上传文件到 OSS
+        try {
+            Thread.sleep(200);
+        } catch (InterruptedException e) {
+            throw new RuntimeException(e);
+        }
         OSS ossClient = new OSSClientBuilder().build(aliProperties.getEndpoint(), aliProperties.getAccessKeyId(), aliProperties.getAccessKeySecret());
         threadPoolUtil.getTaskExecutor().execute(() -> {
             ossClient.putObject(aliProperties.getBucketName(), fileName, file);
@@ -70,5 +79,49 @@ public class AliOSSUtils {
         }
     }
 
- 
+
+    public String addModel(MultipartFile file, String path) {
+        // 获取上传的文件的输入流
+        try {
+            InputStream inputStream = file.getInputStream();
+            // 避免文件覆盖
+            String originalFilename = file.getOriginalFilename();
+            String fileName = path + UUID.randomUUID() + originalFilename.substring(originalFilename.lastIndexOf("."));
+            //上传文件到 OSS
+            OSS ossClient = new OSSClientBuilder().build(aliProperties.getEndpoint(), aliProperties.getAccessKeyId(), aliProperties.getAccessKeySecret());
+            threadPoolUtil.getTaskExecutor().execute(() -> {
+                ossClient.putObject(aliProperties.getBucketName(), fileName, inputStream);
+                ossClient.shutdown();
+            });
+            //文件访问路径
+            return aliProperties.getOsspath().split("//")[0] + "//" + aliProperties.getBucketName() + "." + aliProperties.getOsspath().split("//")[1] + "/" + fileName;
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public Long ossCheckCapacity(String path) {
+        OSS ossClient = new OSSClientBuilder().build(aliProperties.getEndpoint(), aliProperties.getAccessKeyId(), aliProperties.getAccessKeySecret());
+        return calculateFolderLength(ossClient, aliProperties.getBucketName(), path);
+    }
+
+    private static long calculateFolderLength(OSS ossClient, String bucketName, String folder) {
+        long size = 0L;
+        ListObjectsV2Result result = null;
+        do {
+            // MaxKey默认值为100，最大值为1000。
+            ListObjectsV2Request request = new ListObjectsV2Request(bucketName).withPrefix(folder).withMaxKeys(1000);
+            if (result != null) {
+                request.setContinuationToken(result.getNextContinuationToken());
+            }
+            result = ossClient.listObjectsV2(request);
+            List<OSSObjectSummary> sums = result.getObjectSummaries();
+            for (OSSObjectSummary s : sums) {
+                size += s.getSize();
+            }
+        } while (result.isTruncated());
+        return size;
+    }
+
+
 }

src/main/java/com/lqkj/link/util/FileUtils.java

@@ -179,4 +179,49 @@ public class FileUtils {
         }
         return result;
     }
+
+    public static long getFolderSize(File folder) {
+        long totalSize = 0;
+
+        // 检查路径是否存在且为目录
+        if (folder.exists() && folder.isDirectory()) {
+            File[] files = folder.listFiles();
+            if (files != null) { // 处理可能为 null 的情况
+                for (File file : files) {
+                    if (file.isFile()) {
+                        // 累加文件大小
+                        totalSize += file.length();
+                    } else if (file.isDirectory()) {
+                        // 递归计算子目录大小
+                        totalSize += getFolderSize(file);
+                    }
+                }
+            }
+        } else {
+            System.out.println("The specified path does not exist or is not a directory.");
+        }
+
+        return totalSize;
+    }
+
+    public static String convertBytes(long bytes) {
+        if (bytes < 0) {
+            return "Invalid size";
+        }
+
+        // 定义单位
+        String[] units = {"B", "KB", "MB", "GB"};
+        int unitIndex = 0;
+
+        // 逐步转换为更大的单位
+        double size = (double) bytes;
+
+        while (unitIndex < units.length - 1) {
+            size /= 1024;
+            unitIndex++;
+        }
+
+        // 使用 String.format 格式化输出
+        return String.format("%.2f%s", size, units[unitIndex]);
+    }
 }