From 3417cf014a65765e02696c1d121ce58b2b4a8aed Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期二, 08 四月 2025 15:55:36 +0800
Subject: [PATCH] 修改pom.xml
---
src/main/java/com/se/simu/service/GedbService.java | 181 +++++++++++++++++++++++++++++++++++++++++----
1 files changed, 165 insertions(+), 16 deletions(-)
diff --git a/src/main/java/com/se/simu/service/GedbService.java b/src/main/java/com/se/simu/service/GedbService.java
index a286a96..9a82084 100644
--- a/src/main/java/com/se/simu/service/GedbService.java
+++ b/src/main/java/com/se/simu/service/GedbService.java
@@ -10,6 +10,7 @@
import com.se.simu.domain.dto.GeFile;
import com.se.simu.domain.dto.GeLayer;
import com.se.simu.domain.po.DataPo;
+import com.se.simu.helper.CaffeineHelper;
import com.se.simu.helper.RsaHelper;
import com.se.simu.helper.ShpHelper;
import com.se.simu.helper.StringHelper;
@@ -18,7 +19,8 @@
import org.gdal.gdal.WarpOptions;
import org.gdal.gdal.gdal;
import org.gdal.gdalconst.gdalconst;
-import org.springframework.beans.factory.annotation.Value;
+import org.gdal.ogr.Geometry;
+import org.gdal.osr.SpatialReference;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import org.springframework.web.client.RestTemplate;
@@ -33,12 +35,6 @@
import java.util.*;
import java.util.stream.Collectors;
-/**
- * GEDB鏈嶅姟绫�
- *
- * @author WWW
- * @date 2024-09-12
- */
@Slf4j
@Service
@SuppressWarnings("ALL")
@@ -50,6 +46,10 @@
@Resource
RestTemplate restTemplate;
+
+ private final static String DB_KEY = "gedb_db";
+
+ private final static String TOKEN_KEY = "gedb_token";
public boolean test(DataPo data) throws Exception {
createPath(config.getInPath() + File.separator + data.getInPath());
@@ -73,11 +73,26 @@
}
public String getToken() throws Exception {
+ Object obj = CaffeineHelper.get(TOKEN_KEY);
+ if (obj instanceof String) {
+ return obj.toString();
+ }
+
+ String token = getTokenByServer();
+ if (null == token) throw new Exception("鑾峰彇GEDB浠ょ墝澶辫触");
+
+ CaffeineHelper.put(TOKEN_KEY, token);
+
+ return token;
+ }
+
+ private String getTokenByServer() throws Exception {
Map<String, Object> map = new HashMap<>(2);
map.put("userid", config.getUser());
map.put("password", getPassword());
JSONObject obj = restTemplate.postForObject(config.getHost() + "account-service/security/login", map, JSONObject.class);
+ log.info(obj.toString());
JSONObject data = obj.getJSONObject("data");
@@ -94,7 +109,7 @@
return password;
}
- private String getPublicKey() {
+ public String getPublicKey() {
//{"datetime":"2024-09-12 17:24:38","code":200,"data":"MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCtFwJCh2taVTEi05C8qT2oG7Y+rDmJhlO4zicpSeRtiro9LsytePeWI7BXM6sfDU0WeKun1izawcfgGkZgnoJuMBluAOKI1tL0uCrR+DreNLqMVtnXHwoWEIk/hGJedDWaf3q22aGDyEB5h9qCq0JklSShP1Ih4ppap4LmgxdTPQIDAQAB"}
JSONObject obj = restTemplate.getForObject(config.getHost() + "account-service/security/publickey", JSONObject.class);
@@ -102,24 +117,32 @@
}
public GeDb connectGedb(String token, DataPo data) {
- GeDb db = getSeDb(token);
+ GeDb db = getGeDb(token);
db.setBbox(data.getBbox());
data.setEpsg(db.getEpsg());
return db;
}
- private GeDb getSeDb(String token) {
+ public GeDb getGeDb(String token) {
+ Object obj = CaffeineHelper.get(DB_KEY);
+ if (obj instanceof GeDb) {
+ return (GeDb) obj;
+ }
+
Map<String, Object> map = new HashMap<>(1);
map.put("token", token);
- JSONObject obj = restTemplate.postForObject(config.getHost() + "geo-service/entitydb/list/canview", map, JSONObject.class);
- JSONArray data = obj.getJSONArray("data");
+ JSONObject jsonObject = restTemplate.postForObject(config.getHost() + "geo-service/entitydb/list/canview", map, JSONObject.class);
+ JSONArray data = jsonObject.getJSONArray("data");
List<GeDb> list = JSONUtil.toList(data, GeDb.class);
if (CollectionUtils.isEmpty(list)) return null;
- return list.stream().filter(db -> null != db.getName() && db.getName().contains(config.getDbName())).findFirst().orElse(null);
+ GeDb gedb = list.stream().filter(db -> null != db.getName() && db.getName().contains(config.getDbName())).findFirst().orElse(null);
+ if (null != gedb) CaffeineHelper.put(DB_KEY, gedb);
+
+ return gedb;
}
public List<GeLayer> getLayers(String token, GeDb db) {
@@ -148,7 +171,7 @@
return layers;
}
- private String getQueryType(JSONObject jb) {
+ public static String getQueryType(JSONObject jb) {
String qt = jb.getStr("pointlod");
if (!StringHelper.isEmpty(qt)) {
return "point";
@@ -175,6 +198,15 @@
}
}
}
+ }
+
+ public boolean queryBboxCount(String token, GeDb db, List<GeLayer> layers) {
+ for (GeLayer layer : layers) {
+ int count = getCount(token, db, layer);
+ if (0 == count) return false;
+ }
+
+ return true;
}
private int getCount(String token, GeDb db, GeLayer layer) {
@@ -217,18 +249,135 @@
List<GeLayer> layers = getLayers(token, db);
queryData(token, db, layers);
+ checkData(data, db, layers);
createShps(basePath, layers);
+ createZoneShp(basePath, data, db.getSpatialReference());
+ if (data.getPid() > 0) {
+ createFloodShp(basePath, data, db.getSpatialReference());
+ }
+ }
+
+ private void checkData(DataPo data, GeDb db, List<GeLayer> layers) {
+ GeLayer point = getLayerByName(layers, config.getLayerNames().get(0));
+ GeLayer line = getLayerByName(layers, config.getLayerNames().get(1));
+ GeLayer build = getLayerByName(layers, config.getLayerNames().get(2));
+
+ Geometry extent = ShpHelper.createPolygon(db.getSpatialReference(), data.getMinx(), data.getMiny(), data.getMaxx(), data.getMaxy());
+ checkSpatialRange(extent, point);
+ checkSpatialRange(extent, build);
+
+ List<String> bsm = getValues(point, "bsm");
+ List<String> bsm2 = new ArrayList<>(bsm);
+ List<String> qdbsm = getValues(line, "qdbsm");
+ List<String> qdbsm2 = new ArrayList<>(qdbsm);
+ List<String> zdbsm = getValues(line, "zdbsm");
+ List<String> zdbsm2 = new ArrayList<>(zdbsm);
+
+ qdbsm.removeAll(bsm2);
+ zdbsm.removeAll(bsm2);
+ removeValues(line, "qdbsm", qdbsm);
+ removeValues(line, "zdbsm", zdbsm);
+
+ qdbsm = getValues(line, "qdbsm");
+ zdbsm = getValues(line, "zdbsm");
+ bsm.removeAll(qdbsm);
+ bsm.removeAll(zdbsm);
+ removeValues(point, "bsm", bsm);
+
+ GeLayer juncLayer = new GeLayer(point, filterLayerData(point.getData()));
+ juncLayer.setName("闆嗘按鐐�");
+ juncLayer.setShpName(config.getJunctionName());
+ layers.add(juncLayer);
+ }
+
+ private void checkSpatialRange(Geometry extent, GeLayer geLayer) {
+ int i = 0;
+ while (i < geLayer.getData().size()) {
+ JSONObject geom = geLayer.getData().getJSONObject(i).getJSONObject("geometry");
+ Geometry g = ShpHelper.createGeometry(geLayer, geom);
+ g.AssignSpatialReference(extent.GetSpatialReference());
+
+ if (!extent.Intersects(g)) {
+ geLayer.getData().remove(i);
+ continue;
+ }
+ i++;
+ }
+ }
+
+ private GeLayer getLayerByName(List<GeLayer> layers, String name) {
+ return layers.stream().filter(a -> a.getName().equals(name)).findFirst().orElse(null);
+ }
+
+ private List<String> getValues(GeLayer layer, String field) {
+ JSONArray data = layer.getData();
+ List<String> list = new ArrayList<>();
+
+ int i = 0;
+ while (i < data.size()) {
+ JSONObject obj = data.getJSONObject(i).getJSONObject("properties");
+ if (StringHelper.isEmpty(obj.getStr(field))) {
+ data.remove(i);
+ continue;
+ }
+ list.add(obj.getStr(field));
+ i++;
+ }
+
+ return list;
+ }
+
+ private void removeValues(GeLayer layer, String field, List<String> values) {
+ if (CollectionUtils.isEmpty(values)) return;
+
+ int i = 0;
+ JSONArray data = layer.getData();
+ while (i < data.size()) {
+ JSONObject obj = data.getJSONObject(i).getJSONObject("properties");
+ if (values.contains(obj.getStr(field))) {
+ data.remove(i);
+ continue;
+ }
+ i++;
+ }
}
private void createShps(String basePath, List<GeLayer> layers) throws Exception {
for (GeLayer layer : layers) {
- String path = String.format("%s\\%s.shp", basePath, layer.getShpName());
- if (!ShpHelper.createShp(path, layer)) {
+ String path = basePath + File.separator + layer.getShpName();
+ if (layer.getData().isEmpty() || !ShpHelper.createShp(path, layer)) {
throw new Exception(layer.getName() + "锛屽垱寤篠hapeFile鏂囦欢澶辫触锛�");
}
}
}
+ private JSONArray filterLayerData(JSONArray data) {
+ JSONArray arr = new JSONArray();
+ String[] strs = config.getJunctionFilter().split("=");
+ for (int i = 0, c = data.size(); i < c; i++) {
+ JSONObject obj = data.getJSONObject(i).getJSONObject("properties");
+ if (strs[1].equals(obj.getStr(strs[0]))) {
+ arr.put(data.getJSONObject(i));
+ }
+ }
+
+ return arr;
+ }
+
+ private void createZoneShp(String basePath, DataPo data, SpatialReference sr) {
+ String filePath = basePath + File.separator + config.getZoneName();
+ ShpHelper.createShp(filePath, null, sr, data.getMinx(), data.getMiny(), data.getMaxx(), data.getMaxy());
+ }
+
+ private void createFloodShp(String basePath, DataPo data, SpatialReference sr) {
+ String filePath = basePath + File.separator + config.getBarrierName();
+ Map<String, Object> map = new HashMap<>();
+ map.put("height", data.getFloodHeight());
+ map.put("type", data.getFloodType());
+
+ ShpHelper.createShp(filePath, map, sr, data.getFloodMinx(), data.getFloodMiny(), data.getFloodMaxx(), data.getFloodMaxy());
+ }
+
public void copeDem(String token, DataPo data) throws Exception {
GeDb fileDb = getFileDb(token);
String fileId = getFileId(token, fileDb.getDbid());
--
Gitblit v1.9.3