From 0f4da6101dd8552cc91475b1a0468e2847316c7d Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期二, 05 十一月 2024 15:09:59 +0800
Subject: [PATCH] 11

---
 src/main/java/com/se/simu/service/GedbService.java |   18 ++++++++++++++++++
 1 files changed, 18 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/se/simu/service/GedbService.java b/src/main/java/com/se/simu/service/GedbService.java
index 47e7584..c13f549 100644
--- a/src/main/java/com/se/simu/service/GedbService.java
+++ b/src/main/java/com/se/simu/service/GedbService.java
@@ -10,6 +10,7 @@
 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.CaffeineHelper;
 import com.se.simu.helper.RsaHelper;
 import com.se.simu.helper.ShpHelper;
 import com.se.simu.helper.StringHelper;
@@ -51,6 +52,8 @@
     @Resource
     RestTemplate restTemplate;
 
+    private final static String TOKEN_KEY = "gedb_token";
+
     public boolean test(DataPo data) throws Exception {
         createPath(config.getInPath() + File.separator + data.getInPath());
 
@@ -73,11 +76,26 @@
     }
 
     public String getToken() throws Exception {
+        Object obj = CaffeineHelper.get(TOKEN_KEY);
+        if (obj instanceof String) {
+            return obj.toString();
+        }
+
+        String token = getTokenByServer();
+        if (null == token) throw new Exception("鑾峰彇GEDB浠ょ墝澶辫触");
+
+        CaffeineHelper.put(TOKEN_KEY, token);
+
+        return token;
+    }
+
+    private String getTokenByServer() throws Exception {
         Map<String, Object> map = new HashMap<>(2);
         map.put("userid", config.getUser());
         map.put("password", getPassword());
 
         JSONObject obj = restTemplate.postForObject(config.getHost() + "account-service/security/login", map, JSONObject.class);
+        log.info(obj.toString());
 
         JSONObject data = obj.getJSONObject("data");
 

--
Gitblit v1.9.3