完成基础功能开发

13693261870

2024-09-14 d1a6b45132c7df365812237281e91f41f00fe90e

完成基础功能开发

 src/main/java/com/se/simu/controller/TaskController.java

@@ -28,7 +28,10 @@
    @PostMapping(value = "/createTask", produces = "application/json; charset=UTF-8")
    public R<Object> createTask() {
        try {
            Object rs = sedbService.test();
            // 469538.6536261877,4416744.922022615,469853.14714664617,4417049.378602433
            String bbox = "116.64388473935195,39.884315914604464,116.64754729082588,39.887069143903496";
            String taskName = "20240913";
            Object rs = sedbService.test(bbox, taskName);

            return success(rs);
        } catch (Exception ex) {

 src/main/java/com/se/simu/domain/SeFile.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,39 @@
package com.se.simu.domain;

/**
 * SE文件实体类
 *
 * @author WWW
 * @date 2024-09-14
 */
public class SeFile {
    String name;

    Long size;

    String type;

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public Long getSize() {
        return size;
    }

    public void setSize(Long size) {
        this.size = size;
    }

    public String getType() {
        return type;
    }

    public void setType(String type) {
        this.type = type;
    }
}

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

@@ -6,6 +6,7 @@
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;
@@ -18,6 +19,11 @@

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;

@@ -58,26 +64,42 @@
    @Value("#{'${sedb.sysFields}'}")
    List<String> sysFields;

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

    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";

    public boolean test(String bbox, String taskName) throws Exception {
        String token = getToken();

        SeDb db = getSeDb(token);
        db.setBbox(bbox);

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

        createShps(inPath + "\\20240913", layers);
        String basePath = inPath + File.separator + taskName;
        createPath(basePath);
        createShps(basePath, layers);

        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 {
@@ -119,7 +141,7 @@
        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);
        return list.stream().filter(db -> null != db.getName() && db.getName().contains(dbName)).findFirst().orElse(null);
    }

    public List<SeLayer> getLayers(String token, SeDb db) {
@@ -213,12 +235,6 @@
    }

    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)) {
@@ -226,4 +242,83 @@
            }
        }
    }

    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 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();
    }
}

 src/main/resources/application.yml

@@ -46,8 +46,9 @@
  host: http://106.120.22.26:8013/
  user: WUWEIWEI
  pwd: WUWEIWEI
  dbName: é€šå·žæ°´åˆ©ä»¿çœŸ4548
  dbName: é€šå·žæ°´åˆ©ä»¿çœŸ
  pageSize: 2000
  layerNames: ç®¡ç‚¹,管线,建筑物
  shpNames: pipePoint,pipeLine,building
  sysFields: _ext_attr,_meta_id,_attach_files,_x,_y,_z,operatetime,operator,_is_latest,_data_type,groupid,id,shape_length,shape_area
  demName: DEM