package com.se.simu.service;
|
|
import cn.hutool.core.io.FileUtil;
|
import cn.hutool.json.JSONArray;
|
import cn.hutool.json.JSONObject;
|
import cn.hutool.json.JSONUtil;
|
import com.se.simu.domain.SeDb;
|
import com.se.simu.domain.SeField;
|
import com.se.simu.domain.SeLayer;
|
import com.se.simu.helper.RsaHelper;
|
import com.se.simu.helper.ShpHelper;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.CollectionUtils;
|
import org.springframework.util.StringUtils;
|
import org.springframework.web.client.RestTemplate;
|
|
import javax.annotation.Resource;
|
import java.io.File;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* SDDB服务类
|
*
|
* @author WWW
|
* @date 2024-09-12
|
*/
|
@Slf4j
|
@Service
|
@SuppressWarnings("ALL")
|
public class SedbService {
|
@Value("${sys.path.in}")
|
String inPath;
|
|
@Value("${sedb.host}")
|
String host;
|
|
@Value("${sedb.user}")
|
String user;
|
|
@Value("${sedb.pwd}")
|
String pwd;
|
|
@Value("${sedb.dbName}")
|
String dbName;
|
|
@Value("${sedb.pageSize}")
|
Integer pageSize;
|
|
@Value("#{'${sedb.layerNames}'}")
|
List<String> layerNames;
|
|
@Value("#{'${sedb.shpNames}'}")
|
List<String> shpNames;
|
|
@Value("#{'${sedb.sysFields}'}")
|
List<String> sysFields;
|
|
String password;
|
|
@Resource
|
RestTemplate restTemplate;
|
|
public boolean test() throws Exception {
|
// 469538.6536261877,4416744.922022615,469853.14714664617,4417049.378602433
|
String bbox = "116.64388473935195,39.884315914604464,116.64754729082588,39.887069143903496";
|
|
String token = getToken();
|
|
SeDb db = getSeDb(token);
|
db.setBbox(bbox);
|
|
List<SeLayer> layers = getLayers(token, db.getDbid());
|
queryData(token, db, layers);
|
|
createShps(inPath + "\\20240913", layers);
|
|
return true;
|
}
|
|
public String getToken() throws Exception {
|
Map<String, Object> map = new HashMap<>(2);
|
map.put("userid", user);
|
map.put("password", getPassword());
|
|
JSONObject obj = restTemplate.postForObject(host + "account-service/security/login", map, JSONObject.class);
|
|
JSONObject data = obj.getJSONObject("data");
|
|
return data.getStr("token");
|
}
|
|
public String getPassword() throws Exception {
|
if (StringUtils.isEmpty(password)) {
|
String key = getPublicKey();
|
RsaHelper.setPublicKey(key);
|
password = RsaHelper.encrypt(pwd);
|
}
|
|
return password;
|
}
|
|
public String getPublicKey() {
|
//{"datetime":"2024-09-12 17:24:38","code":200,"data":"MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCtFwJCh2taVTEi05C8qT2oG7Y+rDmJhlO4zicpSeRtiro9LsytePeWI7BXM6sfDU0WeKun1izawcfgGkZgnoJuMBluAOKI1tL0uCrR+DreNLqMVtnXHwoWEIk/hGJedDWaf3q22aGDyEB5h9qCq0JklSShP1Ih4ppap4LmgxdTPQIDAQAB"}
|
JSONObject obj = restTemplate.getForObject(host + "account-service/security/publickey", JSONObject.class);
|
|
return obj.getStr("data");
|
}
|
|
public SeDb getSeDb(String token) {
|
Map<String, Object> map = new HashMap<>(1);
|
map.put("token", token);
|
|
JSONObject obj = restTemplate.postForObject(host + "geo-service/entitydb/list/canview", map, JSONObject.class);
|
JSONArray data = obj.getJSONArray("data");
|
|
List<SeDb> list = JSONUtil.toList(data, SeDb.class);
|
if (CollectionUtils.isEmpty(list)) return null;
|
|
return list.stream().filter(db -> dbName.equals(db.getName())).findFirst().orElse(null);
|
}
|
|
public List<SeLayer> getLayers(String token, String dbid) {
|
String uri = String.format("%sgeo-service/entitydb/map/config?dbid=%s&token=%s", host, dbid, token);
|
JSONObject obj = restTemplate.getForObject(uri, JSONObject.class);
|
|
JSONObject data = obj.getJSONObject("data");
|
JSONArray arr = data.getJSONArray("layers");
|
if (null == arr || arr.size() == 0) return null;
|
|
List<SeLayer> layers = new ArrayList<>();
|
for (int i = 0, c = arr.size(); i < c; i++) {
|
JSONObject jb = arr.getJSONObject(i);
|
String name = jb.getStr("name");
|
if (layerNames.contains(name)) {
|
String id = jb.getStr("id");
|
Integer dataType = jb.getInt("_data_type");
|
List<SeField> fields = JSONUtil.toList(jb.getJSONArray("fields"), SeField.class);
|
fields = fields.stream().filter(f -> !sysFields.contains(f.getName())).collect(Collectors.toList());
|
String shpName = shpNames.get(layerNames.indexOf(name));
|
|
layers.add(new SeLayer(id, name, shpName, dataType, fields));
|
}
|
}
|
|
return layers;
|
}
|
|
public void queryData(String token, SeDb db, List<SeLayer> layers) throws Exception {
|
for (SeLayer layer : layers) {
|
int count = getCount(token, db, layer);
|
if (0 == count) throw new Exception(layer.getName() + "，图层数据为空");
|
|
int pageCount = (count - 1) / pageSize + 1;
|
for (int i = 0; i < pageCount; i++) {
|
JSONArray data = query(token, db, layer, i + 1, pageSize);
|
if (null != data && data.size() > 0) {
|
layer.addData(data);
|
}
|
}
|
}
|
}
|
|
private int getCount(String token, SeDb db, SeLayer layer) {
|
Map<String, Object> map = new HashMap<>(6);
|
map.put("token", token);
|
map.put("dbid", db.getDbid());
|
map.put("bbox", db.getBbox());
|
map.put("layerid", layer.getId());
|
map.put("returnCountOnly", true);
|
map.put("inSR", 4326);
|
|
JSONObject obj = restTemplate.postForObject(host + "geo-service/entitydbdata/layer/query", map, JSONObject.class);
|
if (null == obj || 200 != obj.getInt("code")) return 0;
|
|
return obj.getInt("data");
|
}
|
|
private JSONArray query(String token, SeDb db, SeLayer layer, int start, int count) {
|
Map<String, Object> map = new HashMap<>(9);
|
map.put("token", token);
|
map.put("start", start);
|
map.put("count", count);
|
map.put("dbid", db.getDbid());
|
map.put("bbox", db.getBbox());
|
map.put("containCount", false);
|
map.put("layerid", layer.getId());
|
map.put("querytype", layer.getQueryType());
|
map.put("inSR", 4326);
|
|
JSONObject obj = restTemplate.postForObject(host + "geo-service/entitydbdata/layer/query", map, JSONObject.class);
|
if (null == obj || 200 != obj.getInt("code")) return null;
|
|
JSONObject data = obj.getJSONObject("data");
|
|
return data.getJSONArray("features");
|
}
|
|
public void createShps(String basePath, List<SeLayer> layers) throws Exception {
|
File f = new File(basePath);
|
if (f.exists() && f.isDirectory()) {
|
FileUtil.del(f);
|
}
|
f.mkdirs();
|
|
for (SeLayer layer : layers) {
|
String path = String.format("%s\\%s.shp", basePath, layer.getShpName());
|
if (!ShpHelper.createShp(path, layer)) {
|
throw new Exception(layer.getName() + "，创建ShapeFile文件失败！");
|
}
|
}
|
}
|
}
|
