文件上传

src/main/java/com/izouma/wenlvju/service/performance/ProgrammeService.java

@@ -40,7 +40,7 @@ import javax.servlet.ServletOutputStream;
 import javax.servlet.http.HttpServletResponse;
 import javax.transaction.Transactional;
 import java.io.*;
-import java.math.BigDecimal;
+import java.nio.charset.Charset;
 import java.nio.charset.StandardCharsets;
 import java.text.SimpleDateFormat;
 import java.time.LocalDate;
@@ -241,7 +241,7 @@ public class ProgrammeService {
     }
 
     @Transactional(rollbackOn = Exception.class)
-    public void upload(MultipartFile file, Long userId) throws IOException {
+    public void upload(MultipartFile file, Long userId) throws Exception {
         Organization organization = organizationRepo.findByUserId(userId).orElseThrow(new BusinessException("无记录"));
         Long performance = performanceRepo.findNow(LocalDate.now());
         if (ObjectUtil.isNull(performance)) {
@@ -249,17 +249,15 @@ public class ProgrammeService {
         }
 
         File destDir = TempFile.createTempDirectory("import");
-        ZipUtil.unzip(file.getInputStream(), destDir, StandardCharsets.UTF_8);
-
-//        File xlsxFile = FileUtils.findExcel1(destDir);
-//        if (xlsxFile == null) {
-//            Map<String, File> map = FileUtils.findExcel(destDir);
-//            if (map == null) return;
-//
-//            xlsxFile = map.get("file");
-//            destDir = map.get("destDir");
-//        }
-        File xlsxFile = FileUtils.folderMethod1(destDir, null);
+
+
+        try {
+            ZipUtil.unzip(file.getInputStream(), destDir, StandardCharsets.UTF_8);
+        } catch (Exception e) {
+            ZipUtil.unzip(file.getInputStream(), destDir, Charset.forName("gbk"));
+        }
+
+        File xlsxFile = FileUtils.findInDir(destDir, null);
         if (xlsxFile == null) {
             return;
         }
@@ -304,10 +302,10 @@ public class ProgrammeService {
                     programme.setLevelSettingId(mapMap.get(4).get(level));
                 }
 
-                File uploadFile = FileUtils.folderMethod1(destDir, dto.getName());
+                File uploadFile = FileUtils.findInDir(destDir, dto.getName());
                 if (dto.getVideo() != null) {
 //                    File uploadFile = new File(destDir, dto.getVideo());
-                    uploadFile = FileUtils.folderMethod1(destDir, dto.getVideo());
+                    uploadFile = FileUtils.findInDir(destDir, dto.getVideo());
                 }
                 if (ObjectUtil.isNotNull(uploadFile) && uploadFile.exists()) {
                     // 视频
@@ -350,9 +348,9 @@ public class ProgrammeService {
             File uploadImg;
             if (dto.getImg() != null) {
 //                File uploadFile = new File(destDir, dto.getImg());
-                uploadImg = FileUtils.folderMethod1(destDir, dto.getImg());
+                uploadImg = FileUtils.findInDir(destDir, dto.getImg());
             } else {
-                uploadImg = FileUtils.folderMethod1(destDir, dto.getIdNo() + "证件照");
+                uploadImg = FileUtils.findInDir(destDir, dto.getIdNo() + "证件照");
             }
             if (ObjectUtil.isNotNull(uploadImg) && uploadImg.exists()) {
                 participant.setImg(this.saveImg(uploadImg));
@@ -360,9 +358,9 @@ public class ProgrammeService {
 
             File uploadCertificate;
             if (dto.getCertificate() != null) {
-                uploadCertificate = FileUtils.folderMethod1(destDir, dto.getCertificate());
+                uploadCertificate = FileUtils.findInDir(destDir, dto.getCertificate());
             } else {
-                uploadCertificate = FileUtils.folderMethod1(destDir, dto.getIdNo() + "证书");
+                uploadCertificate = FileUtils.findInDir(destDir, dto.getIdNo() + "证书");
             }
             if (ObjectUtil.isNotNull(uploadCertificate) && uploadCertificate.exists()) {
                 participant.setCertificate(this.saveImg(uploadCertificate));

src/main/java/com/izouma/wenlvju/utils/FileUtils.java

@@ -1,7 +1,8 @@
 package com.izouma.wenlvju.utils;
 
-import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.StrUtil;
+import org.apache.commons.io.FilenameUtils;
 import org.apache.commons.lang3.StringUtils;
 
 import java.io.*;
@@ -12,6 +13,7 @@ import java.nio.file.attribute.PosixFileAttributes;
 import java.nio.file.attribute.PosixFilePermission;
 import java.nio.file.attribute.PosixFilePermissions;
 import java.util.*;
+import java.util.regex.Pattern;
 
 public class FileUtils {
 
@@ -241,6 +243,30 @@ public class FileUtils {
         return null;
     }
 
+    public static File findInDir(File dir, String fileName) {
+        if (dir.isDirectory()) {
+            for (File file : dir.listFiles()) {
+                if (file.isDirectory()) {
+                    File f = findInDir(file, fileName);
+                    if (f != null) {
+                        return f;
+                    }
+                } else {
+                    if (fileName != null && file.getName().equalsIgnoreCase(fileName)) {
+                        return file;
+                    } else if (!file.isHidden() && Pattern.matches("xls|xlsx", FilenameUtils.getExtension(file.getName()))) {
+                        return file;
+                    }
+                }
+            }
+        } else if (fileName != null && dir.getName().equalsIgnoreCase(fileName)) {
+            return dir;
+        } else if (!dir.isHidden() && Pattern.matches("xls|xlsx", FilenameUtils.getExtension(dir.getName()))) {
+            return dir;
+        }
+        return null;
+    }
+
     public static File folderMethod1(File file, String fileName) {
         LinkedList<File> list = new LinkedList<>();
 
@@ -248,7 +274,9 @@ public class FileUtils {
             if (null == file.listFiles()) {
                 return null;
             }
-            list.addAll(Arrays.asList(file.listFiles()));
+
+            list.addAll(CollUtil.newArrayList(file.listFiles()));
+
             while (!list.isEmpty()) {
                 File[] files = list.removeFirst().listFiles();
                 if (null == files) {
@@ -264,7 +292,7 @@ public class FileUtils {
                                 return f;
                             }
                         } else {
-                            if ((name.endsWith(".xlsx") || name.endsWith(".xls")) && !name.startsWith(".") && !file.isHidden()) {
+                            if ((name.endsWith(".xlsx") || name.endsWith(".xls")) && !name.startsWith(".") && !f.isHidden()) {
                                 return f;
                             }
                         }

src/main/java/com/izouma/wenlvju/web/performance/ProgrammeController.java

@@ -104,6 +104,8 @@ public class ProgrammeController extends BaseController {
         } catch (IOException e) {
             log.error("上传失败", e);
             throw new BusinessException("上传失败", e.getMessage());
+        } catch (Exception e) {
+            e.printStackTrace();
         }
     }
 

src/main/vue/src/views/performance/ProgrammeOrgList.vue

@@ -589,6 +589,7 @@ export default {
                 });
         },
         downloadTemp() {
+            this.downloading = true;
             this.$axios
                 .get('/programme/excelTemp', {
                     responseType: 'blob'