From a4ae15934058cc2d2d51631bfd48fb17e636a868 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 18 九月 2023 16:35:03 +0800
Subject: [PATCH] 添加插入KML数据功能

---
 src/main/java/com/lf/server/entity/data/MetaFileEntity.java              |   10 +++
 src/main/java/com/lf/server/entity/all/StaticData.java                   |    4 +
 src/main/java/com/lf/server/service/data/UploadService.java              |  138 +++++++++++++++++++++++++++++++++++++++++++++
 src/main/java/com/lf/server/controller/data/upload/UploadController.java |   25 ++++++++
 4 files changed, 175 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/data/upload/UploadController.java b/src/main/java/com/lf/server/controller/data/upload/UploadController.java
index cba61ec..b9ab916 100644
--- a/src/main/java/com/lf/server/controller/data/upload/UploadController.java
+++ b/src/main/java/com/lf/server/controller/data/upload/UploadController.java
@@ -219,6 +219,31 @@
     }
 
     @SysLog()
+    @ApiOperation(value = "鎻掑叆KML鏂囦欢")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "list", value = "鍏冩暟鎹枃浠堕泦鍚�", dataType = "MetaFileEntity", paramType = "body")
+    })
+    @ResponseBody
+    @PostMapping(value = "/insertKml")
+    public ResponseMsg<Object> insertKml(@RequestBody List<MetaFileEntity> list, HttpServletRequest req){
+        try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (null == ue) {
+                return fail("鐢ㄦ埛鏈櫥褰�", null);
+            }
+            if (null == list || list.isEmpty()) {
+                return fail("鍏冩暟鎹枃浠堕泦鍚堜负绌�", null);
+            }
+
+            uploadService.insertKml(ue, list);
+
+            return success("鎴愬姛", list);
+        } catch (Exception ex) {
+            return fail(ex, null);
+        }
+    }
+
+    @SysLog()
     @ApiOperation(value = "鍒犻櫎鍏冩暟鎹�")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "ids", value = "ID鏁扮粍", dataType = "Integer", paramType = "query", allowMultiple = true, example = "1")
diff --git a/src/main/java/com/lf/server/entity/all/StaticData.java b/src/main/java/com/lf/server/entity/all/StaticData.java
index 87a7127..02e7cfc 100644
--- a/src/main/java/com/lf/server/entity/all/StaticData.java
+++ b/src/main/java/com/lf/server/entity/all/StaticData.java
@@ -94,6 +94,8 @@
 
     public final static String MDB = ".mdb";
 
+    public final static String KML = ".kml";
+
     public final static String SHP = ".shp";
 
     public final static String NGDB = "gdb";
@@ -214,7 +216,7 @@
     /**
      * 鎵�鏈夋枃浠舵墿灞曞悕
      */
