1

13693261870

2024-09-23 9981cd49e3fd18691fb6cbe472c93576af58e8df

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

@@ -8,6 +8,7 @@
import com.se.simu.domain.dto.GeField;
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.RsaHelper;
import com.se.simu.helper.ShpHelper;
import com.se.simu.helper.StringHelper;
@@ -82,8 +83,8 @@
    @Resource
    RestTemplate restTemplate;

    public boolean test(String bbox, String taskName) throws Exception {
        String token = getToken();
    public boolean test(DataPo data) throws Exception {
        /*String token = getToken();
        GeDb db = getSeDb(token);
        db.setBbox(bbox);

@@ -100,13 +101,20 @@

        String filePath = inPath + File.separator + fileDb.getDbid();
        downloadFiles(token, filePath, files, fileDb.getDbid(), fileId);
        clipDemFile(filePath, files, basePath, bbox, 4548);
        clipDemFile(filePath, files, basePath, bbox, 4548);*/

        String token = getToken();
        GeDb db = connectGedb(token, data);

        copeVectors(token, data, db);

        copeDem(token, data);

        return true;
    }

    private void createPath(String basePath) {
        File f = new File(basePath);
    private void createPath(String path) {
        File f = new File(path);
        if (f.exists() && f.isDirectory()) {
            FileUtil.del(f);
        }
@@ -125,7 +133,7 @@
        return data.getStr("token");
    }

    public String getPassword() throws Exception {
    private String getPassword() throws Exception {
        if (StringHelper.isEmpty(password)) {
            String key = getPublicKey();
            RsaHelper.setPublicKey(key);
@@ -135,14 +143,22 @@
        return password;
    }

    public String getPublicKey() {
    private 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 GeDb getSeDb(String token) {
    public GeDb connectGedb(String token, DataPo data) {
        GeDb db = getSeDb(token);
        db.setBbox(data.getBbox());
        data.setEpsg(db.getEpsg());

        return db;
    }

    private GeDb getSeDb(String token) {
        Map<String, Object> map = new HashMap<>(1);
        map.put("token", token);

@@ -245,7 +261,15 @@
        return data.getJSONArray("features");
    }

    public void createShps(String basePath, List<GeLayer> layers) throws Exception {
    public void copeVectors(String token, DataPo data, GeDb db) throws Exception {
        String basePath = inPath + File.separator + data.getInPath();

        List<GeLayer> layers = getLayers(token, db);
        queryData(token, db, layers);
        createShps(basePath, layers);
    }

    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)) {
@@ -254,7 +278,17 @@
        }
    }

    public GeDb getFileDb(String token) {
    public void copeDem(String token, DataPo data) throws Exception {
        GeDb fileDb = getFileDb(token);
        String fileId = getFileId(token, fileDb.getDbid());
        List<GeFile> files = getFileNames(token, fileDb.getDbid(), fileId);

        String filePath = inPath + File.separator + fileDb.getDbid();
        downloadFiles(token, filePath, files, fileDb.getDbid(), fileId);
        clipDemFile(filePath, files, data);
    }

    private GeDb getFileDb(String token) {
        String uri = String.format("%sfile-service/docdb/query/canview?token=%s", host, token);
        JSONObject obj = restTemplate.getForObject(uri, JSONObject.class);
        JSONArray data = obj.getJSONArray("data");
@@ -265,7 +299,7 @@
        return list.stream().filter(db -> null != db.getName() && db.getName().contains(dbName)).findFirst().orElse(null);
    }

    public String getFileId(String token, String dbid) {
    private String getFileId(String token, String dbid) {
        String uri = String.format("%sfile-service/doc/catagory/file/query?token=%s&dbid=%s&catagory=%s&count=%d&start=%d&like=",
                host, token, dbid, "image", 9999, 1);

@@ -282,7 +316,7 @@
        return null;
    }

    public List<GeFile> getFileNames(String token, String dbid, String fileId) {
    private List<GeFile> getFileNames(String token, String dbid, String fileId) {
        String uri = String.format("%sfile-service/doc/cluster/struct/list?token=%s&dbid=%s&cluster_fileid=%s&onlychild=true&folder_stairs=",
                host, token, dbid, fileId);

@@ -292,7 +326,7 @@
        return JSONUtil.toList(data, GeFile.class);
    }

    public void downloadFiles(String token, String path, List<GeFile> files, String dbid, String fileId) throws IOException {
    private void downloadFiles(String token, String path, List<GeFile> files, String dbid, String fileId) throws IOException {
        File f = new File(path);
        if (!f.exists() || !f.isDirectory()) {
            f.mkdirs();
@@ -333,40 +367,38 @@
        is.close();
    }

    private void clipDemFile(String filePath, List<GeFile> files, String basePath, String bbox, int epsg) throws Exception {
        String target = basePath + File.separator + demFile;
    private void clipDemFile(String filePath, List<GeFile> files, DataPo data) throws Exception {
        String target = inPath + File.separator + data.getInPath() + File.separator + demFile;
        for (GeFile file : files) {
            if (file.getName().toLowerCase().endsWith(demType)) {
                String source = filePath + File.separator + file.getName();
                clipDem(source, target, bbox, epsg);
                clipDem(source, target, data);
                break;
            }
        }
    }

    private void clipDem(String source, String target, String bbox, int epsg) throws Exception {
    private void clipDem(String source, String target, DataPo data) throws Exception {
        Dataset ds = null;
        try {
            ds = gdal.Open(source, gdalconst.GA_ReadOnly);
            if (null == ds || ds.getRasterCount() < 1 || null == ds.GetSpatialRef()) throw new Exception("DEM数据无效");

            // String bbox = "116.64388473935195,39.884315914604464,116.64754729082588,39.887069143903496";
            String[] bb = bbox.split(",");

            Vector<String> vector = new Vector<>();
            //vector.add("-s_srs");
            //vector.add("EPSG:" + 4326);
            vector.add("-t_srs");
            vector.add("EPSG:" + epsg);
            vector.add("EPSG:" + data.getEpsg());
            vector.add("-r");
            vector.add("bilinear");
            vector.add("-of");
            vector.add("GTiff");
            vector.add("-te");
            vector.add(bb[0]);
            vector.add(bb[1]);
            vector.add(bb[2]);
            vector.add(bb[3]);
            vector.add(data.getMinx().toString());
            vector.add(data.getMiny().toString());
            vector.add(data.getMaxx().toString());
            vector.add(data.getMaxy().toString());
            vector.add("-te_srs");
            vector.add("EPSG:" + 4326);
            WarpOptions warpOptions = new WarpOptions(vector);