From 05ffa1a7f490e1e81d6fbf1ee948db50b1fff6fc Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期二, 14 二月 2023 16:02:52 +0800 Subject: [PATCH] 修改depcode、dircode值 --- src/main/java/com/lf/server/service/data/UploadService.java | 535 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 531 insertions(+), 4 deletions(-) diff --git a/src/main/java/com/lf/server/service/data/UploadService.java b/src/main/java/com/lf/server/service/data/UploadService.java index eb001b4..3749d34 100644 --- a/src/main/java/com/lf/server/service/data/UploadService.java +++ b/src/main/java/com/lf/server/service/data/UploadService.java @@ -1,12 +1,24 @@ package com.lf.server.service.data; -import com.lf.server.entity.data.CoordEntity; -import com.lf.server.entity.data.DirEntity; -import com.lf.server.entity.data.FmeLogEntity; +import com.lf.server.entity.all.BaseEntity; +import com.lf.server.entity.all.StaticData; +import com.lf.server.entity.data.*; +import com.lf.server.entity.sys.DepEntity; +import com.lf.server.entity.sys.UserEntity; +import com.lf.server.helper.*; +import com.lf.server.mapper.all.BasicMapper; import com.lf.server.mapper.data.UploadMapper; +import com.lf.server.service.all.BaseQueryService; +import com.lf.server.service.all.BaseUploadService; +import com.lf.server.service.sys.DepService; +import org.apache.commons.io.FileUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; +import javax.servlet.http.HttpServletRequest; +import java.io.File; +import java.sql.Timestamp; +import java.util.ArrayList; import java.util.List; /** @@ -14,13 +26,30 @@ * @author WWW */ @Service -public class UploadService implements UploadMapper { +public class UploadService extends BaseUploadService implements UploadMapper { @Autowired UploadMapper uploadMapper; + + @Autowired + FmeService fmeService; + + @Autowired + MetaService metaService; + + @Autowired + DepService depService; + + @Autowired + DirService dirService; @Override public List<CoordEntity> selectCoords(String zoning) { return uploadMapper.selectCoords(zoning); + } + + @Override + public Integer selectCount4Coord(String epsgCode) { + return uploadMapper.selectCount4Coord(epsgCode); } @Override @@ -32,4 +61,502 @@ public List<FmeLogEntity> selectFmeLog(String parentid) { return uploadMapper.selectFmeLog(parentid); } + + /** + * 鎻掑叆鏂囦欢 + */ + public void insertFiles(UserEntity ue, List<MetaFileEntity> list, HttpServletRequest req) { + checkMetaFiles(ue, list); + excelLoader(list, req); + loadData(list); + copyFiles(list); + publishFiles(list); + insertMetas(list); + } + + /** + * 妫�鏌ュ厓鏁版嵁鏂囦欢 + */ + private void checkMetaFiles(UserEntity ue, List<MetaFileEntity> list) { + Timestamp createTime = WebHelper.getCurrentTimestamp(); + String tempPath = pathHelper.getConfig().getTempPath(); + + for (MetaFileEntity mf : list) { + mf.setCreateUser(ue.getId()); + mf.setCreateTime(createTime); + mf.setDepcode(ue.getDepcode()); + mf.setPath(tempPath + File.separator + mf.getPath()); + mf.setMsg(null); + + File f = new File(mf.getPath()); + if (!f.exists()) { + mf.setMsg("鏂囦欢涓㈠け"); + } + + MetaEntity old = metaService.selectByGuid(mf.getGuid(), null); + if (null != old) { + mf.setMsg("宸插叆搴�"); + } + } + } + + /** + * Excel鍏ュ簱 + */ + private String excelLoader(List<MetaFileEntity> list, HttpServletRequest req) { + List<MetaFileEntity> xlsList = getExcelFiles(list); + if (xlsList.isEmpty()) { + return ""; + } + + String guid = null; + try { + MetaFileEntity meta = getExcelMeta(xlsList); + guid = fmeService.excelLoader(meta, req); + } catch (Exception ex) { + log.error(ex.getMessage(), ex); + } + + return guid; + } + + /** + * 鑾峰彇Excel鍏冩暟鎹枃浠� + */ + private List<MetaFileEntity> getExcelFiles(List<MetaFileEntity> list) { + List<MetaFileEntity> xlsList = new ArrayList<>(); + for (MetaFileEntity mf : list) { + if (null != mf.getMsg()) { + continue; + } + if (StaticData.XLS.equals(mf.getExtName()) || StaticData.XLSX.equals(mf.getExtName())) { + xlsList.add(mf); + } + } + + return xlsList; + } + + /** + * 鑾峰彇Excel鐨勫厓鏁版嵁 + */ + private MetaFileEntity getExcelMeta(List<MetaFileEntity> xlsList) { + List<String> pathList = new ArrayList<>(); + List<String> dirList = new ArrayList<>(); + List<String> pidList = new ArrayList<>(); + + String xlsBasePath = getXlsPath(xlsList.get(0).getPath()); + for (int i = 0, c = xlsList.size(); i < c; i++) { + MetaFileEntity mf = xlsList.get(i); + + String path = copyXlsFile(xlsBasePath, i, mf); + if (null != path) { + pathList.add(path); + dirList.add(mf.getDircode()); + pidList.add(mf.getEventid()); + } + } + + MetaFileEntity first = xlsList.get(0); + MetaFileEntity meta = new MetaFileEntity(); + meta.setPath(StringHelper.join(pathList, ",")); + meta.setEpsgCode(first.getEpsgCode()); + meta.setName(StringHelper.join(dirList, ";")); + meta.setDepcode(first.getDepcode()); + meta.setVerid(first.getVerid()); + meta.setCreateUser(first.getCreateUser()); + meta.setEventid(StringHelper.join(pidList, ";")); + + return meta; + } + + /** + * 鑾峰彇Xls鐩綍 + */ + private String getXlsPath(String filePath) { + String tempPath = pathHelper.getConfig().getTempPath() + File.separator; + String subPath = filePath.substring(tempPath.length()); + subPath = tempPath + subPath.substring(0, subPath.indexOf(File.separator)).replace("_zip", "") + "_xls"; + + File f = new File(subPath); + if (!f.exists() || !f.isDirectory()) { + f.mkdirs(); + } + + return subPath; + } + + /** + * 澶嶅埗Xls鏂囦欢 + */ + private String copyXlsFile(String xlsBasePath, int i, MetaFileEntity mf) { + try { + String xlsPath = xlsBasePath + File.separator + i; + + File file = new File(xlsPath); + if (!file.exists() || !file.isDirectory()) { + file.mkdirs(); + } + file = new File(mf.getPath()); + + File newFile = new File(xlsPath + File.separator + FileHelper.getFileName(file.getPath())); + FileUtils.copyFile(file, newFile); + + return newFile.getPath(); + } catch (Exception ex) { + log.error(ex.getMessage(), ex); + return null; + } + } + + /** + * 鍔犺浇鏁版嵁 + */ + private void loadData(List<MetaFileEntity> list) { + for (MetaFileEntity mf : list) { + if (null != mf.getMsg() || StringHelper.isEmpty(mf.getEntity())) { + continue; + } + if (StaticData.SHP.equals(mf.getExtName()) || StaticData.GDB.equals(mf.getExtName())) { + loadSpatialData(mf); + } + } + } + + /** + * 鍔犺浇绌洪棿鏁版嵁 + */ + private void loadSpatialData(MetaFileEntity mf) { + BasicMapper basicMapper = ClassHelper.getBasicMapper(mf.getEntity()); + if (null == basicMapper) { + return; + } + + String tabName = BaseQueryService.getTabName(basicMapper); + String className = ClassHelper.getClassName(basicMapper); + Class clazz = ClassHelper.getEntityClass(className); + if (null == clazz || null == tabName) { + return; + } + + File file = new File(mf.getPath()); + if (!file.exists()) { + mf.setMsg("鏂囦欢涓㈠け"); + return; + } + + MetaEntity old = metaService.selectByGuid(mf.getGuid(), tabName); + if (null != old) { + mf.setMsg("宸插叆搴�"); + return; + } + + List<?> list; + if (StaticData.SHP.equals(mf.getExtName())) { + list = ShpHelper.readData(clazz, mf.getPath()); + } else { + list = GdbHelper.readData(clazz, mf.getPath(), mf.getTab()); + } + if (null == list || list.isEmpty()) { + return; + } + setCreateInfo(list, mf); + + int rows = basicMapper.insertBatch(list); + if (rows > 0) { + mf.setEntity(mf.getTab()); + mf.setTab(tabName); + mf.setRows(rows); + } + } + + /** + * 璁剧疆鍒涘缓淇℃伅 + */ + private <T> void setCreateInfo(List<T> list, MetaFileEntity mf) { + try { + if (!(list.get(0) instanceof BaseEntity)) { + return; + } + + for (T t : list) { + BaseEntity be = (BaseEntity) t; + be.setEventid(StringHelper.getGuid()); + be.setParentid(mf.getEventid()); + be.setDirid(mf.getDircode()); + be.setDepid(mf.getDepcode()); + be.setVerid(mf.getVerid()); + be.setCreateuser(mf.getCreateUser()); + be.setCreatetime(mf.getCreateTime()); + } + } catch (Exception ex) { + log.error(ex.getMessage(), ex); + } + } + + /** + * 澶嶅埗鏂囦欢 + */ + private void copyFiles(List<MetaFileEntity> list) { + List<String> gdbList = new ArrayList<>(); + for (MetaFileEntity mf : list) { + if (null != mf.getMsg()) { + continue; + } + + switch (mf.getExtName()) { + case StaticData.MPT: + copyMultiFile(mf, StaticData.MPT_EXT); + break; + case StaticData.IMG: + copyMultiFile(mf, StaticData.IMG_EXT); + break; + case StaticData.TIF: + copyMultiFile(mf, StaticData.TIF_EXT); + break; + case StaticData.TIFF: + copyMultiFile(mf, StaticData.TIFF_EXT); + break; + case StaticData.SHP: + copyMultiFile(mf, StaticData.SHP_EXT); + break; + case StaticData.GDB: + if (gdbList.contains(mf.getPath())) { + String path = findPathByGuid(list, mf); + if (null != path) { + mf.setPath(path); + continue; + } + } + + gdbList.add(mf.getPath()); + copyFolderFile(mf); + break; + default: + copySingleFile(mf); + break; + } + } + } + + /** + * 澶嶅埗鍗曚釜鏂囦欢 + */ + private int copySingleFile(MetaFileEntity mf) { + File file = new File(mf.getPath()); + if (!file.exists()) { + mf.setMsg("鏂囦欢涓㈠け"); + return -1; + } + + MetaEntity old = metaService.selectByGuid(mf.getGuid(), null); + if (null != old) { + mf.setMsg("宸插叆搴�"); + file.delete(); + return 0; + } + + String uploadPath = pathHelper.getUploadFullPath(); + String targetPath = uploadPath + File.separator + mf.getGuid() + mf.getExtName(); + String subPath = FileHelper.getRelativePath(targetPath); + + File newFile = new File(targetPath); + if (newFile.exists()) { + mf.setPath(subPath); + file.delete(); + return 0; + } + + file.renameTo(newFile); + mf.setPath(subPath); + + return 1; + } + + /** + * 澶嶅埗澶氫釜鏂囦欢 + */ + private void copyMultiFile(MetaFileEntity mf, List<String> extList) { + String path = mf.getPath(); + int status = copySingleFile(mf); + if (status < 1) { + for (int i = 0, c = extList.size(); i < c; i++) { + String subPath = path.replace(mf.getExtName(), extList.get(i)); + File file = new File(subPath); + if (file.exists()) { + file.delete(); + } + } + return; + } + + String uploadPath = pathHelper.getConfig().getUploadPath(); + for (int i = 0, c = extList.size(); i < c; i++) { + String sourcePath = path.replace(mf.getExtName(), extList.get(i)); + File file = new File(sourcePath); + if (!file.exists()) { + continue; + } + + String targetPath = uploadPath + File.separator + mf.getPath().replace(mf.getExtName(), extList.get(i)); + File newFile = new File(targetPath); + if (newFile.exists()) { + continue; + } + + file.renameTo(newFile); + } + } + + /** + * 澶嶅埗鏂囦欢澶规枃浠� + */ + private void copyFolderFile(MetaFileEntity mf) { + File file = new File(mf.getPath()); + if (!file.exists()) { + mf.setMsg("鏂囦欢涓㈠け"); + return; + } + + MetaEntity old = metaService.selectByGuid(mf.getGuid(), null); + if (null != old) { + mf.setMsg("宸插叆搴�"); + FileHelper.deleteFiles(file); + return; + } + + String uploadPath = pathHelper.getUploadFullPath(); + String targetPath = uploadPath + File.separator + mf.getGuid() + mf.getExtName(); + String subPath = FileHelper.getRelativePath(targetPath); + + File newFile = new File(targetPath); + if (newFile.exists() && newFile.isDirectory()) { + mf.setPath(subPath); + FileHelper.deleteFiles(file); + return; + } + + newFile.mkdirs(); + File[] files = file.listFiles(); + if (null == files || files.length == 0) { + return; + } + + for (File f : files) { + String subFile = targetPath + File.separator + FileHelper.getFileName(f.getPath()); + f.renameTo(new File(subFile)); + } + mf.setPath(subPath); + } + + /** + * 鏍规嵁GUID鏌ユ壘璺緞 + */ + private String findPathByGuid(List<MetaFileEntity> list, MetaFileEntity mf) { + for (MetaFileEntity meta : list) { + if (meta.getGuid().equals(mf.getGuid()) && !meta.getPath().equals(mf.getPath())) { + return meta.getPath(); + } + } + + return null; + } + + /** + * 鍙戝竷鏂囦欢 + */ + private void publishFiles(List<MetaFileEntity> list) { + for (MetaFileEntity mf : list) { + if (null != mf.getMsg()) { + continue; + } + switch (mf.getExtName()) { + case StaticData.MPT: + linkFiles(mf, StaticData.MPT_EXT, pathHelper.getConfig().getPublishMptPath()); + break; + case StaticData.D3DML: + linkFiles(mf, null, pathHelper.getConfig().getPublish3dmlPath()); + break; + default: + break; + } + } + } + + /** + * 閾炬帴鏂囦欢 + */ + private void linkFiles(MetaFileEntity mf, List<String> exts, String publishPath) { + String uploadPath = pathHelper.getConfig().getUploadPath(); + String source = uploadPath + File.separator + mf.getPath(); + String target = publishPath + File.separator + mf.getName(); + + File sourceFile = new File(source); + if (!sourceFile.exists() || sourceFile.isDirectory()) { + return; + } + + createFileLink(source, target); + if (null == exts) { + return; + } + + for (String ext : exts) { + File f=new File(source.replace(mf.getExtName(), ext)); + if (f.exists()&&!f.isDirectory()) { + createFileLink(source.replace(mf.getExtName(), ext), target.replace(mf.getExtName(), ext)); + } + } + } + + /** + * 鍒涘缓鏂囦欢閾炬帴 + */ + private void createFileLink(String source, String target) { + String cmd = String.format("cmd /c mklink \"%s\" \"%s\"", target, source); + WebHelper.exec(cmd); + } + + /** + * 鎻掑叆鍏冩暟鎹� + */ + private void insertMetas(List<MetaFileEntity> list) { + for (MetaFileEntity mf : list) { + if (null != mf.getMsg()) { + continue; + } + + MetaEntity me = createMeta(mf); + metaService.insert(me); + + mf.setMsg(me.getId() > 0 ? "鎴愬姛" : "澶辫触"); + } + } + + /** + * 鍒涘缓鍏冩暟鎹� + */ + private MetaEntity createMeta(MetaFileEntity mf) { + MetaEntity me = new MetaEntity(); + me.setEventid(mf.getEventid()); + me.setDircode(mf.getDircode()); + me.setDepcode(mf.getDepcode()); + me.setVerid(mf.getVerid()); + me.setName(mf.getName()); + me.setType(mf.getType()); + me.setGuid(mf.getGuid()); + me.setPath(mf.getPath()); + me.setSizes(mf.getSizes()); + if (mf.getRows() > 0) { + me.setTab(mf.getTab()); + } + if (!StringHelper.isEmpty(mf.getEntity())) { + me.setLayer(mf.getEntity()); + } + me.setRows(mf.getRows()); + me.setCreateUser(mf.getCreateUser()); + me.setCreateTime(mf.getCreateTime()); + + return me; + } } -- Gitblit v1.9.3