FengChaoYu il y a 2 mois
Parent
commit
358ab203df

+ 22 - 2
mall-miniapp-service/src/main/java/com/gree/mall/miniapp/config/MulitpartConfig.java

@@ -14,6 +14,10 @@ public class MulitpartConfig {
 
     @Value("${spring.profiles.active}")
     private String env;
+    @Value("${spring.servlet.multipart.max-file-size}")
+    private String maxFileSize;
+    @Value("${spring.servlet.multipart.max-request-size}")
+    private String maxRequestSize;
     /**
      * 文件上传临时路径
      */
@@ -25,8 +29,24 @@ public class MulitpartConfig {
         } else {
             factory.setLocation("/app");
         }
-        factory.setMaxFileSize(DataSize.of(40, DataUnit.MEGABYTES));
-        factory.setMaxRequestSize(DataSize.of(50, DataUnit.MEGABYTES));
+        DataUnit unit = DataUnit.MEGABYTES;
+        long longMaxFileSize = 10L;
+        long longMaxRequestSize = 10L;
+        if (maxFileSize.contains("MB")) {
+            longMaxFileSize = Long.parseLong(maxFileSize.replace("MB", ""));
+            longMaxRequestSize = Long.parseLong(maxRequestSize.replace("MB", ""));
+        } else if (maxFileSize.contains("GB")) {
+            unit = DataUnit.GIGABYTES;
+            longMaxFileSize = Long.parseLong(maxFileSize.replace("GB", ""));
+            longMaxRequestSize = Long.parseLong(maxRequestSize.replace("GB", ""));
+        } else if (maxFileSize.contains("TB")) {
+            unit = DataUnit.TERABYTES;
+            longMaxFileSize = Long.parseLong(maxFileSize.replace("TB", ""));
+            longMaxRequestSize = Long.parseLong(maxRequestSize.replace("TB", ""));
+        }
+
+        factory.setMaxFileSize(DataSize.of(longMaxFileSize, unit));
+        factory.setMaxRequestSize(DataSize.of(longMaxRequestSize, unit));
         return factory.createMultipartConfig();
     }