feat：IDC数据源定时check

jeecg-boot-base-core/src/main/java/org/jeecg/common/TokenContext.java

@@ -12,6 +12,10 @@ public class TokenContext implements AutoCloseable{
         userToken.set(TokenUtils.threadToken());
     }
 
+    public void refreshToken(){
+        userToken.set(TokenUtils.refreshThreadToken());
+    }
+
     @Override
     public void close() throws Exception {
         userToken.remove();

jeecg-boot-base-core/src/main/java/org/jeecg/common/util/TokenUtils.java

@@ -178,8 +178,20 @@ public class TokenUtils {
         String secret = CommonConstant.TEMP_TOKEN_SECRET;
         // 模拟登录生成Token
         String token = JwtUtil.sign(username, secret);
-        // 设置Token缓存有效时间为 5 分钟
+        // 设置Token缓存有效时间为 6h
-        redisUtil.set(RedisConstant.THREAD_TOKEN, token, 5 * 60);
+        redisUtil.set(RedisConstant.THREAD_TOKEN, token, 6 * 60 * 60);
+        return token;
+    }
+
+    public static String refreshThreadToken(){
+        RedisUtil redisUtil = SpringContextUtils.getBean(RedisUtil.class);
+
+        String username = CommonConstant.TEMP_TOKEN_USERNAME;
+        String secret = CommonConstant.TEMP_TOKEN_SECRET;
+        // 模拟登录生成Token
+        String token = JwtUtil.sign(username, secret);
+        // 设置Token缓存有效时间为 6h
+        redisUtil.set(RedisConstant.THREAD_TOKEN, token, 6 * 60 * 60);
         return token;
     }
 }

jeecg-module-abnormal-alarm/src/main/java/org/jeecg/modules/idc/IDCDataFetch.java

@@ -1,6 +1,7 @@
 package org.jeecg.modules.idc;
 
 import cn.hutool.core.map.MapUtil;
+import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
 import lombok.Data;
 import lombok.extern.slf4j.Slf4j;
@@ -26,80 +27,26 @@ import java.util.concurrent.TimeUnit;
 import static org.jeecg.common.util.TokenUtils.getTempToken;
 import static org.jeecg.modules.base.enums.Template.IDC_DATASOURCE_STATUS;
 
-@Component
 @Slf4j
+@Component
 public class IDCDataFetch {
 
-    @Value("${inland.url}")
-    private String urlM; // 本地数据源url，即为主数据源
-
-    @Value("${inland.username}")
-    private String usernameM;
-
-    @Value("${inland.password}")
-    private String passwordM;
-
-    @Value("${oversea.url}")
-    private String urlS; // 国外数据源url，即为从数据源
-
-    @Value("${oversea.username}")
-    private String usernameS;
-
-    @Value("${oversea.password}")
-    private String passwordS;
-
     @Autowired
-    private SystemClient systemClient;
+    private TemplateManager templateManager;
 
     // 定时拾取IDC数据
-    @Scheduled(fixedDelayString = "${request-interval}", timeUnit = TimeUnit.SECONDS)
+    @Scheduled(initialDelay = 5, fixedDelay = 10, timeUnit = TimeUnit.SECONDS)
     public void fetch() {
-        JdbcTemplate template;
-        MessageDTO messageDTO = new MessageDTO(null, IDC_DATASOURCE_STATUS.getCode(), "admin");
-        messageDTO.setType(MessageTypeEnum.XT.getType());
-        ConnR connR = JDBCUtil.isConnection(urlM, usernameM, passwordM);
-        if (connR.isConn()) {
-            try {
-                template = JDBCUtil.template(urlM, usernameM, passwordM);
-                // 查询IDC Data
-
-            } catch (Exception e) {
-                log.error("[inland数据源]IDC数据处理异常: {}", e.getMessage());
-            }
-            return;
-        }
-        UserTokenContext.setToken(getTempToken());
-        // 对发送警告消息时可能出现的异常进行捕获(503) 防止影响后续代码执行
         try {
-            // 给管理员发送预警信息
+            JdbcTemplate template = templateManager.getTemplate();
-            Map<String, Object> data = new HashMap<>();
+            if (ObjectUtil.isNull(template))
-            data.put("param1", "inland");
+                return;
-            data.put("param2", connR.getInfo());
+            // 查询...
-            messageDTO.setData(data);
+            String sql = "select count(1) from gards_stations";
-            systemClient.sendTo(messageDTO);
+            Integer integer = template.queryForObject(sql, Integer.class);
+            System.out.println("========count========>" + integer);
         }catch (Exception e){
-            log.error("发送inland数据源异常信息失败: {}", e.getMessage());
+            log.error("{}", e.getMessage());
         }
-        // 使用备用数据源
-        connR = JDBCUtil.isConnection(urlS, usernameS, passwordS);
-        if (connR.isConn()) {
-            try {
-                template = JDBCUtil.template(urlS, usernameS, passwordS);
-                // 查询IDC Data
-
-            } catch (Exception e) {
-                log.error("[oversea数据源]IDC数据处理异常: {}", e.getMessage());
-            }
-            return;
-        }
-        try {
-            // 给管理员发送预警信息
-            Map<String, Object> data = ParamUtil.set("oversea", connR.getInfo());
-            messageDTO.setData(data);
-            systemClient.sendTo(messageDTO);
-        }catch (Exception e){
-            log.error("发送oversea数据源异常信息失败: {}", e.getMessage());
-        }
-        UserTokenContext.remove();
     }
 }

