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