完成基础功能开发

13693261870

2024-09-14 d1a6b45132c7df365812237281e91f41f00fe90e

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

@@ -1,18 +1,31 @@
package com.se.simu.service;

import cn.hutool.core.io.FileUtil;
import cn.hutool.json.JSONArray;
import cn.hutool.json.JSONObject;
import com.se.simu.domain.SedbLayer;
import cn.hutool.json.JSONUtil;
import com.se.simu.domain.SeDb;
import com.se.simu.domain.SeField;
import com.se.simu.domain.SeFile;
import com.se.simu.domain.SeLayer;
import com.se.simu.helper.RsaHelper;
import com.se.simu.helper.ShpHelper;
import com.se.simu.helper.StringHelper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import org.springframework.util.CollectionUtils;
import org.springframework.web.client.RestTemplate;

import javax.annotation.Resource;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.net.URLConnection;
import java.util.*;
import java.util.stream.Collectors;

/**
 * SDDB服务类
@@ -24,6 +37,9 @@
@Service
@SuppressWarnings("ALL")
public class SedbService {
    @Value("${sys.path.in}")
    String inPath;

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

@@ -33,22 +49,57 @@
    @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;

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

    String password;

    @Resource
    RestTemplate restTemplate;

    public String test() throws Exception {
    public boolean test(String bbox, String taskName) throws Exception {
        String token = getToken();
        SeDb db = getSeDb(token);
        db.setBbox(bbox);

        String dbid = getDbId(token);
        List<SeLayer> layers = getLayers(token, db);
        queryData(token, db, layers);

        List<SedbLayer> layers = getLayers(token, dbid);
        String basePath = inPath + File.separator + taskName;
        createPath(basePath);
        createShps(basePath, layers);

        return dbid;
        SeDb fileDb = getFileDb(token);
        String fileId = getFileId(token, fileDb.getDbid());
        List<SeFile> files = getFileNames(token, fileDb.getDbid(), fileId);

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

        return true;
    }

    private void createPath(String basePath) {
        File f = new File(basePath);
        if (f.exists() && f.isDirectory()) {
            FileUtil.del(f);
        }
        f.mkdirs();
    }

    public String getToken() throws Exception {
@@ -64,7 +115,7 @@
    }

    public String getPassword() throws Exception {
        if (StringUtils.isEmpty(password)) {
        if (StringHelper.isEmpty(password)) {
            String key = getPublicKey();
            RsaHelper.setPublicKey(key);
            password = RsaHelper.encrypt(pwd);
@@ -80,74 +131,194 @@
        return obj.getStr("data");
    }

    public String getDbId(String token) {
    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");

        JSONArray arr = obj.getJSONArray("data");
        if (null == arr || arr.size() == 0) {
            return null;
        }
        List<SeDb> list = JSONUtil.toList(data, SeDb.class);
        if (CollectionUtils.isEmpty(list)) return null;

        JSONObject dbObj = arr.getJSONObject(0);

        return dbObj.getStr("dbid");
        return list.stream().filter(db -> null != db.getName() && db.getName().contains(dbName)).findFirst().orElse(null);
    }

    public List<SedbLayer> getLayers(String token, String dbid) {
        String uri = String.format("%sgeo-service/entitydb/map/config?dbid=%s&token=%s", host, dbid, token);
    public List<SeLayer> getLayers(String token, SeDb db) {
        String uri = String.format("%sgeo-service/entitydb/map/config?dbid=%s&token=%s", host, db.getDbid(), 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;
        }
        if (null == arr || arr.size() == 0) return null;

        List<SedbLayer> layers = new ArrayList<>();
        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");
                JSONArray fields = jb.getJSONArray("fields");
                String queryType = getQueryType(jb);
                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 SedbLayer(id, name, dataType, fields));
                layers.add(new SeLayer(id, name, queryType, fields, shpName, db));
            }
        }

        return layers;
    }

    public List<SedbLayer> getLayers2(String token, String dbid) {
        Calendar calendar = Calendar.getInstance();
        calendar.setTime(new Date());
        calendar.add(Calendar.DAY_OF_MONTH, -1);

        String date = StringHelper.YMD_FORMAT.format(calendar.getTime());
        String uri = String.format("%sgeo-service/statis/layer/data/info?dbid=%s&token=%s&caldate=%s", host, dbid, token, date);
        JSONObject obj = restTemplate.getForObject(uri, JSONObject.class);

        JSONArray arr = obj.getJSONArray("data");
        if (null == arr || arr.size() == 0) {
            return null;
    private String getQueryType(JSONObject jb) {
        String qt = jb.getStr("pointlod");
        if (!StringHelper.isEmpty(qt)) {
            return "point";
        }

        List<SedbLayer> layers = new ArrayList<>();
        for (int i = 0, c = arr.size(); i < c; i++) {
            JSONObject jb = arr.getJSONObject(i);
        qt = jb.getStr("polylinelod");
        if (!StringHelper.isEmpty(qt)) {
            return "polyline";
        }

            String name = jb.getStr("name");
            if (layerNames.contains(name)) {
                String id = jb.getStr("layerid");
                layers.add(new SedbLayer(id, name));
        return "polygon";
    }

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

    public SeDb 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");

        List<SeDb> list = JSONUtil.toList(data, SeDb.class);
        if (CollectionUtils.isEmpty(list)) return null;

        return list.stream().filter(db -> null != db.getName() && db.getName().contains(dbName)).findFirst().orElse(null);
    }

    public 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);

        JSONObject obj = restTemplate.getForObject(uri, JSONObject.class);
        JSONArray items = obj.getJSONObject("data").getJSONArray("items");

        for (int i = 0, c = items.size(); i < c; i++) {
            JSONObject jb = items.getJSONObject(i);
            if (demName.equals(jb.getStr("filename"))) {
                return jb.getStr("fileid");
            }
        }

        return layers;
        return null;
    }

    public List<SeFile> 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);

        JSONObject obj = restTemplate.getForObject(uri, JSONObject.class);
        JSONArray data = obj.getJSONArray("data");

        return JSONUtil.toList(data, SeFile.class);
    }

    public void downloadFiles(String token, String path, List<SeFile> files, String dbid, String fileId) throws IOException {
        File f = new File(path);
        if (!f.exists() || !f.isDirectory()) {
            f.mkdirs();
        }

        for (SeFile seFile : files) {
            String filePath = path + File.separator + seFile.getName();
            f = new File(filePath);
            if (f.exists() && f.length() == seFile.getSize()) {
                continue;
            }
            if (f.exists() && f.length() < seFile.getSize()) {
                f.delete();
            }

            String uri = String.format("%sfile-service/fileparser/cluster/download/%s?token=%s&dbid=%s&cluster_fileid=%s",
                    host, seFile.getName(), token, dbid, fileId);
            downloadFile(uri, filePath);
        }
    }

    private void downloadFile(String uri, String filePath) throws IOException {
        URL url = new URL(uri);
        URLConnection conn = url.openConnection();
        InputStream is = conn.getInputStream();

        byte[] buffer = new byte[1024];
        FileOutputStream fs = new FileOutputStream(filePath);

        int read = 0, sum = 0;
        while ((read = is.read(buffer)) != -1) {
            sum += read;
            fs.write(buffer, 0, read);
        }

        fs.flush();
        fs.close();
        is.close();
    }
}