From ca4cfb3b8c78ef2a25bde11ce6026653732b00fd Mon Sep 17 00:00:00 2001
From: lxu75 <liang.xu.4@volvocars.com>
Date: Thu, 6 Mar 2025 18:43:47 +0800
Subject: [PATCH] =?UTF-8?q?=E8=88=86=E6=83=85=E8=87=AA=E5=8A=A8=E5=8C=96?=
 =?UTF-8?q?=E4=BB=A3=E7=A0=81=E7=BC=96=E5=86=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../center/entity/AiAnalysisRequestLogs.java  |  8 ++--
 .../impl/MqMessageRecordServiceImpl.java      | 45 ++++++++++++++-----
 2 files changed, 39 insertions(+), 14 deletions(-)

diff --git a/ai-analytic-center-api/src/main/java/com/volvo/ai/analytic/center/entity/AiAnalysisRequestLogs.java b/ai-analytic-center-api/src/main/java/com/volvo/ai/analytic/center/entity/AiAnalysisRequestLogs.java
index 457c93a..f5ba13b 100644
--- a/ai-analytic-center-api/src/main/java/com/volvo/ai/analytic/center/entity/AiAnalysisRequestLogs.java
+++ b/ai-analytic-center-api/src/main/java/com/volvo/ai/analytic/center/entity/AiAnalysisRequestLogs.java
@@ -2,13 +2,15 @@ package com.volvo.ai.analytic.center.entity;
 
 import com.baomidou.mybatisplus.annotation.*;
 import com.volvo.common.core.base.BaseEntity;
-import lombok.Builder;
-import lombok.Data;
+import lombok.*;
 
 
-@Data
+@Setter
+@Getter
 @TableName("tt_ai_analysis_request_logs")
 @Builder
+@AllArgsConstructor
+@NoArgsConstructor
 public class AiAnalysisRequestLogs extends BaseEntity {
 
     @TableId(value = "id", type = IdType.AUTO)
diff --git a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/MqMessageRecordServiceImpl.java b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/MqMessageRecordServiceImpl.java
index 8bf4cdf..b540f6e 100644
--- a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/MqMessageRecordServiceImpl.java
+++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/MqMessageRecordServiceImpl.java
@@ -4,6 +4,7 @@ import cn.hutool.core.date.DateUtil;
 import com.alibaba.cloud.commons.lang.StringUtils;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
+import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.volvo.ai.analytic.center.constant.Constant;
 import com.volvo.ai.analytic.center.dto.req.*;
@@ -25,6 +26,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.jdbc.core.BeanPropertyRowMapper;
 import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.scheduling.annotation.Async;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
@@ -88,15 +90,8 @@ public class MqMessageRecordServiceImpl extends ServiceImpl<MqMessageRecordMappe
         }
         // 生成ai分析请求id
         String aiAnalysisRequestId = AiAnalysisUtils.getAiAnalysisRequestId(BusinessTypeEnum.COMMUNITYTARGET.getCode());
-
-        // 保存请求日志
-        aiAnalysisRequestLogsMapper.insert(AiAnalysisRequestLogs.builder()
-                .aiAnalysisRequestId(aiAnalysisRequestId)
-                .businessRequest(message)
-                .difyAgentKey(flowId)
-                .difyRequest(JSON.toJSONString(communityTargetDTO))
-                .aiAnalysisRequestType(BusinessTypeEnum.COMMUNITYTARGET.getCode())
-                .build());
+        // 异步保存请求日志
+        syncSaveRequestLogs(message, aiAnalysisRequestId);
 
         JSONObject difResult = new JSONObject();
         try {
@@ -108,7 +103,6 @@ public class MqMessageRecordServiceImpl extends ServiceImpl<MqMessageRecordMappe
                 communityTargetDTO.getContent().forEach(contentNode -> {
                     if (contentNode.getNodeType().equals(NodeTypeEnum.IMAGE.getCode())) {
                         //构建图片分析请求对象
-
                         DifyImageWorkFlow diFyImageWorkFlow = new DifyImageWorkFlow();
                         diFyImageWorkFlow.setUrl(contentNode.getNodeContent());
 
@@ -133,7 +127,10 @@ public class MqMessageRecordServiceImpl extends ServiceImpl<MqMessageRecordMappe
             DifyCommunityTargetDTO difyCommunityTargetDTO = new DifyCommunityTargetDTO();
             difyCommunityTargetDTO.setTargetId(communityTargetDTO.getTargetId());
             difyCommunityTargetDTO.setTargetType(communityTargetDTO.getTargetType());
-            difyCommunityTargetDTO.setTargetContent(textContent + sb.toString());
+            difyCommunityTargetDTO.setTargetContent(textContent + sb);
+
+            //根据aiAnalysisRequestId更新请求日志表的difyRequest字段
+            syncUpdateDiFyRequest(difyCommunityTargetDTO, aiAnalysisRequestId);
 
             DiFyReq diFyReq = new DiFyReq();
             diFyReq.setUser(user);
@@ -154,6 +151,8 @@ public class MqMessageRecordServiceImpl extends ServiceImpl<MqMessageRecordMappe
                 //返回结果推送到社区的MQ
                 rocketMQTemplate.syncSend("", JSON.toJSONString(difyCommunityTargetResult));
                 log.info("舆情分析发送回调MQ完成: {}", JSON.toJSONString(difyCommunityTargetResult));
+                //异步更新请求日志表的difyResponse字段
+                syncUpdateDiFyResponse(difResult, aiAnalysisRequestId);
             }
         } catch (Exception e) {
             log.error("请求DiFy异常:{}", e);
@@ -167,6 +166,30 @@ public class MqMessageRecordServiceImpl extends ServiceImpl<MqMessageRecordMappe
         return true;
     }
 
+    @Async
+    protected void syncUpdateDiFyRequest(DifyCommunityTargetDTO difyCommunityTargetDTO, String aiAnalysisRequestId) {
+        aiAnalysisRequestLogsMapper.update(new AiAnalysisRequestLogs(),
+                new UpdateWrapper<AiAnalysisRequestLogs>().set("dify_request", JSON.toJSONString(difyCommunityTargetDTO))
+                        .eq("ai_analysis_request_id", aiAnalysisRequestId));
+    }
+
+    @Async
+    protected void syncUpdateDiFyResponse(JSONObject difResult, String aiAnalysisRequestId) {
+        aiAnalysisRequestLogsMapper.update(new AiAnalysisRequestLogs(),
+                new UpdateWrapper<AiAnalysisRequestLogs>().set("business_response", difResult)
+                        .eq("ai_analysis_request_id", aiAnalysisRequestId));
+    }
+
+    @Async
+    protected void syncSaveRequestLogs(String message, String aiAnalysisRequestId) {
+        aiAnalysisRequestLogsMapper.insert(AiAnalysisRequestLogs.builder()
+                .aiAnalysisRequestId(aiAnalysisRequestId)
+                .businessRequest(message)
+                .difyAgentKey(flowId)
+                .aiAnalysisRequestType(BusinessTypeEnum.COMMUNITYTARGET.getCode())
+                .build());
+    }
+
     private void processMqSinceType51(MqFormData oldItem, MqMessageRecord curMQMessageRecord) {
         log.info("processMqSinceType51 辨别数据为分析请求 {}", SubSinceTypeEnum.SINCETYPE51.getCode());
         DiffDefeatanAlysis diffDefeatanAlysis = JSON.parseObject(JSON.toJSONString(oldItem.getData()), DiffDefeatanAlysis.class);