123

13693261870

2024-09-13 0db692fe0e8d9dcbf5c1ca17373893e9ef09875a

 src/main/java/com/se/simu/service/SedbService.java

@@ -1,5 +1,6 @@
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;
@@ -7,7 +8,7 @@
import com.se.simu.domain.SeField;
import com.se.simu.domain.SeLayer;
import com.se.simu.helper.RsaHelper;
import com.se.simu.helper.StringHelper;
import com.se.simu.helper.ShpHelper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
@@ -16,6 +17,7 @@
import org.springframework.web.client.RestTemplate;

import javax.annotation.Resource;
import java.io.File;
import java.util.*;
import java.util.stream.Collectors;

@@ -29,6 +31,9 @@
@Service
@SuppressWarnings("ALL")
public class SedbService {
    @Value("${sys.path.in}")
    String inPath;

    @Value("${sedb.host}")
    String host;

@@ -41,8 +46,14 @@
    @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;
@@ -52,7 +63,8 @@
    @Resource
    RestTemplate restTemplate;

    public String test() throws Exception {
    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();
@@ -61,8 +73,11 @@
        db.setBbox(bbox);

        List<SeLayer> layers = getLayers(token, db.getDbid());
        queryData(token, db, layers);

        return db.getDbid();
        createShps(inPath + "\\20240913", layers);

        return true;
    }

    public String getToken() throws Exception {
@@ -124,38 +139,77 @@
                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, dataType, fields));
                layers.add(new SeLayer(id, name, shpName, dataType, fields));
            }
        }

        return layers;
    }

    public JSONArray query(String token, String dbid, String layerId, String bbox) {
        int start = 1, count = 1000, total = 0;
    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() + "，图层数据为空");

        Map<String, Object> map = new HashMap<>(8);
            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", dbid);
        map.put("layerid", layerId);
        map.put("containCount", true);
        map.put("bbox", bbox);
        map.put("querytype", "entity");
        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");
        total = data.getInt("data");
        if (0 == total) return null;

        JSONArray arr = new JSONArray();
        JSONArray items = data.getJSONArray("items");
        arr.addAll(items);
        return data.getJSONArray("features");
    }

        return arr;
    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文件失败！");
            }
        }
    }
}