From b1e97e3b11b6f95b9e39cd475dca027ba690598c Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 29 五月 2023 10:38:32 +0800
Subject: [PATCH] 1
---
src/main/java/com/lf/server/service/data/UploadService.java | 386 ++++++++++++++++++++++++++++++++++++++++++++++--------
1 files changed, 326 insertions(+), 60 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 4b6806d..2dccb6e 100644
--- a/src/main/java/com/lf/server/service/data/UploadService.java
+++ b/src/main/java/com/lf/server/service/data/UploadService.java
@@ -2,16 +2,15 @@
import com.lf.server.entity.all.BaseEntity;
import com.lf.server.entity.all.StaticData;
-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.data.MetaFileEntity;
+import com.lf.server.entity.data.*;
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;
@@ -23,6 +22,7 @@
/**
* 鏁版嵁涓婁紶鏈嶅姟绫�
+ *
* @author WWW
*/
@Service
@@ -32,6 +32,15 @@
@Autowired
FmeService fmeService;
+
+ @Autowired
+ MetaService metaService;
+
+ @Autowired
+ DepService depService;
+
+ @Autowired
+ DirService dirService;
@Override
public List<CoordEntity> selectCoords(String zoning) {
@@ -58,8 +67,13 @@
*/
public void insertFiles(UserEntity ue, List<MetaFileEntity> list, HttpServletRequest req) {
checkMetaFiles(ue, list);
- dataLoader(list);
- excelLoader(list, req);
+ List<MetaFileEntity> xlsList = getExcelFiles(list);
+ loadData(list);
+ copyFiles(list);
+ insertMetas(list);
+ if (xlsList.size() > 0) {
+ String guid = excelLoader(xlsList, req);
+ }
}
/**
@@ -71,25 +85,91 @@
for (MetaFileEntity mf : list) {
mf.setCreateUser(ue.getId());
- mf.setCreatetime(createTime);
- mf.setDepid(ue.getDepid());
+ 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.setRows(-1);
- mf.setMsg("鏂囦欢涓嶅瓨鍦�");
+ mf.setMsg("鏂囦欢涓㈠け");
+ }
+
+ MetaEntity old = metaService.selectByGuid(mf.getGuid(), null);
+ if (null != old) {
+ mf.setMsg("宸插瓨鍦�");
}
}
}
/**
- * 鍔犺浇鏁版嵁
+ * 鑾峰彇Excel鍏冩暟鎹枃浠�
*/
- private void dataLoader(List<MetaFileEntity> list) {
- loadData(list);
- copyFiles(list);
- insertMetas(list);
+ private List<MetaFileEntity> getExcelFiles(List<MetaFileEntity> list) {
+ List<MetaFileEntity> xlsList = new ArrayList<>();
+ for (MetaFileEntity mf : list) {
+ if (null == mf.getMsg() && isExcel(mf)) {
+ xlsList.add(mf);
+ }
+ }
+ if (xlsList.isEmpty()) {
+ return xlsList;
+ }
+
+ String xlsBasePath = getXlsPath(xlsList.get(0).getPath());
+ for (int i = 0, c = xlsList.size(); i < c; i++) {
+ MetaFileEntity mf = xlsList.get(i);
+ String xlsPath = copyXlsFile(xlsBasePath, i, mf);
+ mf.setXlsPath(xlsPath);
+ }
+
+ return xlsList;
+ }
+
+ /**
+ * 鏄�/鍚︿负Excel
+ */
+ private boolean isExcel(MetaFileEntity mf) {
+ return StaticData.XLS.equals(mf.getExtName()) || StaticData.XLSX.equals(mf.getExtName());
+ }
+
+ /**
+ * 鑾峰彇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;
+ }
}
/**
@@ -97,6 +177,9 @@
*/
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);
}
@@ -106,7 +189,7 @@
/**
* 鍔犺浇绌洪棿鏁版嵁
*/
- private void loadSpatialData(MetaFileEntity mf) {
+ public void loadSpatialData(MetaFileEntity mf) {
BasicMapper basicMapper = ClassHelper.getBasicMapper(mf.getEntity());
if (null == basicMapper) {
return;
@@ -119,11 +202,23 @@
return;
}
- List<?> list = null;
+ 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 {
- GdbHelper.readData(clazz, mf.getPath(), mf.getTab());
+ list = GdbHelper.readData(clazz, mf.getPath(), mf.getTab());
}
if (null == list || list.isEmpty()) {
return;
@@ -132,6 +227,7 @@
int rows = basicMapper.insertBatch(list);
if (rows > 0) {
+ mf.setEntity(mf.getTab());
mf.setTab(tabName);
mf.setRows(rows);
}
@@ -150,11 +246,11 @@
BaseEntity be = (BaseEntity) t;
be.setEventid(StringHelper.getGuid());
be.setParentid(mf.getEventid());
- be.setDirid(mf.getDirid());
- be.setDepid(mf.getDepid());
+ be.setDirid(mf.getDircode());
+ be.setDepid(mf.getDepcode());
be.setVerid(mf.getVerid());
be.setCreateuser(mf.getCreateUser());
- be.setCreatetime(mf.getCreatetime());
+ be.setCreatetime(mf.getCreateTime());
}
} catch (Exception ex) {
log.error(ex.getMessage(), ex);
@@ -165,10 +261,18 @@
* 澶嶅埗鏂囦欢
*/
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.JPG:
+ copyMultiFile(mf, StaticData.JPG_EXT);
break;
case StaticData.IMG:
copyMultiFile(mf, StaticData.IMG_EXT);
@@ -183,6 +287,15 @@
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:
@@ -195,92 +308,245 @@
/**
* 澶嶅埗鍗曚釜鏂囦欢
*/
- private void copySingleFile(MetaFileEntity mf) {
+ 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 createFileLink(String source, String target) {
+ String cmd = String.format("cmd /c mklink \"%s\" \"%s\"", target, source);
+ WebHelper.exec(cmd);
}
/**
* 鎻掑叆鍏冩暟鎹�
- *
- * @param list
*/
private void insertMetas(List<MetaFileEntity> list) {
+ int metaId = insertParentMeta(list);
for (MetaFileEntity mf : list) {
- //
+ if (null != mf.getMsg()) {
+ continue;
+ }
+
+ MetaEntity me = createMeta(mf, metaId);
+ metaService.insert(me);
+
+ mf.setMsg(me.getId() > 0 ? "鎴愬姛" : "澶辫触");
}
+ }
+
+ /**
+ * 鍒涘缓鍏冩暟鎹�
+ */
+ private MetaEntity createMeta(MetaFileEntity mf, int metaId) {
+ MetaEntity me = new MetaEntity();
+ me.setMetaid(metaId);
+ 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;
+ }
+
+ /**
+ * 鎻掑叆鐖跺厓鏁版嵁
+ */
+ private int insertParentMeta(List<MetaFileEntity> list) {
+ for (MetaFileEntity mf : list) {
+ if (null != mf.getMsg() || !mf.getIsMeta()) {
+ continue;
+ }
+
+ MetaEntity me = createMeta(mf, 0);
+ me.setIsmeta((short) 1);
+ metaService.insert(me);
+ mf.setMsg(me.getId() > 0 ? "鎴愬姛" : "澶辫触");
+
+ return me.getId();
+ }
+
+ return 0;
}
/**
* Excel鍏ュ簱
*/
- private String excelLoader(List<MetaFileEntity> list, HttpServletRequest req) {
- List<MetaFileEntity> xlsList = getExcelFiles(list);
- if (xlsList.isEmpty()) {
- return "";
- }
-
- String guid = null;
+ private String excelLoader(List<MetaFileEntity> xlsList, HttpServletRequest req) {
try {
- MetaFileEntity meta = getExcelMeta(xlsList);
- guid = fmeService.excelLoader(meta, req);
+ MetaFileEntity xlsMeta = getExcelMeta(xlsList);
+ if (null != xlsMeta) {
+ return fmeService.excelLoader(xlsMeta, 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) {
- boolean isXls = StaticData.XLS.equals(mf.getExtName()) || StaticData.XLSX.equals(mf.getExtName());
- if (mf.getRows() > -1 && isXls) {
- xlsList.add(mf);
- }
- }
-
- return xlsList;
+ return null;
}
/**
* 鑾峰彇Excel鐨勫厓鏁版嵁
*/
private MetaFileEntity getExcelMeta(List<MetaFileEntity> xlsList) {
+ if (null == xlsList || xlsList.isEmpty()) {
+ return null;
+ }
+
List<String> pathList = new ArrayList<>();
List<String> dirList = new ArrayList<>();
List<String> pidList = new ArrayList<>();
- for (MetaFileEntity mf : xlsList) {
- pathList.add(mf.getPath());
- dirList.add(mf.getDirid() + "");
- pidList.add(mf.getEventid());
+ for (int i = 0, c = xlsList.size(); i < c; i++) {
+ MetaFileEntity mf = xlsList.get(i);
+ if ("鎴愬姛".equals(mf.getMsg()) && null != mf.getXlsPath()) {
+ pathList.add(mf.getXlsPath());
+ dirList.add(mf.getDircode());
+ pidList.add(mf.getEventid());
+ }
}
+ MetaFileEntity first = xlsList.get(0);
MetaFileEntity meta = new MetaFileEntity();
meta.setPath(StringHelper.join(pathList, ","));
- meta.setEpsgCode(xlsList.get(0).getEpsgCode());
- meta.setName(StringHelper.join(dirList, ","));
- meta.setDepid(xlsList.get(0).getDepid());
- meta.setVerid(xlsList.get(0).getVerid());
- meta.setCreateUser(xlsList.get(0).getCreateUser());
- meta.setEventid(StringHelper.join(pidList, ","));
+ 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;
}
--
Gitblit v1.9.3