1
13693261870
2024-09-13 94493e6798ce4be955d1ff1f4d918463423ba9b5
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
package com.se.simu.service;
 
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 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.util.*;
import java.util.stream.Collectors;
 
/**
 * SDDB服务类
 *
 * @author WWW
 * @date   2024-09-12
 */
@Slf4j
@Service
@SuppressWarnings("ALL")
public class SedbService {
    @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 String test() throws Exception {
        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);
 
        return db.getDbid();
    }
 
    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);
                }
            }
        }
    }
 
    public 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");
    }
 
    public 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");
    }
}