From 3af5e8f2335b3d5cd4528f9ee9f326eef5fe2f9a Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期二, 10 一月 2023 10:43:01 +0800 Subject: [PATCH] 3 --- src/main/java/com/lf/server/service/all/BaseUploadService.java | 237 ++++++++++++++++++++++++++++++++++++++--------------------- 1 files changed, 153 insertions(+), 84 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 f0a614d..62fb8d6 100644 --- a/src/main/java/com/lf/server/service/all/BaseUploadService.java +++ b/src/main/java/com/lf/server/service/all/BaseUploadService.java @@ -1,7 +1,10 @@ package com.lf.server.service.all; +import com.lf.server.entity.all.StaticData; import com.lf.server.entity.data.DirEntity; import com.lf.server.entity.data.MetaFileEntity; +import com.lf.server.entity.data.VerEntity; +import com.lf.server.entity.sys.UserEntity; import com.lf.server.helper.*; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; @@ -23,18 +26,6 @@ */ @Service public class BaseUploadService { - private final static String XLS = ".xls"; - - private final static String MDB = ".mdb"; - - private final static String GDB = ".gdb"; - - private final static String SHP_ZIP = ".shp.zip"; - - private final static String GDB_ZIP = ".gdb.zip"; - - private final static String ZIP = ".zip"; - @Autowired public PathHelper pathHelper; @@ -157,7 +148,9 @@ mf.setExtName(extName); mf.setSizes(FileHelper.sizeToMb(f.length())); mf.setPath(subPath + File.separator + fileName); - // mf.setGuid(FileHelper.getFileMd5(f.getPath())) + if (!StaticData.ZIP.equals(extName)) { + mf.setGuid(FileHelper.getFileMd5(f.getPath())); + } list.add(mf); } @@ -190,98 +183,171 @@ /** * 鏌ヨ鏄犲皠 */ - public List<MetaFileEntity> selectMappers(List<MetaFileEntity> metas, DirEntity dir, String epsgCode) { + public List<MetaFileEntity> selectMappers(UserEntity ue, String path, DirEntity dir, VerEntity ver, String epsgCode) { + List<MetaFileEntity> metas = selectFiles(path, StaticData.ALL_EXTENSION); + if (null == metas || metas.isEmpty()) { + return null; + } + List<MetaFileEntity> list = new ArrayList<>(); for (MetaFileEntity meta : metas) { - // - } + meta.setEventid(StringHelper.getGuid()); + meta.setDirid(dir.getId()); + meta.setDepid(ue.getDepid()); + meta.setVerid(ver.getId()); + meta.setCreateUser(ue.getId()); + meta.setEpsgCode(epsgCode); - return list; - } - - /*private List<TabMapperEntity> getMappers(String zipPath, File[] files) { - String temp = pathHelper.getConfig().getTempPath(); - - List<TabMapperEntity> list = new ArrayList<>(); - for (File f : files) { - String fileName = FileHelper.getFileName(f.getPath()); - if (fileName.contains(XLS)) { - String path = f.getPath().replace(temp + File.separator, ""); - list.add(new TabMapperEntity(fileName, "xls", null, path)); - continue; - } - if (fileName.contains(MDB)) { - String path = f.getPath().replace(temp + File.separator, ""); - List<String> tabs = MdbHelper.getTabNames(f.getPath()); - for (String tab : tabs) { - list.add(new TabMapperEntity(fileName, "mdb", tab, path)); + if (StaticData.ZIP.equals(meta.getExtName())) { + List<MetaFileEntity> subs = getMapperFiles(path, meta); + if (null != subs && subs.size() > 0) { + list.addAll(subs); + continue; } - continue; } - if (fileName.contains(SHP_ZIP)) { - String subPath = zipPath + File.separator + f.getName().toLowerCase().replace(".zip", ""); - ZipHelper.unzip(f.getPath(), subPath); - getShpFiles(f.getName(), subPath, list); - continue; - } - if (fileName.contains(GDB_ZIP)) { - String subPath = zipPath + File.separator + f.getName().toLowerCase().replace(".zip", ""); - ZipHelper.unzip(f.getPath(), subPath); - getGdbFiles(f.getName(), subPath, list); - continue; - } - if (fileName.contains(ZIP)) { - // 鏆傛椂涓嶅疄鐜� - } + + meta.setEventid(StringHelper.getGuid()); + list.add(meta); } return list; } - private void getShpFiles(String sourceName, String subPath, List<TabMapperEntity> list) { - List<String> files = new ArrayList<>(); - getShpFiles(subPath, files); - - String root = pathHelper.getConfig().getTempPath() + File.separator; - for (String file : files) { - String name = FileHelper.getFileName(file); - String path = file.replace(root, ""); - list.add(new TabMapperEntity(sourceName, "shp", name, path)); + /** + * 鑾峰彇鏄犲皠鏂囦欢 + */ + private List<MetaFileEntity> getMapperFiles(String path, MetaFileEntity meta) { + String zipFile = pathHelper.getConfig().getTempPath() + File.separator + meta.getPath(); + File file = new File(zipFile); + if (!file.exists() || file.isDirectory()) { + return null; } + + File zipFolder = new File(pathHelper.getConfig().getTempPath() + File.separator + path + "_zip"); + if (!zipFolder.exists() || !zipFolder.isDirectory()) { + zipFolder.mkdirs(); + } + + String subPath = zipFolder + File.separator + meta.getName().toLowerCase().replace(".zip", ""); + ZipHelper.unzip(zipFile, subPath); + + File subFile = new File(subPath); + File[] files = subFile.listFiles(); + if (null == files || files.length == 0) { + return null; + } + + return getMapperFiles(files, meta, subPath); } - private void getShpFiles(String shpPath, List<String> list) { - File file = new File(shpPath); - - File[] files = file.listFiles(); - if (null == files || files.length == 0) { - return; - } - + /** + * 鑾峰彇鏄犲皠鏂囦欢 + */ + private List<MetaFileEntity> getMapperFiles(File[] files, MetaFileEntity meta,String subPath) { + List<MetaFileEntity> list = new ArrayList<>(); for (File f : files) { - if (f.isDirectory()) { - getShpFiles(f.getPath(), list); + boolean isGdb = isGdbFile(f); + if (!isGdb && f.isDirectory()) { continue; } - if (f.getName().toLowerCase().endsWith(".shp")) { - list.add(f.getPath()); + String fileName = FileHelper.getFileName(f.getPath()); + String extName = FileHelper.getExtension(fileName); + if (!StaticData.ALL_EXTENSION.contains(extName)) { + continue; } + + boolean isShp = StaticData.SHP.equals(extName); + if (isGdb) { + String md5 = getGdbMd5(f.getPath()); + continue; + } + + MetaFileEntity mf = createMetaFileEntity(meta); + mf.setName(fileName); + mf.setExtName(extName); + mf.setSizes(FileHelper.sizeToMb(f.length())); + mf.setPath(f.getPath().substring(subPath.length())); + if (isShp) { + mf.setTab(fileName); + mf.setGuid(getShpMd5(f.getPath())); + } else { + mf.setGuid(FileHelper.getFileMd5(f.getPath())); + } + list.add(mf); } + + return list; } - private void getGdbFiles(String sourceName, String subPath, List<TabMapperEntity> list) { - List<String> files = new ArrayList<>(); - getGdbFiles(subPath, files); + /** + * 鍒涘缓鍏冩暟鎹枃浠跺疄浣撶被 + */ + private MetaFileEntity createMetaFileEntity(MetaFileEntity meta) { + MetaFileEntity mf = new MetaFileEntity(); + mf.setEventid(StringHelper.getGuid()); + mf.setDirid(meta.getDirid()); + mf.setDepid(meta.getDepid()); + mf.setVerid(meta.getVerid()); + mf.setCreateUser(meta.getCreateUser()); + mf.setEpsgCode(meta.getEpsgCode()); - String root = pathHelper.getConfig().getTempPath() + File.separator; - for (String file : files) { - String path = file.replace(root, ""); - List<String> tabs = GdbHelper.getTabNames(file); - for (String tab : tabs) { - list.add(new TabMapperEntity(sourceName, "gdb", tab, path)); + return mf; + } + + /** + * 鑾峰彇SHP鏂囦欢鐨凪D5鐮� + */ + private String getShpMd5(String shpPath) { + List<String> files = new ArrayList<>(); + files.add(shpPath); + + for (int i = 1, c = StaticData.SHP_EXTENSION.size(); i < c; i++) { + String path = shpPath.replace(".shp", StaticData.SHP_EXTENSION.get(i)); + + File f = new File(path); + if (f.exists() && !f.isDirectory()) { + files.add(path); } } + + return getFilesMd5(files); + } + + /** + * 鑾峰彇GDB鏂囦欢鐨凪D5鐮� + */ + private String getGdbMd5(String path) { + File[] files = new File(path).listFiles(); + if (null == files || files.length == 0) { + return StringHelper.getGuid(); + } + + List<String> list = new ArrayList<>(); + for (File f : files) { + list.add(f.getPath()); + } + + return getFilesMd5(list); + } + + /** + * 鑾峰彇澶氭枃浠剁殑MD5 + */ + private String getFilesMd5(List<String> list) { + StringBuilder sb = new StringBuilder(); + for (String str : list) { + String md5 = FileHelper.getFileMd5(str); + if (null != md5) { + sb.append(sb + ","); + } + } + + if (sb.length() > 0) { + return FileHelper.getStringMd5(sb.toString()); + } + + return StringHelper.getGuid(); } private void getGdbFiles(String shpPath, List<String> list) { @@ -304,10 +370,13 @@ getGdbFiles(f.getPath(), list); } - }*/ + } + /** + * 鏄�/鍚︿负GDB鏂囦欢 + */ private boolean isGdbFile(File f) { - if (f.getName().toLowerCase().endsWith(GDB)) { + if (f.isDirectory() && f.getName().toLowerCase().endsWith(StaticData.GDB)) { File[] files = f.listFiles(); if (null == files || files.length == 0) { return false; @@ -324,7 +393,7 @@ } /** - * 鑾峰彇鍙傛暟 + * 鑾峰彇鍙傛暟 * * Enumeration<String> headers = req.getHeaderNames(); * Enumeration<String> attributes = req.getAttributeNames(); */ -- Gitblit v1.9.3