今天和大家分享shp数据的读取入库。

程序代码使用IDEA编写，使用mybatis、gdal，数据库是postgresql，安装postGIS扩展。废话不多说，从controller层开始上代码。

@PostMapping("add")public ReturnModel add(){    // shp文件地址。处理过程中没有做坐标转换，这里的shp数据的坐标系采用WGS84，对应的EPSG代码为4326    String shpPath = "E:\\data_cd\\build\\build4326.shp";    // 返回给前端的模型    ReturnModel returnModel = new ReturnModel();    // 处理程序。这里仅为示例，采用同步的形式    // 一般数据处理是一个耗时操作，建议处理为异步的形式    shpService.shpHandler(returnModel, shpPath);    returnModel.setCode(1);    returnModel.setMessage("添加成功");    return returnModel;}

service实现类

@Overridepublic void shpHandler(ReturnModel returnModel, String path) {    ogr.RegisterAll();    // 支持中文路径    gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "YES");    // 属性表字段支持中文    gdal.SetConfigOption("SHAPE_ENCODING", "CP936");    // 打开shp文件是一个datasource对象    Driver shpDriver = ogr.GetDriverByName("ESRI Shapefile");    String info = "shp文件处理中";    if (shpDriver == null) {        info = "ESRI Shapefile 驱动不可用";        log.error(info);        returnModel.setCode(-1);        returnModel.setMessage(info);        return;    }    DataSource ds = shpDriver.Open(path);    if (ds == null) {        info = "文件打开失败";        log.error(info);        returnModel.setCode(-1);        returnModel.setMessage(info);        return;    }    Layer layer = ds.GetLayer(0);    if (layer == null) {        info = "打开图层失败";        log.error(info);        returnModel.setCode(-1);        returnModel.setMessage(info);        return;    }    // 要素数量    long count = layer.GetFeatureCount();    if (count == 0) {        info = "该文件内无数据";        log.error(info);        returnModel.setCode(-1);        returnModel.setMessage(info);        return;    }    // 获取图形和属性信息。构建入库的list    List<FeatureModel> featureList = getFeatureList(layer);    // 批量入库    shpMapper.insertFeatureList(featureList);}

读取数据的具体实现

private List<FeatureModel> getFeatureList(Layer layer) {    // 图层id    String setid = UUID.randomUUID().toString();    long count = layer.GetFeatureCount();    String name = layer.GetName();    // 坐标系    SpatialReference spatialRef = layer.GetSpatialRef();    List<FeatureModel> featureModelList = new ArrayList<>();    for (int i = 0; i < count; i++) {        FeatureModel featureModel = new FeatureModel();        // 设置图层uuid        featureModel.setSetid(setid);        // 设置要素uuid        featureModel.setUuid(UUID.randomUUID().toString());        // 设置坐标系        featureModel.setCoordinateSystem(spatialRef.toString());        // 图形feature        Feature feature = layer.GetFeature(i);        Geometry geometry = feature.GetGeometryRef();        // WKT形式的图形字符串        String geometryWkt = geometry.ExportToWkt();        featureModel.setGeometryWKT(geometryWkt);        // 属性字段        FeatureDefn featureDefn = layer.GetLayerDefn();        int fieldCount = feature.GetFieldCount();        Map<String, Object> fieldMap = new HashMap<>();        for (int j = 0; j < fieldCount; j++) {            FieldDefn fieldDefn = featureDefn.GetFieldDefn(j);            int type = fieldDefn.GetFieldType();            // 字段类型            String fieldTypeName = fieldDefn.GetFieldTypeName(type);            // 属性名称            String fieldName = fieldDefn.GetName();            // 属性值            String value = feature.GetFieldAsString(j);            fieldMap.put(fieldName, value);        }        JSONObject fieldJson = new JSONObject(fieldMap);        // 属性信息存为json格式        featureModel.setFieldInfo(fieldJson.toString());        featureModelList.add(featureModel);    }    return featureModelList;}

mapper层

@Mapper@Repositorypublic interface ShpMapper {    int insertFeatureList(@Param("featureList") List<FeatureModel> featureModelList);}

mapper.xml

<insert id="insertFeatureList">    insert into feature_build (uuid, setid, geometry, info, "system") values    <foreach collection="featureList" item="item" index="index" separator=",">        (#{item.uuid}, #{item.setid}, ST_GeomFromText(#{item.geometryWKT}, 4326), #{item.fieldInfo},        #{item.coordinateSystem})    </foreach></insert>

本文涉及到的两个model

@Datapublic class ReturnModel {    // -1处理失败；0 处理中；1处理成功    private int code;    // 返回数据    private Object data;    // 返回的提示信息    private String message;}@Datapublic class FeatureModel {    // 图形的唯一uuid    private String uuid;    // 标识每一个shp文件    private String setid;    // 坐标系    private String coordinateSystem;    // 属性信息    private String fieldInfo;    // 图形WKT串    private String geometryWKT;}

入库结果

点击图中的眼睛按钮，可以查看图形。如果这里的图形结果不对，请查看数据的坐标系，和入库时（mapper.xml）指定的坐标系是否一致。

最后：

1、创建spring boot的web项目：IDEA构建spring boot项目

2、mybaitis连接到pg库：mybaitis连接数据库