From 5a8b2ff0f50b5034831a4d0e30dce3fc58ddcbbb Mon Sep 17 00:00:00 2001
From: xueqingkun <xueqingkun@126.com>
Date: Wed, 24 Jul 2024 15:04:15 +0800
Subject: [PATCH] =?UTF-8?q?1.=20=E8=B0=83=E6=95=B4=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E5=A4=A7=E5=B0=8F=E9=99=90=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/supervision/common/enums/ResultStatusEnum.java   | 1 +
 .../com/supervision/config/ExceptionHandlerConfig.java   | 9 +++++++++
 src/main/resources/application.yml                       | 4 ++++
 3 files changed, 14 insertions(+)

diff --git a/src/main/java/com/supervision/common/enums/ResultStatusEnum.java b/src/main/java/com/supervision/common/enums/ResultStatusEnum.java
index 9b6790f..07bd215 100644
--- a/src/main/java/com/supervision/common/enums/ResultStatusEnum.java
+++ b/src/main/java/com/supervision/common/enums/ResultStatusEnum.java
@@ -36,6 +36,7 @@ public enum ResultStatusEnum {
 
     ILLEGAL_ARGUMENT(508, "参数校验失败!"),
     RUNTIME_EXCEPTION(509, "程序运行异常!"),
+    EXCEED_FILE_SIZE(510, "文件大小超出限制!"),
     IMPORT_COMPANY_FORMAT_ERROR(521,"Excel表格格式错误！"),
     IMPORT_COMPANY_FAIL(522,"部分数据导入失败"),
     INSERT_FAIL(600,"新增失败"),
diff --git a/src/main/java/com/supervision/config/ExceptionHandlerConfig.java b/src/main/java/com/supervision/config/ExceptionHandlerConfig.java
index b79a4fa..603585d 100644
--- a/src/main/java/com/supervision/config/ExceptionHandlerConfig.java
+++ b/src/main/java/com/supervision/config/ExceptionHandlerConfig.java
@@ -7,6 +7,7 @@ import org.springframework.context.annotation.Configuration;
 import org.springframework.web.bind.annotation.ExceptionHandler;
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.bind.annotation.RestControllerAdvice;
+import org.springframework.web.multipart.MaxUploadSizeExceededException;
 
 /**
  * 统一异常处理器配置
@@ -39,4 +40,12 @@ public class ExceptionHandlerConfig {
         log.error("<<<=========运行异常=========");
         return R.fail(ResultStatusEnum.RUNTIME_EXCEPTION,exception.getMessage());
     }
+
+    @ExceptionHandler(MaxUploadSizeExceededException.class)
+    public R<?> handleMaxSizeException(MaxUploadSizeExceededException exception) {
+        log.error("=========文件大小超出限制异常=========>>>");
+        log.error(exception.getMessage(), exception);
+        log.error("<<<=========文件大小超出限制异常=========");
+        return R.fail(ResultStatusEnum.EXCEED_FILE_SIZE,exception.getMessage());
+    }
 }
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index 2708ad0..3d05568 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -4,6 +4,10 @@ spring:
   main:
     allow-bean-definition-overriding: true
     allow-circular-references: true
+  servlet:
+    multipart:
+      max-file-size: 10MB
+      max-request-size: 200MB
   mvc:
     path match:
       matching-strategy: ANT_PATH_MATCHER