springboot中使用gdal将表中的空间数据转shapefile文件

springboot中使用gdal将表中的空间数据转shapefile文件

代码:

java 复制代码
// 样本导出-将样本表导出为shapefile,复制样本shp文件到临时目录下  sampleDir是文件夹path
    public void setYbShapeFile(Yb yb, File sampleDir) {
        // 创建 前时项 和 后时项 文件夹
        File ybDir = new File(sampleDir, "样本");
        ybDir.mkdirs();
        String tableName = yb.getLabelLayerName();
        boolean isYbLabelName = tableName != null && !tableName.equals("");
        String shapefileName = new File(ybDir, tableName + ".shp").getAbsolutePath();

        if (isYbLabelName) {
            // 查询数据
            SqlRowSet resultSet = jdbcTemplate.queryForRowSet("SELECT objectid, ST_AsText(shape) AS shape FROM " + tableName);

            // 使用GDAL库将查询到的数据转换为Shapefile
            ogr.RegisterAll();
            Driver driver = ogr.GetDriverByName("ESRI Shapefile");
            if (driver == null) {
                throw new RuntimeException("GDAL 驱动加载失败: ESRI Shapefile");
            }

            DataSource ds = null;
            Layer layer = null;
            try {
                ds = driver.CreateDataSource(shapefileName);
                if (ds == null) {
                    throw new RuntimeException("无法创建 Shapefile 数据源: " + shapefileName);
                }

                // 定义坐标参考系统(以 WGS 84 为例)
                SpatialReference srs = new SpatialReference();
                srs.ImportFromEPSG(4326);

                // 创建图层并指定参考系统
                layer = ds.CreateLayer(tableName, srs, ogr.wkbUnknown);
                if (layer == null) {
                    throw new RuntimeException("无法创建图层: " + tableName);
                }

                // 定义字段,根据实际数据库字段调整
                layer.CreateField(new FieldDefn("objectid", ogr.OFTInteger));

                // 遍历结果集,添加要素到图层
                while (resultSet.next()) {
                    Feature feature = new Feature(layer.GetLayerDefn());
                    feature.SetField("objectid", resultSet.getInt("objectid"));

                    // 获取几何数据并设置
                    String wktGeometry = resultSet.getString("shape");
                    if (wktGeometry != null && !wktGeometry.isEmpty()) {
                        Geometry geom = ogr.CreateGeometryFromWkt(wktGeometry);
                        if (geom == null) {
                            throw new RuntimeException("无效的WKT几何: " + wktGeometry);
                        }
                        feature.SetGeometry(geom);
                    }

                    // 添加Feature到图层
                    if (layer.CreateFeature(feature) != 0) {
                        throw new RuntimeException("无法创建要素");
                    }

                    feature.delete(); // 释放资源
                }

                // 同步数据到磁盘
                layer.SyncToDisk();
            } catch (Exception e) {
                e.printStackTrace();
                throw new RuntimeException("导出Shapefile时发生错误: " + e.getMessage(), e);
            } finally {
                if (ds != null) {
                    ds.delete(); // 确保数据源关闭
                }

                if (layer != null) {
                    layer.delete(); // 显式释放图层资源
                }
            }
        }
    }

示例表:

转为shp文件后:

相关推荐
Java中文社群7 小时前
Hutool被卖半年多了,现状是逆袭还是沉寂?
java·后端
程序员蜗牛8 小时前
9个Spring Boot参数验证高阶技巧,第8,9个代码量直接减半!
后端
yeyong8 小时前
咨询kimi关于设计日志告警功能,还是有启发的
后端
库森学长8 小时前
2025年,你不能错过Spring AI,那个汲取了LangChain灵感的家伙!
后端·openai·ai编程
爱吃苹果的日记本8 小时前
开学第一课
java
Java水解8 小时前
Spring Boot 启动流程详解
spring boot·后端
学历真的很重要9 小时前
Claude Code Windows 原生版安装指南
人工智能·windows·后端·语言模型·面试·go
渣哥9 小时前
Java 集合框架详解:常见集合类及分类方式
java
转转技术团队9 小时前
让AI成为你的编程助手:如何高效使用Cursor
后端·cursor
shellvon9 小时前
你怎么被识别的?从TLS到Canvas的设备追踪术
后端·算法