-    public final static List<String> ALL_EXTENSION = new ArrayList<>(Arrays.asList(".txt", ".xml", ".pdf", ".xls", ".xlsx", ".doc", ".docx", ".ppt", ".pptx", ".shp", ".gdb", ".mdb", ".dwg", ".las", ".laz", ".cpt", ".mpt", ".ei.mpt", ".fly", ".efb", ".g3d", ".fbx", ".obj", ".3dm", ".3dml", ".osgb", ".rvt", ".ifc", ".jpg", ".jp2", ".png", ".img", ".tif", ".tiff", ".dem", ".bmp", ".gif", ".rmvb", ".rm", ".mp3", ".mp4", ".avi", ".wma", ".wmv", ".7z", ".rar", ".zip", ".lz"));
+    public final static List<String> ALL_EXTENSION = new ArrayList<>(Arrays.asList(".txt", ".xml", ".pdf", ".xls", ".xlsx", ".doc", ".docx", ".ppt", ".pptx", ".shp", ".gdb", ".mdb", ".dwg", ".las", ".laz", ".cpt", ".mpt", ".ei.mpt", ".fly", ".efb", ".g3d", ".fbx", ".obj", ".3dm", ".3dml", ".osgb", ".rvt", ".ifc", ".jpg", ".jp2", ".png", ".img", ".tif", ".tiff", ".dem", ".bmp", ".gif", ".rmvb", ".rm", ".mp3", ".mp4", ".avi", ".wma", ".wmv", ".7z", ".rar", ".zip", ".lz", ".kml"));
 
     /**
      * 鎻掑叆鎺掗櫎瀛楁
diff --git a/src/main/java/com/lf/server/entity/data/MetaFileEntity.java b/src/main/java/com/lf/server/entity/data/MetaFileEntity.java
index 4423c9b..f85ae60 100644
--- a/src/main/java/com/lf/server/entity/data/MetaFileEntity.java
+++ b/src/main/java/com/lf/server/entity/data/MetaFileEntity.java
@@ -50,6 +50,8 @@
 
     private String xlsPath;
 
+    private String medium;
+
     public String getEventid() {
         return eventid;
     }
@@ -209,4 +211,12 @@
     public void setXlsPath(String xlsPath) {
         this.xlsPath = xlsPath;
     }
+
+    public String getMedium() {
+        return medium;
+    }
+
+    public void setMedium(String medium) {
+        this.medium = medium;
+    }
 }
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 bd0c0af..0b0faf2 100644
--- a/src/main/java/com/lf/server/service/data/UploadService.java
+++ b/src/main/java/com/lf/server/service/data/UploadService.java
@@ -17,6 +17,10 @@
 
 import javax.servlet.http.HttpServletRequest;
 import java.io.File;
+import java.lang.reflect.Field;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Paths;
 import java.sql.Timestamp;
 import java.util.ArrayList;
 import java.util.List;
@@ -496,7 +500,7 @@
             MetaEntity me = createMeta(mf, metaId);
             if (StaticData.NGDB.equals(me.getType())) {
                 if (guids.contains(me.getGuid())) {
-                    me.setIsmeta((short)-1);
+                    me.setIsmeta((short) -1);
                 } else {
                     guids.add(me.getGuid());
                 }
@@ -605,4 +609,136 @@
 
         return meta;
     }
+
+    /**
+     * 鎻掑叆KML鏂囦欢
+     */
+    public void insertKml(UserEntity ue, List<MetaFileEntity> list) {
+        checkMetaFiles(ue, list);
+        loadKml(list);
+        copyFiles(list);
+        insertMetas(list);
+    }
+
+    /**
+     * 鍔犺浇Kml
+     */
+    private void loadKml(List<MetaFileEntity> list) {
+        for (MetaFileEntity mf : list) {
+            if (null != mf.getMsg() || StringHelper.isEmpty(mf.getEntity()) || !StaticData.KML.equals(mf.getExtName())) {
+                continue;
+            }
+
+            String geom = readKml(mf.getPath());
+            if (StringHelper.isEmpty(geom)) {
+                continue;
+            }
+
+            loadKmlData(mf, geom);
+        }
+    }
+
+    /**
+     * 璇诲彇KML
+     */
+    private String readKml(String path) {
+        try {
+            List<String> list = Files.readAllLines(Paths.get(path), StandardCharsets.UTF_8);
+            for (String str : list) {
+                if (!str.contains("<coordinates>")) {
+                    continue;
+                }
+
+                return getWktFromCoordinates(str);
+            }
+
+            return null;
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+            return null;
+        }
+    }
+
+    /**
+     * 鑾峰彇KWT
+     */
+    private String getWktFromCoordinates(String coords) {
+        String[] strs = coords.replace("<coordinates>", "").replace("</coordinates>", "").split(" ");
+        if (strs.length == 0) {
+            return null;
+        }
+
+        // MULTILINESTRING((108.99843373100003 34.736292580000054,108.99818970700005 34.73764945000005,108.99831780400007 34.738880547000065,108.99812184300004 34.739564029000064))
+        StringBuilder sb = new StringBuilder();
+        sb.append("MULTILINESTRING((");
+        for (String str : strs) {
+            if (StringHelper.isEmpty(str)) {
+                continue;
+            }
+
+            String[] vals = str.split(StaticData.COMMA);
+            sb.append(vals[0]).append(" ").append(vals[1]).append(StaticData.COMMA);
+        }
+        sb.deleteCharAt(sb.length() - 1);
+        sb.append("))");
+
+        return sb.toString();
+    }
+
+    /**
+     * 鍔犺浇Kml鏁版嵁
+     */
+    private void loadKmlData(MetaFileEntity mf, String geom) {
+        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;
+        }
+
+        String name = FileHelper.getFileName(mf.getPath());
+        List<?> list = createKmlEntity(clazz, geom, name);
+        if (null == list || list.isEmpty()) {
+            return;
+        }
+        mf.setRecords(list.size());
+        setCreateInfo(list, mf);
+
+        int rows = batchInserts(basicMapper, list);
+        if (rows > 0) {
+            mf.setEntity(mf.getTab());
+            mf.setTab(tabName);
+            mf.setRows(rows);
+        }
+    }
+
+    /**
+     * 鍒涘缓KML瀹炰綋绫�
+     */
+    private <T> List<T> createKmlEntity(Class clazz, String geom, String name) {
+        try {
+            T t = (T) clazz.newInstance();
+
+            Field gField = clazz.getSuperclass().getDeclaredField("geom");
+            gField.setAccessible(true);
+            gField.set(t, geom);
+
+            Field pField = clazz.getSuperclass().getDeclaredField("pipename");
+            pField.setAccessible(true);
+            pField.set(t, name);
+
+            List<T> list = new ArrayList<>();
+            list.add(t);
+
+            return list;
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+            return null;
+        }
+    }
 }

--
Gitblit v1.9.3