jeecg-module-abnormal-alarm/src/main/java/org/jeecg/modules/idc/TemplateManager.java

@@ -0,0 +1,119 @@
+package org.jeecg.modules.idc;
+
+import com.alibaba.nacos.api.config.annotation.NacosValue;
+import feign.FeignException;
+import lombok.extern.slf4j.Slf4j;
+import org.jeecg.common.TokenContext;
+import org.jeecg.common.api.dto.message.MessageDTO;
+import org.jeecg.common.constant.enums.MessageTypeEnum;
+import org.jeecg.common.util.JDBCUtil;
+import org.jeecg.modules.base.dto.ConnR;
+import org.jeecg.modules.feignclient.SystemClient;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.cloud.context.config.annotation.RefreshScope;
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
+import java.util.concurrent.TimeUnit;
+
+import static org.jeecg.modules.base.enums.Template.IDC_DATASOURCE_STATUS;
+
+@Slf4j
+@Component
+@RefreshScope
+public class TemplateManager {
+
+    @Value("${inland.url}")
+    private String urlM; // 本地数据源url，即为主数据源
+
+    @Value("${inland.username}")
+    private String usernameM;
+
+    @Value("${inland.password}")
+    private String passwordM;
+
+    @Value("${oversea.url}")
+    private String urlS; // 国外数据源url，即为从数据源
+
+    @Value("${oversea.username}")
+    private String usernameS;
+
+    @Value("${oversea.password}")
+    private String passwordS;
+
+    @Autowired
+    private SystemClient systemClient;
+
+    private final String templateKey = "template";
+
+    private final ConcurrentMap<String, JdbcTemplate> templateMap = new ConcurrentHashMap<>();
+
+    private TemplateManager(){}
+
+    private void setTemplate(JdbcTemplate template){
+        templateMap.put(templateKey, template);
+    }
+
+    public JdbcTemplate getTemplate(){
+        return templateMap.get(templateKey);
+    }
+
+    /*
+    *   定时检查inland/oversea数据源的的连接状态
+    * */
+    @Scheduled(fixedDelay = 10, timeUnit = TimeUnit.SECONDS)
+    private void checkTemplate(){
+        try (TokenContext tokenContext = new TokenContext()){
+            MessageDTO messageDTO = new MessageDTO(null, IDC_DATASOURCE_STATUS.getCode(), "admin");
+            messageDTO.setType(MessageTypeEnum.XT.getType());
+            ConnR connR = JDBCUtil.isConnection(urlM, usernameM, passwordM);
+            // 如果inland数据源不可用 尝试oversea数据源
+            JdbcTemplate template;
+            if (connR.isConn()){
+                template = JDBCUtil.template(urlM, usernameM, passwordM);
+                this.setTemplate(template);
+            } else {
+                // 为当前线程设置临时Token 避免OpenFeign调用其它服务接口时403
+                tokenContext.setToken();
+
+                // 给admin发送数据源异常消息
+                Map<String, Object> data = new HashMap<>();
+                data.put("param1", "inland");
+                data.put("param2", connR.getInfo());
+                messageDTO.setData(data);
+                try { systemClient.sendTo(messageDTO); }
+                catch (FeignException.Unauthorized e) {
+                    // 刷新临时Token 重新发送信息
+                    tokenContext.refreshToken();
+                    systemClient.sendTo(messageDTO);
+                }
+                catch(Exception e) {log.error("发送inland数据源异常信息失败: {}" , e.getMessage());}
+                connR = JDBCUtil.isConnection(urlS, usernameS, passwordS);
+                if (connR.isConn()){
+                    template = JDBCUtil.template(urlS, usernameS, passwordS);
+                    this.setTemplate(template);
+                    return;
+                }
+                // 给admin发送数据源异常消息
+                data.put("param1", "oversea");
+                data.put("param2", connR.getInfo());
+                messageDTO.setData(data);
+                try { systemClient.sendTo(messageDTO); }
+                catch (FeignException.Unauthorized e) {
+                    // 刷新临时Token 重新发送信息
+                    tokenContext.refreshToken();
+                    systemClient.sendTo(messageDTO);
+                }
+                catch(Exception e) {log.error("发送oversea数据源异常信息失败: {}" , e.getMessage());}
+            }
+        }catch (Exception e){
+            log.error("定时检查inland/oversea数据源状态异常: {}", e.getMessage());
+        }
+    }
+}