com.biz.common.jts.ShapefileToGeoJSON Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of biz-all Show documentation
BizX 是一个灵活而高效的业务开发框架, 其中也有很多为业务开发所需要的工具类的提供。
The newest version!
package com.biz.common.jts;

import com.biz.common.concurrent.ExecutorsUtils;
import lombok.extern.slf4j.Slf4j;
import org.geotools.data.shapefile.ShapefileDataStore;
import org.geotools.data.simple.SimpleFeatureCollection;
import org.geotools.geojson.feature.FeatureJSON;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.nio.charset.Charset;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * Shapefile 转换为 GeoJSON 的工具类。
 * 该类提供了递归遍历目录下的 .shp 文件，并将其转换为 .geojson 文件的功能。
 *
 * 示例代码：
 * {@code
 * File directory = new File("/path/to/directory");
 * ShapefileToGeoJSON.traverseDirectory(directory);
 * }
 *
 * 注意：该类依赖于 GeoTools 库和自定义的 ExecutorsUtils 工具类。
 *
 * @author francis
 * @version 1.0.1
 * @since 1.0.1
 **/
@Slf4j
public class ShapefileToGeoJSON {

    private static final String CHARSET = "GBK";
    private static final String SHP = ".shp";
    private static final String GEO_JSON = ".geojson";
    private static final ExecutorService EXECUTOR_SERVICE = ExecutorsUtils.buildScheduledExecutorService();

    /**
     * 递归遍历目录下所有的.shp文件，并将其转换为 .geojson 文件。
     *
     * @param directory 需要遍历的文件夹
     * @throws IOException 如果遍历过程中发生 I/O 错误
     */
    public static void traverseDirectory(File directory) throws IOException {
        // 获取文件夹下的所有文件和文件夹
        List files;
        try (Stream paths = Files.list(directory.toPath())) {
            files = paths.map(Path::toFile).collect(Collectors.toList());
        } catch (IOException e) {
            log.warn("Error listing files", e);
            return;
        }

        files.stream()
                .filter(File::isDirectory)
                .forEach(file -> EXECUTOR_SERVICE.submit(() -> {
                    try {
                        traverseDirectory(file);
                    } catch (IOException e) {
                        log.error("Error traversing directory", e);
                    }
                }));

        files.stream()
                .filter(file -> file.getName().toLowerCase().endsWith(SHP))
                .forEach(file -> EXECUTOR_SERVICE.submit(() -> convertShpToGeoJSON(file)));

        EXECUTOR_SERVICE.shutdown();

        try {
            // 等待所有任务完成或超时
            boolean terminated = EXECUTOR_SERVICE.awaitTermination(1, TimeUnit.HOURS);
            if (!terminated) {
                log.warn("ExecutorService did not terminate within the specified time.");
            }
        } catch (InterruptedException e) {
            log.error("The conversion process is interrupted", e);
            Thread.currentThread().interrupt(); // 恢复中断状态
        }
    }

    /**
     * 将 .shp 文件转换为 .geojson 文件。
     *
     * @param shpFile 需要转换的 .shp 文件
     */
    public static void convertShpToGeoJSON(File shpFile) {
        String shpFilePath = shpFile.getAbsolutePath();
        String geoJsonFilePath = shpFilePath.substring(0, shpFilePath.lastIndexOf(".")) + GEO_JSON;

        try {
            // 获取 SHP 文件的数据存储
            ShapefileDataStore dataStore = new ShapefileDataStore(shpFile.toURI().toURL());
            // 处理中文乱码
            dataStore.setCharset(Charset.forName(CHARSET));

            // 获取特征集合
            SimpleFeatureCollection featureCollection = dataStore.getFeatureSource().getFeatures();

            // 创建 FeatureJSON 对象
            FeatureJSON featureJSON = new FeatureJSON();
            try (FileWriter writer = new FileWriter(geoJsonFilePath)) {
                // 将特征集合写入 GeoJSON 文件
                featureJSON.writeFeatureCollection(featureCollection, writer);
            }

            // 释放数据存储资源
            dataStore.dispose();
        } catch (IOException e) {
            log.error("Failed to convert Shapefile to GeoJSON", e);
        }
    }
}