feat(storage): 支持分片上传初始化及相关功能

Author: Time-w

continew-starter-storage/src/main/java/top/continew/starter/storage/autoconfigure/LocalStorageAutoConfiguration.java

@@ -16,6 +16,7 @@
 
 package top.continew.starter.storage.autoconfigure;
 
+import cn.hutool.core.util.StrUtil;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
 import org.springframework.context.annotation.Bean;
 import top.continew.starter.core.constant.PropertiesConstants;
@@ -52,6 +53,15 @@ public LocalStorageAutoConfiguration(StorageProperties storageProperties) {
     public void register(List<StorageStrategy> strategies) {
         for (LocalStorageConfig config : storageProperties.getLocal()) {
             if (config.isEnabled()) {
+                if (config.getMultipartUploadThreshold() == null || config.getMultipartUploadThreshold() <= 0) {
+                    config.setMultipartUploadThreshold(storageProperties.getMultipartUploadThreshold());
+                }
+                if (config.getMultipartUploadPartSize() == null || config.getMultipartUploadPartSize() <= 0) {
+                    config.setMultipartUploadPartSize(storageProperties.getMultipartUploadPartSize());
+                }
+                if (StrUtil.isBlank(config.getMultipartTempDir())) {
+                    config.setMultipartTempDir(storageProperties.getLocalMultipartTempDir());
+                }
                 strategies.add(new LocalStorageStrategy(config));
             }
         }

continew-starter-storage/src/main/java/top/continew/starter/storage/autoconfigure/OssStorageAutoConfiguration.java

@@ -52,6 +52,12 @@ public OssStorageAutoConfiguration(StorageProperties properties) {
     public void register(List<StorageStrategy> strategies) {
         for (OssStorageConfig config : properties.getOss()) {
             if (config.isEnabled()) {
+                if (config.getMultipartUploadThreshold() == null || config.getMultipartUploadThreshold() <= 0) {
+                    config.setMultipartUploadThreshold(properties.getMultipartUploadThreshold());
+                }
+                if (config.getMultipartUploadPartSize() == null || config.getMultipartUploadPartSize() <= 0) {
+                    config.setMultipartUploadPartSize(properties.getMultipartUploadPartSize());
+                }
                 strategies.add(new OssStorageStrategy(config));
             }
         }

continew-starter-storage/src/main/java/top/continew/starter/storage/autoconfigure/properties/LocalStorageConfig.java

@@ -16,6 +16,8 @@
 
 package top.continew.starter.storage.autoconfigure.properties;
 
+import top.continew.starter.storage.common.constant.StorageConstant;
+
 /**
  * 本地存储配置
  *
@@ -44,6 +46,21 @@ public class LocalStorageConfig {
      */
     private String endpoint;
 
+    /**
+     * 多部分上传阈值（字节）
+     */
+    private Long multipartUploadThreshold;
+
+    /**
+     * 多部分上传的部分大小（字节）
+     */
+    private Long multipartUploadPartSize;
+
+    /**
+     * 分片上传临时目录
+     */
+    private String multipartTempDir = StorageConstant.DEFAULT_LOCAL_MULTIPART_TEMP_DIR;
+
     public boolean isEnabled() {
         return enabled;
     }
@@ -75,4 +92,28 @@ public String getEndpoint() {
     public void setEndpoint(String endpoint) {
         this.endpoint = endpoint;
     }
+
+    public Long getMultipartUploadThreshold() {
+        return multipartUploadThreshold;
+    }
+
+    public void setMultipartUploadThreshold(Long multipartUploadThreshold) {
+        this.multipartUploadThreshold = multipartUploadThreshold;
+    }
+
+    public Long getMultipartUploadPartSize() {
+        return multipartUploadPartSize;
+    }
+
+    public void setMultipartUploadPartSize(Long multipartUploadPartSize) {
+        this.multipartUploadPartSize = multipartUploadPartSize;
+    }
+
+    public String getMultipartTempDir() {
+        return multipartTempDir;
+    }
+
+    public void setMultipartTempDir(String multipartTempDir) {
+        this.multipartTempDir = multipartTempDir;
+    }
 }

continew-starter-storage/src/main/java/top/continew/starter/storage/autoconfigure/properties/OssStorageConfig.java

@@ -74,12 +74,12 @@ public class OssStorageConfig {
     /**
      * 多部分上传阈值（字节）
      */
-    private long multipartUploadThreshold = StorageConstant.DEFAULT_FILE_SIZE;
+    private Long multipartUploadThreshold;
 
     /**
      * 多部分上传的部分大小（字节）
      */
-    private long multipartUploadPartSize = StorageConstant.DEFAULT_FILE_SIZE;
+    private Long multipartUploadPartSize;
 
     /**
      * 请求超时时间（秒）
@@ -168,19 +168,19 @@ public void setTransferAccelerationEnabled(boolean transferAccelerationEnabled)
         this.transferAccelerationEnabled = transferAccelerationEnabled;
     }
 
-    public long getMultipartUploadThreshold() {
+    public Long getMultipartUploadThreshold() {
         return multipartUploadThreshold;
     }
 
-    public void setMultipartUploadThreshold(long multipartUploadThreshold) {
+    public void setMultipartUploadThreshold(Long multipartUploadThreshold) {
         this.multipartUploadThreshold = multipartUploadThreshold;
     }
 
-    public long getMultipartUploadPartSize() {
+    public Long getMultipartUploadPartSize() {
         return multipartUploadPartSize;
     }
 
-    public void setMultipartUploadPartSize(long multipartUploadPartSize) {
+    public void setMultipartUploadPartSize(Long multipartUploadPartSize) {
         this.multipartUploadPartSize = multipartUploadPartSize;
     }
 

continew-starter-storage/src/main/java/top/continew/starter/storage/autoconfigure/properties/StorageProperties.java

@@ -43,6 +43,21 @@ public class StorageProperties {
      */
     private DefaultStorageSource defaultStorageSource = DefaultStorageSource.DYNAMIC;
 
+    /**
+     * 全局分片阈值（字节）
+     */
+    private long multipartUploadThreshold = StorageConstant.DEFAULT_MULTIPART_UPLOAD_THRESHOLD;
+
+    /**
+     * 全局分片大小（字节）
+     */
+    private long multipartUploadPartSize = StorageConstant.DEFAULT_MULTIPART_UPLOAD_PART_SIZE;
+
+    /**
+     * 全局本地分片临时目录
+     */
+    private String localMultipartTempDir = StorageConstant.DEFAULT_LOCAL_MULTIPART_TEMP_DIR;
+
     /**
      * 本地存储配置列表
      */
@@ -84,4 +99,28 @@ public List<OssStorageConfig> getOss() {
     public void setOss(List<OssStorageConfig> oss) {
         this.oss = oss;
     }
+
+    public long getMultipartUploadThreshold() {
+        return multipartUploadThreshold;
+    }
+
+    public void setMultipartUploadThreshold(long multipartUploadThreshold) {
+        this.multipartUploadThreshold = multipartUploadThreshold;
+    }
+
+    public long getMultipartUploadPartSize() {
+        return multipartUploadPartSize;
+    }
+
+    public void setMultipartUploadPartSize(long multipartUploadPartSize) {
+        this.multipartUploadPartSize = multipartUploadPartSize;
+    }
+
+    public String getLocalMultipartTempDir() {
+        return localMultipartTempDir;
+    }
+
+    public void setLocalMultipartTempDir(String localMultipartTempDir) {
+        this.localMultipartTempDir = localMultipartTempDir;
+    }
 }

continew-starter-storage/src/main/java/top/continew/starter/storage/common/constant/StorageConstant.java

@@ -46,6 +46,21 @@ public class StorageConstant {
      */
     public static final Long DEFAULT_FILE_SIZE = 1024 * 1024 * 10L;
 
+    /**
+     * 默认分片阈值（字节）
+     */
+    public static final Long DEFAULT_MULTIPART_UPLOAD_THRESHOLD = DEFAULT_FILE_SIZE;
+
+    /**
+     * 默认分片大小（字节）
+     */
+    public static final Long DEFAULT_MULTIPART_UPLOAD_PART_SIZE = 5 * 1024 * 1024L;
+
+    /**
+     * 本地分片上传临时目录
+     */
+    public static final String DEFAULT_LOCAL_MULTIPART_TEMP_DIR = "/tmp";
+
     /**
      * 默认的对象ACL
      */