From 377e7a0a05632464bb6c7a78f50b038ba505cfef Mon Sep 17 00:00:00 2001
From: cy <1664593601@qq.com>
Date: 星期三, 11 十月 2023 15:05:50 +0800
Subject: [PATCH] fix(quartz): 1.增加超时重试机制2.数据条数不统一时删除拉取到的数据

---
 src/main/java/com/integrated/zyyt/util/ZyytUtil.java |  135 ++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 121 insertions(+), 14 deletions(-)

diff --git a/src/main/java/com/integrated/zyyt/util/ZyytUtil.java b/src/main/java/com/integrated/zyyt/util/ZyytUtil.java
index acf13b3..bd3fc66 100644
--- a/src/main/java/com/integrated/zyyt/util/ZyytUtil.java
+++ b/src/main/java/com/integrated/zyyt/util/ZyytUtil.java
@@ -1,7 +1,9 @@
 package com.integrated.zyyt.util;
 
 import cn.hutool.core.convert.Convert;
+import cn.hutool.core.io.IORuntimeException;
 import cn.hutool.http.HttpRequest;
+import cn.hutool.http.HttpResponse;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.integrated.zyyt.ZyytConstant;
@@ -13,8 +15,12 @@
 import com.integrated.zyyt.enetity.business.ZyytTShkdrb;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.HttpStatus;
 import org.springframework.jdbc.core.JdbcTemplate;
 import org.springframework.stereotype.Component;
+import org.springframework.util.CollectionUtils;
+import org.springframework.util.StringUtils;
 
 import java.time.LocalDate;
 import java.util.HashMap;
@@ -31,6 +37,27 @@
 @Component
 @Slf4j
 public class ZyytUtil {
+    /**
+     * 瓒呮椂閲嶈瘯娆℃暟
+     */
+    private static int TRYNUM = 3;
+
+    @Value("${pullData.tryNum}")
+    public void setTRYNUM(Integer tryNum) {
+        tryNum = tryNum <= 0 ? 3 : tryNum;
+        this.TRYNUM = tryNum;
+    }
+
+    /**
+     * http瓒呮椂鏃堕棿璁剧疆
+     */
+    private static int TRYTIMEOUT = 3 * 1000;
+
+    @Value("${pullData.tryTimeOut}")
+    public void setTRYTIMEOUT(Integer tryTimeOut) {
+        this.TRYTIMEOUT = tryTimeOut * 1000;
+    }
+
     private static JdbcTemplate jdbcTemplate;
 
     @Autowired
@@ -67,19 +94,54 @@
         params.put("ciphertext", eciphrtext);
         params.put("client", client);
         params.put("secret", secret);
-
-        String httpResult = HttpRequest.post(url)
+        HttpRequest httpRequest = HttpRequest.post(url)
                 .form(params)
                 .header("X-EOS-SourceSysKey", sourceSysKey)
-                .execute()
-                .body();
+                .timeout(TRYTIMEOUT);
+        log.info("瓒呮椂鏃堕棿涓簕}姣", TRYTIMEOUT);
         log.info("url =銆嬨�媨}", url);
         log.info("params =銆嬨�媨}", params);
+        HttpResponse httpResponse = null;
+        int tryNumTemp = 1;
+        boolean isTimeOut = false;
+        try {
+            httpResponse = httpRequest.execute();
+        } catch (IORuntimeException e) {
+//            e.printStackTrace();
+            // 瓒呮椂閲嶈瘯
+            isTimeOut = true;
+            log.error(e.getMessage());
+        }
+        while (tryNumTemp <= TRYNUM && isTimeOut) {
+            log.error("璇锋眰鎺ュ彛瓒呮椂锛岃繘琛岀{}娆¢噸璇曪紒", tryNumTemp);
+            try {
+                httpResponse = httpRequest.execute();
+                isTimeOut = false;
+            } catch (IORuntimeException e) {
+//                e.printStackTrace();
+                // 瓒呮椂閲嶈瘯
+                log.error(e.getMessage());
+            }
+            tryNumTemp++;
+        }
+        if (httpResponse == null) {
+            log.error("璇锋眰鎺ュ彛瓒呮椂锛岃鑱旂郴鎺ュ彛鎻愪緵鏂癸紒");
+            return null;
+        }
+        int responseStatus = httpResponse.getStatus();
+        if (HttpStatus.NOT_FOUND.value() == responseStatus) {
+            log.error("鎺ュ彛API寮傚父锛岃鑱旂郴鎺ュ彛鎻愪緵鏂癸紒HttpStatus锛歿}", responseStatus);
+            return null;
+        } else if (HttpStatus.OK.value() != responseStatus) {
+            log.error("璇锋眰鎺ュ彛鐘舵�佹湁璇紒HttpStatus锛歿}", responseStatus);
+            return null;
+        }
+        String httpResult = httpResponse.body();
         log.info("result =銆嬨�媨}", httpResult);
         try {
             return JSONObject.parseObject(httpResult, TokenResult.class);
         } catch (Exception e) {
-            log.error(e.getMessage());
+            log.error("Token瑙f瀽澶辫触锛�", e.getMessage());
             return null;
         }
     }
