From b69e86316f5a8802a8cc2ff552bc798b4c27aba4 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期二, 06 十二月 2022 09:39:59 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/helper/GdbHelper.java |  133 ++++++++++++++++++++++++++++----------------
 1 files changed, 84 insertions(+), 49 deletions(-)

diff --git a/src/main/java/com/lf/server/helper/GdbHelper.java b/src/main/java/com/lf/server/helper/GdbHelper.java
index 48d1d14..33c0e36 100644
--- a/src/main/java/com/lf/server/helper/GdbHelper.java
+++ b/src/main/java/com/lf/server/helper/GdbHelper.java
@@ -9,10 +9,7 @@
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.LocalTime;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * GDB甯姪绫�
@@ -21,11 +18,50 @@
 public class GdbHelper {
     private final static Log log = LogFactory.getLog(GdbHelper.class);
 
-    public static List<String> excludeFields = new ArrayList<String>();
+    private final static String OBJECT ="java.lang.Object";
 
-    static {
-        excludeFields.add("gid");
-        excludeFields.add("shape_leng");
+    private static List<String> excludeFields = new ArrayList<>(Arrays.asList("gid", "objectid", "dirid", "depid", "verid","createtime", "createuser", "updateuser", "updatetime", "shape_leng", "shape_area"));
+
+    /**
+     * 閿�姣佽祫婧�
+     */
+    public static void delete(Layer layer) {
+        try {
+            if (null != layer) {
+                layer.delete();
+            }
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+        }
+    }
+
+    /**
+     * 閿�姣佽祫婧�
+     */
+    public static void delete(DataSource dataSource, Driver driver) {
+        try {
+            if (null != dataSource) {
+                dataSource.delete();
+            }
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+        }
+
+        try {
+            if (null != driver) {
+                driver.delete();
+            }
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+        }
+    }
+
+    /**
+     * 閿�姣佽祫婧�
+     */
+    public static void delete(Layer layer, DataSource dataSource, Driver driver) {
+        delete(layer);
+        delete(dataSource, driver);
     }
 
     /**
@@ -33,13 +69,16 @@
      */
     public static List<String> getTabNames(String filePath) {
         List<String> list = new ArrayList<>();
+
+        Driver driver = null;
+        DataSource dataSource = null;
         try {
-            org.gdal.ogr.Driver driver = ogr.GetDriverByName("OpenFileGDB");
+            driver = ogr.GetDriverByName("OpenFileGDB");
             if (null == driver) {
                 return list;
             }
 
-            DataSource dataSource = driver.Open(filePath, 0);
+            dataSource = driver.Open(filePath, 0);
             if (null == dataSource) {
                 return list;
             }
@@ -50,11 +89,10 @@
 
                 layer.delete();
             }
-
-            dataSource.delete();
-            driver.delete();
         } catch (Exception ex) {
             log.error(ex.getMessage(), ex);
+        } finally {
+            GdbHelper.delete(dataSource, driver);
         }
 
         return list;
@@ -65,20 +103,22 @@
      */
     public static <T> List<T> readData(Class clazz, String filePath, String layerName) {
         List<T> list = new ArrayList<>();
+
+        Driver driver = null;
+        DataSource dataSource = null;
         try {
-            org.gdal.ogr.Driver driver = ogr.GetDriverByName("OpenFileGDB");
+            driver = ogr.GetDriverByName("OpenFileGDB");
             if (null == driver) {
                 return list;
             }
 
-            DataSource dataSource = driver.Open(filePath, 0);
+            dataSource = driver.Open(filePath, 0);
             if (null == dataSource) {
                 return list;
             }
 
             for (int i = 0, count = dataSource.GetLayerCount(); i < count; i++) {
                 Layer layer = dataSource.GetLayer(i);
-
                 if (layer.GetName().equals(layerName)) {
                     GdbHelper.readLayer(clazz, layer, list);
                     break;
@@ -86,11 +126,10 @@
 
                 layer.delete();
             }
-
-            dataSource.delete();
-            driver.delete();
         } catch (Exception ex) {
             log.error(ex.getMessage(), ex);
+        } finally {
+            GdbHelper.delete(dataSource, driver);
         }
 
         return list;
@@ -102,11 +141,10 @@
     public static <T> void readLayer(Class clazz, Layer layer, List<T> list) {
         try {
             Field gField = getGeomField(clazz);
-            Map<Integer, Field> map = getFieldMapper(clazz, layer);
-            if (map.size() == 0) {
-                return;
-            }
-            if (1 > layer.GetFeatureCount()) {
+
+            Map<Integer, Field> map = new HashMap<>(3);
+            getFieldMapper(clazz, layer, map);
+            if (map.isEmpty() || 0 == layer.GetFeatureCount()) {
                 return;
             }
 
@@ -124,9 +162,7 @@
         } catch (Exception ex) {
             log.error(ex.getMessage(), ex);
         } finally {
-            if (null != layer) {
-                layer.delete();
-            }
+            GdbHelper.delete(layer);
         }
     }
 
@@ -135,7 +171,7 @@
      */
     private static Field getGeomField(Class clazz) {
         try {
-            Field gField = clazz.getDeclaredField("geom");
+            Field gField = clazz.getSuperclass().getDeclaredField("geom");
             gField.setAccessible(true);
 
             return gField;
@@ -147,16 +183,14 @@
     /**
      * 鑾峰彇瀛楁鏄犲皠
      */
-    private static <T> Map<Integer, Field> getFieldMapper(Class clazz, Layer layer) {
-        Map<Integer, Field> map = new HashMap<>(3);
-
+    private static <T> void getFieldMapper(Class clazz, Layer layer, Map<Integer, Field> map) {
         try {
             FeatureDefn fd = layer.GetLayerDefn();
             for (int i = 0, count = fd.GetFieldCount(); i < count; i++) {
                 FieldDefn fieldDefn = fd.GetFieldDefn(i);
                 try {
                     String name = fieldDefn.GetName().toLowerCase();
-                    if (excludeFields.contains(name)){
+                    if (excludeFields.contains(name)) {
                         continue;
                     }
 
@@ -168,11 +202,13 @@
                     //
                 }
             }
+
+            if (OBJECT != clazz.getSuperclass().getName()) {
+                getFieldMapper(clazz.getSuperclass(), layer, map);
+            }
         } catch (Exception ex) {
             //
         }
-
-        return map;
     }
 
     /**
@@ -199,21 +235,21 @@
      * 璁剧疆鍊�
      */
     private static <T> void setValue(T t, Feature f, Field field, Integer i) throws Exception {
-        switch (field.getType().toString()) {
-            case "class java.math.BigDecimal":
-            case "class java.lang.Double":
+        switch (field.getType().getName()) {
+            case "java.math.BigDecimal":
+            case "java.lang.Double":
             case "double":
                 field.set(t, f.GetFieldAsDouble(i));
                 break;
-            case "class java.lang.Long":
+            case "java.lang.Long":
             case "long":
                 field.set(t, f.GetFieldAsInteger64(i));
                 break;
-            case "class java.lang.Integer":
+            case "java.lang.Integer":
             case "int":
                 field.set(t, f.GetFieldAsInteger(i));
                 break;
-            case "class java.sql.Timestamp":
+            case "java.sql.Timestamp":
                 field.set(t, getTimestamp(f, i));
                 break;
             default:
@@ -237,10 +273,10 @@
      * wkbLinearRing = 101
      */
     private static <T> void setGeom(T t, Feature f, Field gField) throws Exception {
-        Geometry geometry = f.GetGeometryRef();
-        if (null != geometry) {
-            String wkt = geometry.ExportToWkt();
-            switch (geometry.GetGeometryType()) {
+        String geo = "null";
+        if (null != f.GetGeometryRef()) {
+            String wkt = f.GetGeometryRef().ExportToWkt();
+            switch (f.GetGeometryRef().GetGeometryType()) {
                 case 2:
                     wkt = wkt.replace("LINESTRING (", "MULTILINESTRING ((") + ")";
                     break;
@@ -250,9 +286,10 @@
                 default:
                     break;
             }
-
-            gField.set(t, String.format("ST_GeomFromText('%s')", wkt));
+            geo = String.format("ST_GeomFromText('%s')", wkt);
         }
+
+        gField.set(t, geo);
     }
 
     /**
@@ -278,8 +315,6 @@
                 LocalTime.of(pnHour[0], pnMinute[0], s, ns)
         );
 
-        Timestamp timestamp = Timestamp.valueOf(localDateTime);
-
-        return timestamp;
+        return Timestamp.valueOf(localDateTime);
     }
 }

--
Gitblit v1.9.3