From e44478a894a45b684112b4680852792641315413 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 21 十一月 2022 09:48:26 +0800
Subject: [PATCH] 数据上传+数据入库:修改获取文件功能

---
 src/main/java/com/lf/server/service/data/DataLoaderService.java |   39 +++++++++++++++++++++++++++++++++++++--
 1 files changed, 37 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/lf/server/service/data/DataLoaderService.java b/src/main/java/com/lf/server/service/data/DataLoaderService.java
index 5846fce..765fef7 100644
--- a/src/main/java/com/lf/server/service/data/DataLoaderService.java
+++ b/src/main/java/com/lf/server/service/data/DataLoaderService.java
@@ -15,7 +15,9 @@
 import org.springframework.stereotype.Service;
 
 import java.io.File;
+import java.io.IOException;
 import java.lang.reflect.Field;
+import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -34,9 +36,42 @@
 
     private final static String MDB = ".mdb";
 
-    private final static String SHP = ".shp";
+    private final static String SHP = ".shp.zip";
 
-    private final static String GDB = ".gdb";
+    private final static String GDB = ".gdb.zip";
+
+    /**
+     * 鏌ヨ鏂囦欢
+     */
+    public List<MetaFileEntity> selectFiles(String subPath) throws IOException {
+        String root = pathHelper.getConfig().getTempPath() + File.separator + subPath;
+
+        File file = new File(root);
+        if (!file.exists() && !file.isDirectory()) {
+            return null;
+        }
+        File[] files = file.listFiles();
+        if (null == files || files.length == 0) {
+            return null;
+        }
+
+        List<MetaFileEntity> list = new ArrayList<MetaFileEntity>();
+        for (File f : files) {
+            String fileName = FileHelper.getFileName(f.getPath());
+            double sizes = FileHelper.sizeToMb(f.length());
+            String filePath = subPath + File.separator + fileName;
+
+            MetaFileEntity mf = new MetaFileEntity();
+            mf.setName(fileName);
+            mf.setSizes(sizes);
+            mf.setPath(filePath);
+            mf.setGuid(FileHelper.getFileMd5(f.getPath()));
+
+            list.add(mf);
+        }
+
+        return list;
+    }
 
     /**
      * 鎻掑叆鏂囦欢

--
Gitblit v1.9.3