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 |  485 ++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 457 insertions(+), 28 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 8743472..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,20 +1,23 @@
 package com.lf.server.service.data;
 
+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.DepEntity;
 import com.lf.server.entity.sys.UserEntity;
-import com.lf.server.helper.StringHelper;
+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 org.apache.commons.collections4.queue.PredicatedQueue;
+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;
 
@@ -29,6 +32,15 @@
 
     @Autowired
     FmeService fmeService;
+
+    @Autowired
+    MetaService metaService;
+
+    @Autowired
+    DepService depService;
+
+    @Autowired
+    DirService dirService;
 
     @Override
     public List<CoordEntity> selectCoords(String zoning) {
@@ -55,23 +67,35 @@
      */
     public void insertFiles(UserEntity ue, List<MetaFileEntity> list, HttpServletRequest req) {
         checkMetaFiles(ue, list);
-        loadExcel(list, req);
+        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.setDepid(ue.getDepid());
+            mf.setCreateTime(createTime);
+            mf.setDepcode(ue.getDepcode());
+            mf.setPath(tempPath + File.separator + mf.getPath());
+            mf.setMsg(null);
 
-            File f = new File(tempPath + File.separator + mf.getPath());
+            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("宸插叆搴�");
             }
         }
     }
@@ -79,14 +103,21 @@
     /**
      * Excel鍏ュ簱
      */
-    private void loadExcel(List<MetaFileEntity> list, HttpServletRequest req) {
+    private String excelLoader(List<MetaFileEntity> list, HttpServletRequest req) {
         List<MetaFileEntity> xlsList = getExcelFiles(list);
         if (xlsList.isEmpty()) {
-            return;
+            return "";
         }
 
-        MetaFileEntity meta = getExcelMeta(xlsList);
-        String guid = fmeService.excelLoader(meta, req);
+        String guid = null;
+        try {
+            MetaFileEntity meta = getExcelMeta(xlsList);
+            guid = fmeService.excelLoader(meta, req);
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+        }
+
+        return guid;
     }
 
     /**
@@ -95,8 +126,10 @@
     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) {
+            if (null != mf.getMsg()) {
+                continue;
+            }
+            if (StaticData.XLS.equals(mf.getExtName()) || StaticData.XLSX.equals(mf.getExtName())) {
                 xlsList.add(mf);
             }
         }
@@ -112,22 +145,418 @@
         List<String> dirList = new ArrayList<>();
         List<String> pidList = new ArrayList<>();
 
-        String tempPath = pathHelper.getConfig().getTempPath();
-        for (MetaFileEntity mf : xlsList) {
-            pathList.add(tempPath + File.separator + mf.getPath());
-            dirList.add(mf.getDirid() + "");
-            pidList.add(mf.getEventid());
+        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(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;
     }
+
+    /**
+     * 鑾峰彇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