From 3600a4aecf7bf3636a3985b77a65cc7c185afc04 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 11 十一月 2024 16:15:29 +0800
Subject: [PATCH] 1

---
 src/main/java/com/se/simu/service/DbService.java |   70 +++++++++++++++++++++++++++++++++-
 1 files changed, 67 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/se/simu/service/DbService.java b/src/main/java/com/se/simu/service/DbService.java
index eaac946..15305bb 100644
--- a/src/main/java/com/se/simu/service/DbService.java
+++ b/src/main/java/com/se/simu/service/DbService.java
@@ -1,16 +1,21 @@
 package com.se.simu.service;
 
+import cn.hutool.json.JSONObject;
 import com.se.simu.config.PropertiesConfig;
 import com.se.simu.domain.dto.GeDb;
-import com.se.simu.helper.HttpHelper;
-import com.se.simu.helper.StringHelper;
+import com.se.simu.domain.vo.QueryVo;
+import com.se.simu.helper.*;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.http.HttpStatus;
 import org.springframework.stereotype.Service;
+import org.springframework.web.client.RestTemplate;
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import java.util.Date;
+import java.util.HashMap;
+import java.util.Map;
 
 @Slf4j
 @Service
@@ -20,13 +25,18 @@
     PropertiesConfig config;
 
     @Resource
+    RestTemplate restTemplate;
+
+    @Resource
     GedbService gedbService;
+
+    private final static String PUBLIC_KEY = "where_public_key";
 
     private final static long ONE_DAY = 24 * 60 * 60 * 1000;
 
     public void info(HttpServletRequest req, HttpServletResponse res) throws Exception {
         String token = gedbService.getToken();
-        GeDb db = gedbService.getSeDb(token);
+        GeDb db = gedbService.getGeDb(token);
         String date = StringHelper.YMD_FORMAT.format(new Date(System.currentTimeMillis() - ONE_DAY));
 
         String uri = String.format("%sgeo-service/statis/layer/data/info?dbid=%s&token=%s&caldate=%s", config.getHost(), db.getDbid(), token, date);
@@ -34,4 +44,58 @@
         HttpHelper helper = new HttpHelper();
         helper.service(req, res, uri);
     }
+
+    public void config(HttpServletRequest req, HttpServletResponse res) throws Exception {
+        String token = gedbService.getToken();
+        GeDb db = gedbService.getGeDb(token);
+
+        String uri = String.format("%sgeo-service/entitydb/map/config?dbid=%s&token=%s", config.getHost(), db.getDbid(), token);
+
+        HttpHelper helper = new HttpHelper();
+        helper.service(req, res, uri);
+    }
+
+    private String getPublicKey() {
+        Object obj = CaffeineHelper.get(PUBLIC_KEY);
+        if (obj instanceof GeDb) {
+            return (String) obj;
+        }
+
+        JSONObject jsonObject = restTemplate.getForObject(config.getHost() + "geo-service/setting/publickey", JSONObject.class);
+
+        String key = jsonObject.getStr("data");
+        if (!StringHelper.isEmpty(key)) {
+            CaffeineHelper.put(PUBLIC_KEY, key);
+        }
+
+        return key;
+    }
+
+    public void query(QueryVo vo, HttpServletRequest req, HttpServletResponse res) throws Exception {
+        String token = gedbService.getToken();
+        GeDb db = gedbService.getGeDb(token);
+        String url = config.getHost() + "geo-service/entitydbdata/layer/query";
+
+        Map<String, Object> map = new HashMap<>(6);
+        map.put("token", token);
+        map.put("dbid", db.getDbid());
+        map.put("layerid", vo.getLayerid());
+        //map.put("returnCountOnly", true);
+        //map.put("inSR", 4326);
+        map.put("containCount", vo.getContainCount());
+        map.put("count", vo.getCount());
+        map.put("start", vo.getStart());
+        map.put("querytype", vo.getQuerytype());
+        if (StringHelper.isEmpty(vo.getWhere())) {
+            RsaHelper.setPublicKey(getPublicKey());
+            String where = RsaHelper.encrypt(vo.getWhere());
+            map.put("where", vo.getWhere());
+        }
+        if (!StringHelper.isEmpty(vo.getGeometry())) {
+            map.put("geometry", vo.getGeometry());
+        }
+
+        String rs = restTemplate.postForObject(url, map, String.class);
+        WebHelper.writeStr2Page(res, HttpStatus.OK, rs);
+    }
 }

--
Gitblit v1.9.3