GeoServer系列-通过mongodb发布geojson数据

前言

geoserver支持的数据存储种类多样，客户的空间数据种类更杂，比如dwg,excel,kml,mysql等等，针对多样的空间文件，一般系统会统一解析保存空间属性，很容易就能转换成geojson，所以大部分空间数据文件的发布，都可以转为geojson的发布，业务系统保存属性信息的同时空间信息也能顺便发布出去，遇到重新发布时就很方便，不需要一直保存用户上传的文件，省去许多文件IO操作的麻烦

1,pom必要的依赖

    <dependency>
        <groupId>org.springframework.data</groupId>
        <artifactId>spring-data-mongodb</artifactId>
        <version>${spring-data-mongodb.version}</version>
    </dependency>
    <dependency>
        <groupId>it.geosolutions</groupId>
        <artifactId>geoserver-manager</artifactId>
        <version>${geoserver.version}</version>
    </dependency>
1
2
3
4
5
6
7
8
9
10

2,单元测试demo

import com.cqdh.group.its.db.util.GSMongodbDatastoreEncoder;
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoClients;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import it.geosolutions.geoserver.rest.GeoServerRESTManager;
import it.geosolutions.geoserver.rest.GeoServerRESTPublisher;
import it.geosolutions.geoserver.rest.GeoServerRESTReader;
import it.geosolutions.geoserver.rest.decoder.RESTDataStore;
import it.geosolutions.geoserver.rest.decoder.RESTLayer;
import it.geosolutions.geoserver.rest.encoder.GSLayerEncoder;
import it.geosolutions.geoserver.rest.encoder.feature.GSFeatureTypeEncoder;
import org.bson.Document;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.json.JSONTokener;

import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.List;

/**
 * @Description 发布geojson文件到geoserver
 * @Date 2023/3/2 16:08
 * @Author 余乐
 **/
public class GeoJsonToGeoServer {
	
    public static void main(String[] args) throws IOException, JSONException {
        String jsonFilePath = "C:\\Users\\CDLX\\Desktop\\geojson\\TEST004_SHP_SHUIYU.geojson";
        String url = "http://127.0.0.1:8089/geoserver";
        String username = "admin";
        String passwd = "geoserver";
        String ws = "yule";
        String storeName = "151mongodb";
        String mongodbUrl = "mongodb://192.168.1.1:27017/its-table";
        String db = "its-table";
        String collectname = "TEST004_GEOJSON_SHUIYU";
        GeoJsonToGeoServer toGeoServer = new GeoJsonToGeoServer();
        //1保存geojson空间数据到mongodb
        toGeoServer.saveToMongodb(jsonFilePath, mongodbUrl, db, collectname);
        //2发布mongodb空间表到geoserver
        toGeoServer.publicMongodb(url, username, passwd, collectname, mongodbUrl, ws, storeName);
    }

    /**
     * 保存json数据到mongodb
     *
     * @param filePath   geojson全路径
     * @param mongodbUrl mongodb地址
     * @param db         数据库
     * @param collect    表名
     * @return
     * @throws IOException
     * @throws JSONException
     */
    public void saveToMongodb(String filePath, String mongodbUrl, String db, String collect) throws IOException, JSONException {
        // 1. 读取 GeoJSON 文件
        BufferedReader reader = new BufferedReader(new FileReader(filePath));
        StringBuilder stringBuilder = new StringBuilder();
        String line = null;
        while ((line = reader.readLine()) != null) {
            stringBuilder.append(line);
        }
        reader.close();
        String geoJsonString = stringBuilder.toString();

        // 2. 解析 GeoJSON 数据
        JSONTokener tokener = new JSONTokener(geoJsonString);
        JSONObject geoJson = new JSONObject(tokener);

        // 3. 连接 MongoDB
        MongoClient mongoClient = MongoClients.create(mongodbUrl);
        MongoDatabase mongoDatabase = mongoClient.getDatabase(db);
        MongoCollection<Document> mongoCollection = mongoDatabase.getCollection(collect);
        //先清空表再写入
        mongoCollection.drop();
        //创建索引，否则geoserver找不到该表，导致发布失败
        mongoCollection.createIndex(new Document("geometry", "2dsphere"));

        // 4. 将解析后的 GeoJSON 数据插入 MongoDB
        JSONArray features = geoJson.getJSONArray("features");
        for (int i = 0; i < features.length(); i++) {
            JSONObject feature = features.getJSONObject(i);
            Document document = Document.parse(feature.toString());
            mongoCollection.insertOne(document);
        }

        // 5. 关闭 MongoDB 连接
        mongoClient.close();
    }


    /**
     * 发布mongodb表到geoserver
     *
     * @param url        geoserverUrl
     * @param username   geoserver用户名
     * @param passwd     geoserver密码
     * @param tale_name  空间表名
     * @param mongodbUrl mongodb地址
     * @param ws         geoserver工作空间
     * @param store_name geoserver存储仓库
     * @throws MalformedURLException
     */
    public void publicMongodb(String url, String username, String passwd, String tale_name, String mongodbUrl, String ws, String store_name) throws MalformedURLException {
        String srs = "EPSG:4490";
        //判断工作区（workspace）是否存在，不存在则创建
        URL restURL = new URL(url);
        //获取管理对象
        GeoServerRESTManager manager = new GeoServerRESTManager(restURL, username, passwd);
        //获取发布对象
        GeoServerRESTPublisher publisher = manager.getPublisher();
        //获取所有的工作空间名称
        List<String> workspaces = manager.getReader().getWorkspaceNames();
        //判断工作空间是否存在
        if (!workspaces.contains(ws)) {
            //创建一个新的存储空间
            boolean createws = publisher.createWorkspace(ws);
            System.out.println("create ws : " + createws);
        } else {
            System.out.println("workspace已经存在了,ws :" + ws);
        }

        //判断数据存储（datastore）是否已经存在，不存在则创建
        RESTDataStore restStore = manager.getReader().getDatastore(ws, store_name);
        if (restStore == null) {
            //创建momgodb数据存储
            GSMongodbDatastoreEncoder store = new GSMongodbDatastoreEncoder(store_name);
            store.setSchema(mongodbUrl);
            store.setData_store(mongodbUrl);
            boolean createStore = manager.getStoreManager().create(ws, store);
            System.out.println("create store  " + store_name + ":" + createStore);
        } else {
            System.out.println("数据存储已经存在了,store:" + store_name);
        }

        //判断图层是否已经存在，不存在则创建并发布
        GeoServerRESTReader reader = new GeoServerRESTReader(url, username, passwd);
        RESTLayer layer = reader.getLayer(ws, tale_name);
        if (layer == null) {
            GSFeatureTypeEncoder pds = new GSFeatureTypeEncoder();
            pds.setTitle(tale_name);
            pds.setName(tale_name);
            //设置坐标系，预览有图片位置不对都是这个参数引起的
            pds.setSRS(srs);
            GSLayerEncoder layerEncoder = new GSLayerEncoder();
            boolean publish = manager.getPublisher().publishDBLayer(ws, store_name, pds, layerEncoder);
            System.out.println("publish : " + publish);
        } else {
            System.out.println("表已经发布过了,table:" + tale_name);
        }
    }

}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160

3,执行结果

workspace已经存在了,ws :yule
数据存储已经存在了,store:151mongodb
publish : true
1
2
3

检查数据库

4,预览图层