代码优化

alien-job/src/main/java/shop/alien/job/store/AiTagJob.java

@@ -78,7 +78,7 @@ public class AiTagJob {
      */
     @XxlJob("getSaveTagsTask")
     public R<String> getSaveTagsTask() {
-        log.info("登录第三方服务获取token..." + loginUrl);
+        log.info("登录Ai服务获取token..." + loginUrl);
         //构建请求参数
         MultiValueMap<String, String> formData = new LinkedMultiValueMap<>();
         formData.add("username", userName);    // 表单字段 1：用户名
@@ -90,6 +90,7 @@ public class AiTagJob {
         HttpEntity<MultiValueMap<String, String>> requestEntity = new HttpEntity<>(formData, headers);
         ResponseEntity<String> postForEntity = null;
         try {
+            log.info("请求Ai服务登录接口===================>");
             postForEntity = restTemplate.postForEntity(loginUrl, requestEntity, String.class);
         } catch (Exception e) {
             log.error("类:PostMethod 方法:post", e);
@@ -97,24 +98,25 @@ public class AiTagJob {
 
         if (postForEntity != null) {
             if (postForEntity.getStatusCodeValue() == 200) {
-                log.info("postForEntity.getBody()\t" + postForEntity.getBody());
+                log.info("请求Ai服务登录成功 postForEntity.getBody()\t" + postForEntity.getBody());
                 String responseBody = postForEntity.getBody();
                 JSONObject jsonObject = JSONObject.parseObject(responseBody);
                 JSONObject dataJson = jsonObject.getJSONObject("data");
                 String accessToken = dataJson.getString("access_token");
-                //调用 获取AI服务 标签主表get请求
+                //调用 创建Ai标签评论数据方法
                 getSaveTagsTaskInfo(accessToken);
+                return R.success("任务执行成功");
             } else {
                 log.error("请求AI服务 登录接口失败 http状态：" + postForEntity.getStatusCode());
-                return null;
+                return R.fail("请求AI服务 登录接口失败 http状态:" + postForEntity.getStatusCode());
             }
         }
-
-        return null;
+        XxlJobHelper.handleFail("登录任务执行失败 返回异常");
+        return R.fail("登录任务执行失败 返回异常");
     }
 
     public R<String> getSaveTagsTaskInfo(String token) {
-        log.info("开始从第三方服务创建Ai标签评论数据...");
+        log.info("开始从Ai服务创建店铺评价标签评论数据...");
         log.info("restTemplate:" + restTemplate + "url:" + saveTagUrl);
 
         // 获取当前日期
@@ -134,8 +136,8 @@ public class AiTagJob {
         // 1. 构建请求头
         HttpHeaders headers = new HttpHeaders();
         headers.setContentType(MediaType.APPLICATION_JSON);
-        headers.set("Authorization", "Bearer " + token);
         // Token 认证
+        headers.set("Authorization", "Bearer " + token);
 
         AnalysisRequest jsonBody = new AnalysisRequest();
         jsonBody.setStart_time(startDate + "T00:00:00");
@@ -154,19 +156,23 @@ public class AiTagJob {
                 log.info("postForEntity.getBody()\t" + response.getStatusCode());
                 XxlJobHelper.handleSuccess("创建标签成功 状态码" + response.getStatusCode());
                 return R.success("创建成功");
+            } else {
+                log.error("请求AI服务 创建标签数据失败 http状态：" + response.getStatusCode());
+                XxlJobHelper.handleFail("请求AI服务 创建标签数据失败 http状态" + response.getStatusCode());
+                return R.fail("请求AI服务 创建标签数据失败 http状态：" + response.getStatusCode());
             }
         }
-        log.error("请求AI服务 创建标签数据失败 http状态：" + response.getStatusCode());
-        XxlJobHelper.handleFail("任务执行失败 状态码" + response.getStatusCode());
-        return R.fail("创建失败 状态码:" + response.getStatusCode());
+        log.error("创建标签任务执行失败 返回异常");
+        XxlJobHelper.handleFail("创建标签任务执行失败 返回异常");
+        return R.fail("创建标签任务执行失败 返回异常");
     }
 
     /**
      * 拉取Ai服务标签主表数据
      */
     @XxlJob("getTagMainTask")
-    public String getTagMainTask() {
-        log.info("登录第三方服务获取token..." + loginUrl);
+    public R<String> getTagMainTask() {
+        log.info("登录Ai服务获取token..." + loginUrl);
         //构建请求参数
         MultiValueMap<String, String> formData = new LinkedMultiValueMap<>();
         formData.add("username", userName);    // 表单字段 1：用户名
@@ -185,23 +191,25 @@ public class AiTagJob {
 
         if (postForEntity != null) {
             if (postForEntity.getStatusCodeValue() == 200) {
-                log.info("postForEntity.getBody()\t" + postForEntity.getBody());
+                log.info("请求Ai服务登录成功 postForEntity.getBody()\t" + postForEntity.getBody());
                 String responseBody = postForEntity.getBody();
                 JSONObject jsonObject = JSONObject.parseObject(responseBody);
                 JSONObject dataJson = jsonObject.getJSONObject("data");
                 String accessToken = dataJson.getString("access_token");
                 //调用 获取AI服务 标签主表get请求
                 getTagMainTaskInfo(accessToken);
+                return R.success("任务执行成功");
             } else {
                 log.error("请求AI服务 登录接口失败 http状态：" + postForEntity.getStatusCode());
-                return null;
+                return R.fail("请求AI服务 登录接口失败 http状态:" + postForEntity.getStatusCode());
             }
         }
-        return null;
+        XxlJobHelper.handleFail("登录任务执行失败 返回异常");
+        return R.fail("登录任务执行失败 返回异常");
     }
 
     public R<String> getTagMainTaskInfo(String token) {
-        log.info("开始从第三方服务获取标签主表信息...");
+        log.info("开始从Ai服务获取标签主表信息...");
         log.info("restTemplate:" + restTemplate + "url:" + tagMainUrl);
         // 1. 构建请求头
         HttpHeaders headers = new HttpHeaders();
@@ -246,9 +254,9 @@ public class AiTagJob {
                 int num = tagsMainMapper.insertBatchTagsMain(tagsMainList);
                 XxlJobHelper.handleSuccess("拉取主标签数据成功 执行" + num + "条");
                 return R.success("拉取主标签数据成功 执行" + num + "条");
-            }else{
-                XxlJobHelper.handleSuccess("获取回显数据为空 状态码:"+ responseEntity.getStatusCodeValue());
-                return R.success("请求成功 获取回显数据为空 状态码:"+ responseEntity.getStatusCodeValue());
+            } else {
+                XxlJobHelper.handleSuccess("获取回显数据为空 状态码:" + responseEntity.getStatusCodeValue());
+                return R.success("请求成功 获取回显数据为空 状态码:" + responseEntity.getStatusCodeValue());
             }
         }
         XxlJobHelper.handleFail("任务执行失败 状态码" + responseEntity.getStatusCodeValue());
@@ -257,11 +265,11 @@ public class AiTagJob {
 
 
     /**
-     * 拉取Ai服务标签数据
+     * 拉取Ai服务子标签数据
      */
     @XxlJob("getTagsSynonymTask")
-    public String getTagsSynonymTask() {
-        log.info("登录第三方服务获取token..." + loginUrl);
+    public R<String> getTagsSynonymTask() {
+        log.info("登录Ai服务获取token..." + loginUrl);
         //构建请求参数
         MultiValueMap<String, String> formData = new LinkedMultiValueMap<>();
         formData.add("username", userName);    // 表单字段 1：用户名
@@ -280,24 +288,25 @@ public class AiTagJob {
 
         if (postForEntity != null) {
             if (postForEntity.getStatusCodeValue() == 200) {
-                log.info("postForEntity.getBody()\t" + postForEntity.getBody());
+                log.info("请求Ai服务登录成功 postForEntity.getBody()\t" + postForEntity.getBody());
                 String responseBody = postForEntity.getBody();
                 JSONObject jsonObject = JSONObject.parseObject(responseBody);
                 JSONObject dataJson = jsonObject.getJSONObject("data");
                 String accessToken = dataJson.getString("access_token");
                 //调用 获取AI服务 标签子表get请求
                 getTagsSynonymTaskInfo(accessToken);
+                return R.success("任务执行成功");
             } else {
                 log.error("请求AI服务 登录接口失败 http状态：" + postForEntity.getStatusCode());
-                return null;
+                return R.fail("请求AI服务 登录接口失败 http状态:" + postForEntity.getStatusCode());
             }
         }
-
-        return null;
+        XxlJobHelper.handleFail("登录任务执行失败 返回异常");
+        return R.fail("登录任务执行失败 返回异常");
     }
 
     public R<String> getTagsSynonymTaskInfo(String token) {
-        log.info("开始从第三方服务获取数据 获取所有标签子表信息...");
+        log.info("开始从Ai服务获取数据 获取所有标签子表信息...");
         log.info("restTemplate:" + restTemplate + "url:" + tagsSynonymUrl);
         // 1. 构建请求头
         HttpHeaders headers = new HttpHeaders();
@@ -344,22 +353,21 @@ public class AiTagJob {
                 int num = tagsSynonymMapper.insertBatchTagsSynonym(tagsSynonymList);
                 XxlJobHelper.handleSuccess("拉取子标签数据成功 执行" + num + "条");
                 return R.success("拉取子标签数据成功 执行" + num + "条");
-            }else{
-                XxlJobHelper.handleSuccess("获取回显数据为空"+"状态码:"+responseEntity.getStatusCodeValue());
-                return R.success("请求成功 获取回显数据为空"+"状态码:"+responseEntity.getStatusCodeValue());
+            } else {
+                XxlJobHelper.handleSuccess("获取回显数据为空" + "状态码:" + responseEntity.getStatusCodeValue());
+                return R.success("请求成功 获取回显数据为空" + "状态码:" + responseEntity.getStatusCodeValue());
             }
         }
-        log.info("AI服务子表请求失败，状态码:" + responseEntity.getStatusCodeValue());
         XxlJobHelper.handleFail("任务执行失败 状态码" + responseEntity.getStatusCodeValue());
-        return R.fail("AI服务子表请求失败，状态码:"+responseEntity.getStatusCodeValue());
+        return R.fail("任务执行失败 状态码" + responseEntity.getStatusCodeValue());
     }
 
     /**
-     * 拉取Ai服务获取所有推荐信息数据
+     * 拉取Ai服务获取推荐信息数据
      */
     @XxlJob("getStoreCommentSummaryTask")
-    public String getStoreCommentSummaryTask() {
-        log.info("登录第三方服务获取token..." + loginUrl);
+    public R<String> getStoreCommentSummaryTask() {
+        log.info("登录Ai服务获取token..." + loginUrl);
         //构建请求参数
         MultiValueMap<String, String> formData = new LinkedMultiValueMap<>();
         formData.add("username", userName);    // 表单字段 1：用户名
@@ -378,24 +386,25 @@ public class AiTagJob {
 
         if (postForEntity != null) {
             if (postForEntity.getStatusCodeValue() == 200) {
-                log.info("postForEntity.getBody()\t" + postForEntity.getBody());
+                log.info("请求Ai服务登录成功 postForEntity.getBody()\t" + postForEntity.getBody());
                 String responseBody = postForEntity.getBody();
                 JSONObject jsonObject = JSONObject.parseObject(responseBody);
                 JSONObject dataJson = jsonObject.getJSONObject("data");
                 String accessToken = dataJson.getString("access_token");
                 //调用 获取AI服务 获取所有推荐信表 get请求
                 getStoreCommentSummaryTaskInfo(accessToken);
+                return R.success("任务执行成功");
             } else {
                 log.error("请求AI服务 登录接口失败 http状态：" + postForEntity.getStatusCode());
-                return null;
+                return R.fail("请求AI服务 登录接口失败 http状态:" + postForEntity.getStatusCode());
             }
         }
-
-        return null;
+        XxlJobHelper.handleFail("登录任务执行失败 返回异常");
+        return R.fail("登录任务执行失败 返回异常");
     }
 
     public R<String> getStoreCommentSummaryTaskInfo(String token) {
-        log.info("开始从第三方服务获取数据 获取推荐信息数据 店铺评论总结表...");
+        log.info("开始从Ai服务获取数据 获取推荐信息数据 店铺评论总结表...");
         log.info("restTemplate:" + restTemplate + "url:" + allSummaryUrl);
         // 1. 构建请求头
         HttpHeaders headers = new HttpHeaders();
@@ -442,9 +451,9 @@ public class AiTagJob {
                 int num = storeCommentSummaryMapper.insertBatchStoreCommentSummary(storeCommentSummaries);
                 XxlJobHelper.handleSuccess("拉取子标签数据成功 执行" + num + "条");
                 return R.success("拉取子标签数据成功 执行" + num + "条");
-            }else {
-                XxlJobHelper.handleSuccess("获取回显数据为空"+"状态码:"+responseEntity.getStatusCodeValue());
-                return R.success("请求成功 获取回显数据为空"+"状态码:"+responseEntity.getStatusCodeValue());
+            } else {
+                XxlJobHelper.handleSuccess("获取回显数据为空" + "状态码:" + responseEntity.getStatusCodeValue());
+                return R.success("请求成功 获取回显数据为空" + "状态码:" + responseEntity.getStatusCodeValue());
             }
         }
         log.info("AI服务 推荐信息数据请求失败，状态码：" + responseEntity.getStatusCodeValue());