From 138b959cc11dc9a73b0c766030b99ba1180d8650 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 11 十一月 2024 14:32:48 +0800
Subject: [PATCH] 11

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

diff --git a/src/main/java/com/se/simu/service/GedbService.java b/src/main/java/com/se/simu/service/GedbService.java
index 35e5d3c..81dd818 100644
--- a/src/main/java/com/se/simu/service/GedbService.java
+++ b/src/main/java/com/se/simu/service/GedbService.java
@@ -35,12 +35,6 @@
 import java.util.*;
 import java.util.stream.Collectors;
 
-/**
- * GEDB鏈嶅姟绫�
- *
- * @author WWW
- * @date   2024-09-12
- */
 @Slf4j
 @Service
 @SuppressWarnings("ALL")
@@ -52,6 +46,8 @@
 
     @Resource
     RestTemplate restTemplate;
+
+    private final static String DB_KEY = "gedb_db";
 
     private final static String TOKEN_KEY = "gedb_token";
 
@@ -128,17 +124,25 @@
         return db;
     }
 
-    private GeDb getSeDb(String token) {
+    public GeDb getSeDb(String token) {
+        Object obj = CaffeineHelper.get(DB_KEY);
+        if (obj instanceof GeDb) {
+            return (GeDb) obj;
+        }
+
         Map<String, Object> map = new HashMap<>(1);
         map.put("token", token);
 
-        JSONObject obj = restTemplate.postForObject(config.getHost() + "geo-service/entitydb/list/canview", map, JSONObject.class);
-        JSONArray data = obj.getJSONArray("data");
+        JSONObject jsonObject = restTemplate.postForObject(config.getHost() + "geo-service/entitydb/list/canview", map, JSONObject.class);
+        JSONArray data = jsonObject.getJSONArray("data");
 
         List<GeDb> list = JSONUtil.toList(data, GeDb.class);
         if (CollectionUtils.isEmpty(list)) return null;
 
-        return list.stream().filter(db -> null != db.getName() && db.getName().contains(config.getDbName())).findFirst().orElse(null);
+        GeDb gedb = list.stream().filter(db -> null != db.getName() && db.getName().contains(config.getDbName())).findFirst().orElse(null);
+        if (null != gedb) CaffeineHelper.put(DB_KEY, gedb);
+
+        return gedb;
     }
 
     public List<GeLayer> getLayers(String token, GeDb db) {

--
Gitblit v1.9.3