From ed8c7a5effd0d423ce1118b680ecdca6fe732609 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期三, 02 七月 2025 16:43:13 +0800 Subject: [PATCH] Merge branch 'master' of http://192.168.11.205:9000/r/P2022036_Service --- src/main/java/com/lf/server/service/all/BaseUploadService.java | 539 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 519 insertions(+), 20 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 76966f8..eb21b50 100644 --- a/src/main/java/com/lf/server/service/all/BaseUploadService.java +++ b/src/main/java/com/lf/server/service/all/BaseUploadService.java @@ -1,15 +1,21 @@ 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.helper.FileHelper; -import com.lf.server.helper.PathHelper; -import com.lf.server.helper.StringHelper; +import com.lf.server.entity.data.VerEntity; +import com.lf.server.entity.sys.UserEntity; +import com.lf.server.helper.*; +import com.lf.server.service.data.DirService; +import com.lf.server.service.sys.TokenService; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.support.StandardMultipartHttpServletRequest; +import javax.annotation.Resource; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.File; @@ -21,16 +27,30 @@ * 鐖朵笂浼犳湇鍔� * @author WWW */ +@Service public class BaseUploadService { - @Autowired - PathHelper pathHelper; + @Resource + protected DirService dirService; - private final Log log = LogFactory.getLog(getClass()); + @Resource + protected PathHelper pathHelper; + + @Resource + protected TokenService tokenService; + + public final Log log = LogFactory.getLog(getClass()); + + /** + * 鏌ヨ鐩綍 + */ + public String selectPath() { + return pathHelper.getTempPathName(); + } /** * 涓婁紶鏂囦欢 */ - public <T> List<MetaFileEntity> uploadData(T t, String path, HttpServletRequest req, HttpServletResponse res) throws Exception { + public <T> List<MetaFileEntity> uploadData(T t, String path, boolean isGetGuid, HttpServletRequest req, HttpServletResponse res) throws Exception { StandardMultipartHttpServletRequest request = (StandardMultipartHttpServletRequest) req; req.setCharacterEncoding("utf-8"); res.setContentType("application/json;charset=utf-8"); @@ -38,9 +58,8 @@ if (t != null) { setEntity(t, request); } - List<MetaFileEntity> list = getFiles(path, request); - return list; + return getFiles(path, isGetGuid, request); } /** @@ -83,25 +102,28 @@ /** * 鑾峰彇鏂囦欢 */ - private List<MetaFileEntity> getFiles(String subPath, StandardMultipartHttpServletRequest req) throws Exception { - List<MetaFileEntity> list = new ArrayList<MetaFileEntity>(); + public List<MetaFileEntity> getFiles(String subPath, boolean isGetGuid, StandardMultipartHttpServletRequest req) throws Exception { + List<MetaFileEntity> list = new ArrayList<>(); String path = pathHelper.getTempPath(subPath); Iterator<String> iterator = req.getFileNames(); while (iterator.hasNext()) { MultipartFile file = req.getFile(iterator.next()); - if (StringHelper.isEmpty(file.getOriginalFilename())) { + if (null == file || StringHelper.isEmpty(file.getOriginalFilename())) { continue; } - double sizes = FileHelper.sizeToMb(file.getSize()); MetaFileEntity mf = new MetaFileEntity(); mf.setName(file.getOriginalFilename()); + + double sizes = FileHelper.sizeToMb(file.getSize()); mf.setSizes(sizes); mf.setPath(path + File.separator + mf.getName()); file.transferTo(new File(mf.getPath())); - mf.setGuid(FileHelper.getFileMd5(mf.getPath())); + if (isGetGuid) { + mf.setGuid(FileHelper.getFileMd5(mf.getPath())); + } list.add(mf); } @@ -110,12 +132,41 @@ } /** - * 鑷悊瀹炰綋闆嗗悎 + * 鏌ヨ鏂囦欢 */ - public void dealEntities(List<MetaFileEntity> list) { - for (MetaFileEntity entity : list) { - entity.setPath(FileHelper.getRelativePath(entity.getPath())); + public List<MetaFileEntity> selectFiles(String subPath, boolean hasMd5) { + 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<>(); + for (File f : files) { + String fileName = FileHelper.getFileName(f.getPath()); + String extName = FileHelper.getExtension(fileName); + if (!isFileValid(f.getName())) { + continue; + } + + MetaFileEntity mf = new MetaFileEntity(); + mf.setName(fileName); + mf.setExtName(extName); + mf.setSizes(FileHelper.sizeToMb(f.length())); + mf.setPath(subPath + File.separator + fileName); + if (hasMd5 && !StaticData.ZIP.equals(extName)) { + mf.setGuid(FileHelper.getFileMd5(f.getPath())); + } + + list.add(mf); + } + + return list; } /** @@ -141,12 +192,460 @@ } /** - * 鑾峰彇鍙傛暟 + * 鏌ヨ鏄犲皠 + */ + public List<MetaFileEntity> selectMappers(UserEntity ue, String path, DirEntity dir, VerEntity ver, String epsgCode) { + List<MetaFileEntity> metas = selectFiles(path, true); + if (null == metas || metas.isEmpty()) { + return null; + } + + List<MetaFileEntity> list = new ArrayList<>(); + for (MetaFileEntity meta : metas) { + meta.setEventid(StringHelper.getGuid()); + meta.setDepcode(ue.getDepcode()); + meta.setVerid(ver.getId()); + meta.setCreateUser(ue.getId()); + meta.setEpsgCode(epsgCode); + + if (StaticData.ZIP.equals(meta.getExtName())) { + List<MetaFileEntity> subs = getMapperFiles(path, meta); + if (null != subs && subs.size() > 0) { + list.addAll(subs); + continue; + } + } + list.add(meta); + } + setMetaType(list); + setDircode(list, dir); + + return list; + } + + /** + * 鑾峰彇鏄犲皠鏂囦欢 + */ + private List<MetaFileEntity> getMapperFiles(String path, MetaFileEntity meta) { + String tempPath = pathHelper.getConfig().getTempPath(); + String zipFile = tempPath + 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", "") .replace("锛�","").replace("锛�","").replace(" ","") + String subPath = zipFolder.getPath(); + ZipHelper.unzip(zipFile, subPath); + + List<File> files = new ArrayList<>(); + getFilesByPath(files, subPath); + + return getMapperFiles(files, meta, tempPath.length() + 1); + } + + /** + * 鏍规嵁璺緞鑾峰彇鏂囦欢 + */ + private void getFilesByPath(List<File> list, String path) { + File file = new File(path); + if (!file.isDirectory()) { + if (isFileValid(file.getName())) { + list.add(file); + } + return; + } + + if (isGdbFile(file) || isOsgbFile(file)) { + list.add(file); + return; + } + + File[] files = file.listFiles(); + if (null == files) { + return; + } + for (File f : files) { + if (f.isDirectory()) { + getFilesByPath(list, f.getPath()); + } else { + if (isFileValid(f.getName())) { + list.add(f); + } + } + } + } + + /** + * 鏂囦欢鏄�/鍚︽湁鏁� + */ + private boolean isFileValid(String fileName) { + String extName = FileHelper.getExtension(fileName); + if (fileName.startsWith(StaticData.TILDE) || !StaticData.ALL_EXTENSION.contains(extName)) { + return false; + } + + fileName = fileName.toLowerCase(); + for (String ext : StaticData.MAPPER_EXCLUDE_EXT) { + if (fileName.contains(ext)) { + return false; + } + } + + return true; + } + + /** + * 鑾峰彇鏄犲皠鏂囦欢 + */ + private List<MetaFileEntity> getMapperFiles(List<File> files, MetaFileEntity meta, int start) { + List<MetaFileEntity> list = new ArrayList<>(); + for (File f : files) { + boolean isGdb = isGdbFile(f); + boolean isOsgb = isOsgbFile(f); + boolean isFile = isGdb || isOsgb; + if (f.isDirectory() && !isFile) { + continue; + } + + String fileName = FileHelper.getFileName(f.getPath()); + String extName = isOsgb ? StaticData.OSGB : FileHelper.getExtension(fileName); + if (!isFileValid(f.getName())) { + continue; + } + + if (isGdb) { + List<MetaFileEntity> rs = getGdbMappers(f, meta, start); + if (null != rs && rs.size() > 0) { + list.addAll(rs); + } + continue; + } + if (isOsgb) { + list.add(getOsgbMapper(f, meta, start)); + continue; + } + + MetaFileEntity mf = createMetaFileEntity(f, meta, fileName, extName); + mf.setPath(f.getPath().substring(start)); + list.add(mf); + } + + return list; + } + + /** + * 鍒涘缓鍏冩暟鎹枃浠跺疄浣撶被 + */ + private MetaFileEntity createMetaFileEntity(File f, MetaFileEntity meta, String fileName, String extName) { + boolean isShp = StaticData.SHP.equals(extName); + + MetaFileEntity mf = createMetaFileEntity(meta); + mf.setEventid(StringHelper.getGuid()); + mf.setName(fileName); + mf.setExtName(extName); + + if (isShp) { + List<String> shpFiles = getShpFiles(f.getPath()); + mf.setTab(fileName.replace(StaticData.SHP, "")); + mf.setSizes(getFilesSize(shpFiles)); + mf.setGuid(getFilesMd5(shpFiles)); + } else { + mf.setSizes(FileHelper.sizeToMb(f.length())); + mf.setGuid(FileHelper.getFileMd5(f.getPath())); + } + + return mf; + } + + /** + * 鍒涘缓鍏冩暟鎹枃浠跺疄浣撶被 + */ + private MetaFileEntity createMetaFileEntity(MetaFileEntity meta) { + MetaFileEntity mf = new MetaFileEntity(); + mf.setDepcode(meta.getDepcode()); + mf.setVerid(meta.getVerid()); + mf.setCreateUser(meta.getCreateUser()); + mf.setEpsgCode(meta.getEpsgCode()); + + return mf; + } + + /** + * 鑾峰彇SHP鏂囦欢闆嗗悎 + */ + private List<String> getShpFiles(String shpPath) { + List<String> list = new ArrayList<>(); + list.add(shpPath); + + for (int i = 0, c = StaticData.SHP_EXT.size(); i < c; i++) { + String path = shpPath.replace(".shp", StaticData.SHP_EXT.get(i)); + + File f = new File(path); + if (f.exists() && !f.isDirectory()) { + list.add(path); + } + } + + return list; + } + + /** + * 鑾峰彇GDB鏂囦欢闆嗗悎 + */ + private List<String> getGdbFiles(String path) { + List<String> list = new ArrayList<>(); + + File[] files = new File(path).listFiles(); + if (null == files || files.length == 0) { + return list; + } + for (File f : files) { + list.add(f.getPath()); + } + + return list; + } + + /** + * 鑾峰彇澶氭枃浠剁殑MD5 + */ + private String getFilesMd5(List<String> files) { + List<String> list = new ArrayList<>(); + for (String file : files) { + String md5 = FileHelper.getFileMd5(file); + if (null != md5) { + list.add(md5); + } + } + + if (list.size() > 0) { + String str = StringHelper.join(list, ","); + return FileHelper.getStringMd5(str); + } + + return StringHelper.getGuid(); + } + + /** + * 鑾峰彇澶氭枃浠剁殑澶у皬 + */ + private double getFilesSize(List<String> files) { + long size = 0L; + for (String file : files) { + File f = new File(file); + if (f.exists() && !f.isDirectory()) { + size += f.length(); + } + } + + return FileHelper.sizeToMb(size); + } + + /** + * 鑾峰彇GDB鏂囦欢鏄犲皠 + */ + private List<MetaFileEntity> getGdbMappers(File f, MetaFileEntity meta, int start) { + List<String> tabs = GdbHelper.getTabNames(f.getPath()); + if (null == tabs || tabs.size() == 0) { + return null; + } + + String fileName = FileHelper.getFileName(f.getPath()); + String extName = FileHelper.getExtension(fileName); + + List<String> gdbFiles = getGdbFiles(f.getPath()); + String md5 = getFilesMd5(gdbFiles); + double sizes = getFilesSize(gdbFiles); + + List<MetaFileEntity> list = new ArrayList<>(); + for (String tab : tabs) { + MetaFileEntity mf = createMetaFileEntity(meta); + mf.setEventid(StringHelper.getGuid()); + mf.setName(fileName + "\\" + tab); + mf.setExtName(extName); + mf.setSizes(FileHelper.sizeToMb(f.length())); + mf.setPath(f.getPath().substring(start)); + mf.setTab(tab); + mf.setSizes(sizes); + mf.setGuid(md5); + list.add(mf); + } + + return list; + } + + /** + * 鏄�/鍚︿负GDB鏂囦欢 + */ + private boolean isGdbFile(File f) { + if (f.isDirectory() && f.getName().toLowerCase().endsWith(StaticData.GDB)) { + File gdb = new File(f.getPath() + File.separator + "gdb"); + + return gdb.exists() && !gdb.isDirectory(); + } + + return false; + } + + /** + * 璁剧疆鍏冩暟鎹枃浠剁殑绫诲瀷 + */ + private void setMetaType(List<MetaFileEntity> list) { + for (MetaFileEntity mf : list) { + if (null != mf.getExtName()) { + mf.setType(mf.getExtName().replace(".", "")); + } + } + } + + /** + * 澶勭悊鐩綍 + */ + public void copePath(List<MetaFileEntity> list) { + String basePath = pathHelper.getConfig().getTempPath() + File.separator; + for (MetaFileEntity mf : list) { + mf.setPath(mf.getPath().replace(basePath, "")); + } + } + + /** + * 鏄�/鍚︿负OSGB鏂囦欢 + */ + private boolean isOsgbFile(File f) { + if (f.isDirectory()) { + File meta = new File(f.getPath() + File.separator + "metadata.xml"); + File data = new File(f.getPath() + File.separator + "Data"); + + return meta.exists() && !meta.isDirectory() && data.exists() && data.isDirectory(); + } + + return false; + } + + /** + * 鑾峰彇OSGB鏂囦欢鏄犲皠 + */ + private MetaFileEntity getOsgbMapper(File f, MetaFileEntity meta, int start) { + String fileName = FileHelper.getFileName(f.getPath()); + + List<String> files = new ArrayList<>(); + FileHelper.getFilesByPath(files, f.getPath()); + + String md5 = getFilesMd5(files); + double sizes = getFilesSize(files); + + MetaFileEntity mf = createMetaFileEntity(meta); + mf.setEventid(StringHelper.getGuid()); + mf.setName(fileName); + mf.setExtName(StaticData.OSGB); + mf.setSizes(FileHelper.sizeToMb(f.length())); + mf.setPath(f.getPath().substring(start)); + mf.setSizes(sizes); + mf.setGuid(md5); + + return mf; + } + + /** + * 璁剧疆鐩綍缂栫爜 + */ + private void setDircode(List<MetaFileEntity> list, DirEntity dir) { + List<DirEntity> dirs = dirService.selectRecursiveById(dir.getId()); + for (MetaFileEntity mfe : list) { + String filePath = mfe.getPath().toLowerCase().replace("/", "\\"); + String fileName = mfe.getName().toLowerCase(); + + String code = findDirByPath(dirs, filePath, fileName); + if (StringHelper.isEmpty(code)) { + DirEntity baseDir = findBaseDirByPath(dirs, filePath); + if (null != baseDir) { + createDirByPath(baseDir, mfe.getPath().replace("/", "\\")); + dirs = dirService.selectRecursiveById(dir.getId()); + code = findDirByPath(dirs, filePath, fileName); + } + } + code = StringHelper.isEmpty(code) ? dir.getCode() : code; + mfe.setDircode(code); + } + } + + /** + * 鏍规嵁璺緞鏌ユ壘鐩綍 + */ + private String findDirByPath(List<DirEntity> dirs, String filePath,String fileName) { + for (DirEntity de : dirs) { + if (filePath.contains(de.getFullName().toLowerCase() + "\\" + fileName)) { + return de.getCode(); + } + } + + return null; + } + + /** + * 鏍规嵁璺緞鏌ユ壘鐖剁洰褰� + */ + private DirEntity findBaseDirByPath(List<DirEntity> dirs, String filePath) { + for (DirEntity de : dirs) { + if (StringHelper.isEmpty(de.getCode()) || de.getCode().length() != StaticData.FOUR) { + continue; + } + if (filePath.contains(de.getFullName().toLowerCase() + "\\")) { + return de; + } + } + + return null; + } + + /** + * 鏍规嵁璺緞鍒涘缓鐩綍 + */ + private void createDirByPath(DirEntity baseDir, String filePath) { + int start = filePath.toLowerCase().indexOf(baseDir.getFullName().toLowerCase() + "\\") + baseDir.getFullName().length() + 1; + int end = filePath.lastIndexOf("\\"); + if (end <= start) { + return; + } + if (0 == baseDir.getLevel()) { + baseDir.setLevel(2); + } + + HttpServletRequest req = WebHelper.getRequest(); + UserEntity ue = tokenService.getCurrentUser(req); + + String[] strs = filePath.substring(start, end).split("\\\\"); + for (String str : strs) { + DirEntity entity = dirService.selectDirByName(str, baseDir.getId()); + if (null == entity) { + entity = new DirEntity(); + entity.setPid(baseDir.getId()); + entity.setName(str); + entity.setOrderNum(dirService.selectMaxOrderNum()); + entity.setLevel(baseDir.getLevel() + 1); + entity.setCreateUser(ue.getId()); + + dirService.insert(entity); + baseDir = dirService.selectDir(entity.getId()); + } else { + baseDir = entity; + } + } + } + + /** + * 鑾峰彇鍙傛暟 * * Enumeration<String> headers = req.getHeaderNames(); * Enumeration<String> attributes = req.getAttributeNames(); */ public Map<String, String> getParams(StandardMultipartHttpServletRequest req) { - Map<String, String> map = new HashMap<String, String>(3); + Map<String, String> map = new HashMap<>(3); Enumeration<String> enumeration = req.getParameterNames(); while (enumeration.hasMoreElements()) { -- Gitblit v1.9.3