From 2d2d9138fbe5c9c43f070be59a2d10f18fbf4c3b Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 23 十一月 2022 09:05:06 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/helper/MdbHelper.java |   36 +++++++++++++++++++++++++++++++++---
 1 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/lf/server/helper/MdbHelper.java b/src/main/java/com/lf/server/helper/MdbHelper.java
index 1b515bf..71bb0fe 100644
--- a/src/main/java/com/lf/server/helper/MdbHelper.java
+++ b/src/main/java/com/lf/server/helper/MdbHelper.java
@@ -23,16 +23,24 @@
         List<String> list = new ArrayList<>();
         try {
             org.gdal.ogr.Driver driver = ogr.GetDriverByName("PGeo");
-            if (driver == null) {
+            if (null == driver) {
                 return list;
             }
 
             DataSource 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);
-
                 list.add(layer.GetName());
+
+                layer.delete();
             }
+
+            dataSource.delete();
+            driver.delete();
         } catch (Exception ex) {
             log.error(ex.getMessage(), ex);
         }
@@ -43,10 +51,32 @@
     /**
      * 璇诲彇鏁版嵁
      */
-    public static <T> List<T> readData(Class clazz, String filePath) {
+    public static <T> List<T> readData(Class clazz, String filePath, String layerName) {
         List<T> list = new ArrayList<>();
         try {
+            org.gdal.ogr.Driver driver = ogr.GetDriverByName("PGeo");
+            if (null == driver) {
+                return list;
+            }
 
+            DataSource 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;
+                }
+
+                layer.delete();
+            }
+
+            dataSource.delete();
+            driver.delete();
         } catch (Exception ex) {
             log.error(ex.getMessage(), ex);
         }

--
Gitblit v1.9.3