From 81ea26775c37749ba7857b134d6383a901804abd Mon Sep 17 00:00:00 2001
From: yjz <3131811435@qq.com>
Date: Wed, 7 Aug 2024 15:58:37 +0800
Subject: [PATCH] =?UTF-8?q?feat(=E6=96=87=E6=A1=A3=E6=8B=86=E5=88=86):=20?=
 =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=9B=BE=E7=89=87=E6=A0=A1=E9=AA=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/FileSplitItemsEOServiceImpl.java     | 29 +++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/laws-modules/src/main/java/com/jero/modules/split/service/impl/FileSplitItemsEOServiceImpl.java b/laws-modules/src/main/java/com/jero/modules/split/service/impl/FileSplitItemsEOServiceImpl.java
index 843bcd70..caced60c 100644
--- a/laws-modules/src/main/java/com/jero/modules/split/service/impl/FileSplitItemsEOServiceImpl.java
+++ b/laws-modules/src/main/java/com/jero/modules/split/service/impl/FileSplitItemsEOServiceImpl.java
@@ -1490,6 +1490,7 @@ public class FileSplitItemsEOServiceImpl extends ServiceImpl<FileSplitItemsEOMap
                                         try {
 //                                            String paths = splitUrl+imageName;
 //                MinioUtil.uploadByte(bytev, filepathandname);
+                                            imgCheck(multipartFile);
                                             String upload = ObsBootUtil.upload(multipartFile, "temp");
                                             OSSFile ossFile = new OSSFile();
                                             ossFile.setId(String.valueOf(SnowflakeUtils.snowflake()));
@@ -1556,6 +1557,7 @@ public class FileSplitItemsEOServiceImpl extends ServiceImpl<FileSplitItemsEOMap
                                         try {
 //                                            String paths = splitUrl+imageName;
 //                MinioUtil.uploadByte(bytev, filepathandname);
+                                            imgCheck(multipartFile);
                                             String upload = ObsBootUtil.upload(multipartFile, "temp");
                                             OSSFile ossFile = new OSSFile();
                                             ossFile.setId(String.valueOf(SnowflakeUtils.snowflake()));
@@ -1618,6 +1620,7 @@ public class FileSplitItemsEOServiceImpl extends ServiceImpl<FileSplitItemsEOMap
                                     try {
 //                                            String paths = splitUrl+imageName;
 //                MinioUtil.uploadByte(bytev, filepathandname);
+                                        imgCheck(multipartFile);
                                         String upload = ObsBootUtil.upload(multipartFile, "temp");
                                         OSSFile ossFile = new OSSFile();
                                         ossFile.setId(String.valueOf(SnowflakeUtils.snowflake()));
@@ -4484,6 +4487,7 @@ public class FileSplitItemsEOServiceImpl extends ServiceImpl<FileSplitItemsEOMap
 //                bizPath = "";
 //            }
         }
+        imgCheck(file);
 //        OSSFile oSSFile = ossFileService.uploadLocalForSplit(file, bizPath,state,cut);
         String upload = ObsBootUtil.upload(file, "temp");
         if (oConvertUtils.isNotEmpty(upload)) {
@@ -4499,9 +4503,9 @@ public class FileSplitItemsEOServiceImpl extends ServiceImpl<FileSplitItemsEOMap
             result.setResult(ossFile);
             result.setSuccess(true);
         } else {
-            if("cut".equals(LanguageEnum.CN.getValue())){
+            if(LanguageEnum.CN.equals(MessageUtils.getLanguage())) {
                 result.setMessage("上传失败！");
-            }else{
+            }else {
                 result.setMessage("fail to upload！");
             }
             result.setSuccess(false);
@@ -4509,4 +4513,25 @@ public class FileSplitItemsEOServiceImpl extends ServiceImpl<FileSplitItemsEOMap
         return result;
     }
 
+    private void imgCheck(MultipartFile file) {
+        // 格式校验
+        String orgName = file.getOriginalFilename();
+        String fileType = orgName.substring(orgName.lastIndexOf(".")).toLowerCase();
+        List<String> list = Arrays.asList(".jpg", ".jpeg", ".png");
+        if (!list.contains(fileType)){
+            if(LanguageEnum.CN.equals(MessageUtils.getLanguage())) {
+                throw new JeroBootException("只能上传: jpg、jpeg、png类型！");
+            }else{
+                throw new JeroBootException("Only jpg, jpeg, png can be uploaded！");
+            }
+        }
+        // 上传文件不超过20mb
+        if (file.getSize() > 20 * 1024 * 1024) {
+            if(LanguageEnum.CN.equals(MessageUtils.getLanguage())) {
+                throw new JeroBootException("上传文件大小不能超过20MB！");
+            }else {
+                throw new JeroBootException("Upload file size cannot exceed 20MB！");
+            }
+        }
+    }
 }