@@ -94,6 +156,10 @@
      * @return
      */
     public static <T> ZyytDataResult<T> getData(String url, String bearToken, String sourceSysKey, JSONObject params, int pageNum, int pageSize, Class<T> tClass) {
+        if (StringUtils.isEmpty(bearToken)) {
+            log.error("bearToken涓嶈兘涓簄ull锛�");
+            return null;
+        }
         HashMap header = new HashMap();
         header.put("Authorization", "Bearer " + bearToken);
         header.put("X-EOS-SourceSysKey", sourceSysKey);
@@ -102,23 +168,55 @@
         log.info("url =銆嬨�媨}", trueUrl);
         log.info("header=>>bearToken銆嬨�媨}\r\nSourceSysKey=>>{}", bearToken, sourceSysKey);
         log.info("params =銆嬨�媨}", params);
-        String httpResult = HttpRequest.post(trueUrl)
+        HttpRequest httpRequest = HttpRequest.post(trueUrl)
                 .addHeaders(header)
                 .contentType("application/json")
                 .body(params.toJSONString())
-                .timeout(1000 * 50)
-                .execute()
-                .body();
+                .timeout(TRYTIMEOUT);
+        log.info("瓒呮椂鏃堕棿涓簕}姣", TRYTIMEOUT);
+        HttpResponse httpResponse = null;
+
+        int tryNumTemp = 1;
+        boolean isTimeOut = false;
+        try {
+            httpResponse = httpRequest.execute();
+        } catch (IORuntimeException e) {
+            // 瓒呮椂閲嶈瘯
+            isTimeOut = true;
+            log.error(e.getMessage());
+        }
+        while (tryNumTemp <= TRYNUM && isTimeOut) {
+            log.error("璇锋眰鎺ュ彛瓒呮椂锛岃繘琛岀{}娆¢噸璇曪紒", tryNumTemp);
+            try {
+                httpResponse = httpRequest.execute();
+                isTimeOut = false;
+            } catch (IORuntimeException e) {
+//                e.printStackTrace();
+                // 瓒呮椂閲嶈瘯
+                log.error(e.getMessage());
+            }
+            tryNumTemp++;
+        }
+        if (httpResponse == null) {
+            log.error("璇锋眰鎺ュ彛瓒呮椂锛岃鑱旂郴鎺ュ彛鎻愪緵鏂癸紒");
+            return null;
+        }
+        int responseStatus = httpResponse.getStatus();
+        if (HttpStatus.NOT_FOUND.value() != responseStatus) {
+            log.error("鎺ュ彛API寮傚父锛岃鑱旂郴鎺ュ彛鎻愪緵鏂癸紒HttpStatus锛歿}", responseStatus);
+            return null;
+        } else if (HttpStatus.OK.value() != responseStatus) {
+            log.error("璇锋眰鎺ュ彛鐘舵�佹湁璇紒HttpStatus锛歿}", responseStatus);
+            return null;
+        }
+        String httpResult = httpResponse.body();
         try {
             JSONObject respJson = JSONObject.parseObject(httpResult);
-            log.info("寮�濮嬭В鏋愮粨鏋�4");
             Long contentSize = respJson.getLong("contentSize");
             if (contentSize != null && contentSize.compareTo(0L) > 0) {
                 JSONArray dataArray = respJson.getJSONArray("content");
-                log.info("寮�濮嬭В鏋愮粨鏋�5");
                 List<T> content = dataArray.toJavaList(tClass);
                 respJson.remove("content");
-                log.info("寮�濮嬭В鏋愮粨鏋�6");
                 ZyytDataResult<T> zyytDataResult = respJson.toJavaObject(ZyytDataResult.class);
                 zyytDataResult.setContent(content);
                 return zyytDataResult;
@@ -128,7 +226,7 @@
             }
         } catch (Exception e) {
             log.error("getData 寮傚父");
-            e.printStackTrace();
+//            e.printStackTrace();
             return null;
         }
     }
@@ -144,7 +242,12 @@
         String sql = "SELECT*FROM (SELECT*FROM ZYYT_RECORD WHERE BATCH_NO='" + getBatchNo(date) + "'  AND TABLE_NAME='" + tableName + "' ORDER BY START_TIME DESC) tmp WHERE ROWNUM<=1";
         Map<String, Object> map = null;
         try {
-            map = jdbcTemplate.queryForMap(sql);
+//            map = jdbcTemplate.queryForMap(sql);
+            List<Map<String, Object>> maps = jdbcTemplate.queryForList(sql);
+            if (CollectionUtils.isEmpty(maps)) {
+                return map;
+            }
+            map = maps.get(0);
         } catch (Exception e) {
             log.error("sql=={} 鏌ヨ鍙戠敓浜嗗紓甯�,鍙兘鍥犱负娌℃湁鏌ヨ鍒版暟鎹�", sql);
             log.error(e.getMessage());
@@ -278,6 +381,10 @@
     public static Long getApiTotalElements(String tableName, LocalDate date) {
         Zyyt businessEntity = getBusinessEntity(tableName, date);
         ZyytDataResult zyytDataResult = businessEntity.getData(0, 1);
+        if (zyytDataResult == null) {
+            log.info("鎷夊彇鏁版嵁澶辫触锛屾媺鍙栬繑鍥炲�间负null");
+            return 0l;
+        }
         log.info("鎬诲叡鎷夊彇鍒版暟鎹畕}鏉�", zyytDataResult.getTotalElements());
         return zyytDataResult.getTotalElements();
     }

--
Gitblit v1.9.1