From 65b7935caa01d5bc3e80066eab9576d329a5867d Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期日, 08 一月 2023 14:16:07 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/service/all/BaseUploadService.java |   43 +++++++++++++++++++++++++------------------
 1 files changed, 25 insertions(+), 18 deletions(-)

diff --git a/src/main/java/com/lf/server/service/all/BaseUploadService.java b/src/main/java/com/lf/server/service/all/BaseUploadService.java
index 0e98187..bd4f342 100644
--- a/src/main/java/com/lf/server/service/all/BaseUploadService.java
+++ b/src/main/java/com/lf/server/service/all/BaseUploadService.java
@@ -1,6 +1,6 @@
 package com.lf.server.service.all;
 
-import com.lf.server.entity.data.MetaFileEntity;
+import com.lf.server.entity.data.MetaEntity;
 import com.lf.server.helper.FileHelper;
 import com.lf.server.helper.PathHelper;
 import com.lf.server.helper.StringHelper;
@@ -31,16 +31,9 @@
     public final Log log = LogFactory.getLog(getClass());
 
     /**
-     * 鏌ヨ鐩綍
-     */
-    public String selectPath() {
-        return pathHelper.getTempPathName();
-    }
-
-    /**
      * 鏌ヨ鏂囦欢
      */
-    public List<MetaFileEntity> selectFiles(String subPath) throws IOException {
+    public List<MetaEntity> selectFiles(String subPath, List<String> extList) throws IOException {
         String root = pathHelper.getConfig().getTempPath() + File.separator + subPath;
 
         File file = new File(root);
@@ -52,13 +45,20 @@
             return null;
         }
 
-        List<MetaFileEntity> list = new ArrayList<MetaFileEntity>();
+        List<MetaEntity> list = new ArrayList<MetaEntity>();
         for (File f : files) {
             String fileName = FileHelper.getFileName(f.getPath());
+            if (null != extList) {
+                String extName = FileHelper.getExtension(fileName);
+                if (!extList.contains(extName)) {
+                    continue;
+                }
+            }
+
             double sizes = FileHelper.sizeToMb(f.length());
             String filePath = subPath + File.separator + fileName;
 
-            MetaFileEntity mf = new MetaFileEntity();
+            MetaEntity mf = new MetaEntity();
             mf.setName(fileName);
             mf.setSizes(sizes);
             mf.setPath(filePath);
@@ -71,9 +71,16 @@
     }
 
     /**
+     * 鏌ヨ鐩綍
+     */
+    public String selectPath() {
+        return pathHelper.getTempPathName();
+    }
+
+    /**
      * 涓婁紶鏂囦欢
      */
-    public <T> List<MetaFileEntity> uploadData(T t, String path, HttpServletRequest req, HttpServletResponse res) throws Exception {
+    public <T> List<MetaEntity> uploadData(T t, String path, HttpServletRequest req, HttpServletResponse res) throws Exception {
         StandardMultipartHttpServletRequest request = (StandardMultipartHttpServletRequest) req;
         req.setCharacterEncoding("utf-8");
         res.setContentType("application/json;charset=utf-8");
@@ -81,7 +88,7 @@
         if (t != null) {
             setEntity(t, request);
         }
-        List<MetaFileEntity> list = getFiles(path, request);
+        List<MetaEntity> list = getFiles(path, request);
 
         return list;
     }
@@ -126,8 +133,8 @@
     /**
      * 鑾峰彇鏂囦欢
      */
-    public List<MetaFileEntity> getFiles(String subPath, StandardMultipartHttpServletRequest req) throws Exception {
-        List<MetaFileEntity> list = new ArrayList<MetaFileEntity>();
+    public List<MetaEntity> getFiles(String subPath, StandardMultipartHttpServletRequest req) throws Exception {
+        List<MetaEntity> list = new ArrayList<MetaEntity>();
 
         String path = pathHelper.getTempPath(subPath);
         Iterator<String> iterator = req.getFileNames();
@@ -138,7 +145,7 @@
             }
 
             double sizes = FileHelper.sizeToMb(file.getSize());
-            MetaFileEntity mf = new MetaFileEntity();
+            MetaEntity mf = new MetaEntity();
             mf.setName(file.getOriginalFilename());
             mf.setSizes(sizes);
             mf.setPath(path + File.separator + mf.getName());
@@ -155,11 +162,11 @@
     /**
      * 鍒犻櫎鏂囦欢
      */
-    public Integer deleteFiles(List<MetaFileEntity> list) {
+    public Integer deleteFiles(List<MetaEntity> list) {
         String root = pathHelper.getConfig().getTempPath();
 
         int count = 0;
-        for (MetaFileEntity entity : list) {
+        for (MetaEntity entity : list) {
             if (!StringHelper.isEmpty(entity.getPath())) {
                 String file = root + File.separator + entity.getPath();
 

--
Gitblit v1.9.3