From 88f20021e6887574f6ea456fa2ed5928c74cf995 Mon Sep 17 00:00:00 2001
From: wangshuai <daws172839@163.com>
Date: Sat, 27 Jul 2024 15:50:57 +0800
Subject: [PATCH] =?UTF-8?q?2=EF=BC=9B=E5=9B=BE=E7=89=87=E4=B8=8A=E4=BC=A0?=
 =?UTF-8?q?=E6=9D=A1=E4=BB=B6=E6=8B=A6=E6=88=AA=EF=BC=88=E6=A0=BC=E5=BC=8F?=
 =?UTF-8?q?jpg=20png=EF=BC=8C=E5=A4=A7=E5=B0=8F50M=E7=AD=89=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../web/controller/common/CommonController.java      |  2 +-
 .../com/ruoyi/common/utils/file/FileUploadUtils.java | 12 ++++++++++++
 .../com/ruoyi/common/utils/file/MimeTypeUtils.java   |  4 ++++
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java
index 7b1455b..bf5876d 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java
@@ -83,7 +83,7 @@ public class CommonController
             // 上传文件路径
             String filePath = RuoYiConfig.getUploadPath();
             // 上传并返回新文件名称
-            String fileName = FileUploadUtils.upload(filePath, file);
+            String fileName = FileUploadUtils.uploadImg(filePath, file);
             String url = serverConfig.getUrl() + fileName;
             AjaxResult ajax = AjaxResult.success();
             ajax.put("url", url);
diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUploadUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUploadUtils.java
index d5455c4..187626f 100644
--- a/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUploadUtils.java
+++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/FileUploadUtils.java
@@ -86,6 +86,18 @@ public class FileUploadUtils
         }
     }
 
+    public static final String uploadImg(String baseDir, MultipartFile file) throws IOException
+    {
+        try
+        {
+            return upload(baseDir, file, MimeTypeUtils.DEFAULT_IMG_ALLOWED_EXTENSION);
+        }
+        catch (Exception e)
+        {
+            throw new IOException(e.getMessage(), e);
+        }
+    }
+
     /**
      * 文件上传
      *
diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/MimeTypeUtils.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/MimeTypeUtils.java
index f968f1a..d786cb0 100644
--- a/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/MimeTypeUtils.java
+++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/file/MimeTypeUtils.java
@@ -38,6 +38,10 @@ public class MimeTypeUtils
             // pdf
             "pdf" };
 
+    public static final String[] DEFAULT_IMG_ALLOWED_EXTENSION = {
+            // 图片
+            "jpg", "png"};
+
     public static String getExtension(String prefix)
     {
         switch (prefix)