摘要提取接口访问异常处理

src/main/java/com/zhiwei/brandkbs2/controller/app/AppToolsetController.java

@@ -17,6 +17,8 @@ import com.zhiwei.brandkbs2.util.RedisUtil;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.apache.commons.lang3.StringUtils;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.http.HttpEntity;
@@ -41,6 +43,7 @@ import java.util.*;
 @Api(tags = "工具集", description = "工具集")
 @Auth(role = RoleEnum.CUSTOMER)
 public class AppToolsetController {
+    public static final Logger log = LogManager.getLogger(AppToolsetController.class);
 
     @Resource(name = "redisUtil")
     private RedisUtil redisUtil;
@@ -163,10 +166,17 @@ public class AppToolsetController {
      * @param projectId 项目id
      * @return
      */
-    private JSONObject getUrlInfo(String url, String projectId){
-        String linkedGroupId = projectService.getProjectVOById(projectId).getBrandLinkedGroupId();
-        JSONObject jsonObject = restTemplate.getForEntity(articleInfoUrl, JSONObject.class, url, linkedGroupId, UserThreadLocal.getNickname()).getBody();
-        if (Objects.isNull(jsonObject) || !jsonObject.getBoolean("status")){
+    private JSONObject getUrlInfo(String url, String projectId) {
+        JSONObject jsonObject;
+        // 即使抛出异常(大概率会是timeout)，也要保证批量时其他链接正常执行
+        try {
+            String linkedGroupId = projectService.getProjectVOById(projectId).getBrandLinkedGroupId();
+            jsonObject = restTemplate.getForEntity(articleInfoUrl, JSONObject.class, url, linkedGroupId, UserThreadLocal.getNickname()).getBody();
+        } catch (Exception e) {
+            log.info("url:{},访问链接信息提取接口异常-", url, e);
+            return null;
+        }
+        if (Objects.isNull(jsonObject) || !jsonObject.getBoolean("status")) {
             return null;
         }
         return jsonObject.getJSONObject("data");
@@ -178,16 +188,23 @@ public class AppToolsetController {
      * @return
      */
     private String getArticleSummaryResult(String text){
+        // 即使抛出异常(大概率会是timeout)，也要保证批量时其他链接正常执行
+        String errorString = "访问超时,请稍后重试此条数据";
         if (Objects.isNull(text)){
-            return null;
+            return errorString;
+        }
+        try {
+            // 拼接提示词模板
+            String resultText = TEXT_SUMMARY_PREVIOUS + StringUtils.substring(text, 0, 5000);
+            // 请求参数 请求头
+            HttpHeaders headers = new HttpHeaders();
+            headers.setContentType(MediaType.TEXT_PLAIN);
+            HttpEntity<String> request = new HttpEntity<>(resultText, headers);
+            ResponseEntity<String> response = restTemplate.postForEntity(articleSummaryUrl, request, String.class);
+            return response.getBody();
+        }catch (Exception e){
+            log.info("访问摘要提取接口异常-", e);
+            return errorString;
         }
-        // 拼接提示词模板
-        String resultText = TEXT_SUMMARY_PREVIOUS + StringUtils.substring(text, 0, 5000);
-        // 请求参数 请求头
-        HttpHeaders headers = new HttpHeaders();
-        headers.setContentType(MediaType.TEXT_PLAIN);
-        HttpEntity<String> request = new HttpEntity<>(resultText, headers);
-        ResponseEntity<String> response = restTemplate.postForEntity(articleSummaryUrl, request, String.class);
-        return response.getBody();
     }
 }