From f52e94aa5c28e2b74382e254bdac3cb1a3bc390a Mon Sep 17 00:00:00 2001
From: zren25 <zhen.ren.2@volvocars.com>
Date: Wed, 2 Apr 2025 18:39:35 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9ai=E5=9B=9E=E8=B0=83mq?=
 =?UTF-8?q?=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../center/mq/CorpushIsLikeConsumer.java         | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpushIsLikeConsumer.java b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpushIsLikeConsumer.java
index c9ad5bb..190b597 100644
--- a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpushIsLikeConsumer.java
+++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpushIsLikeConsumer.java
@@ -1,8 +1,11 @@
 package com.volvo.ai.analytic.center.mq;
 
 import com.alibaba.fastjson.JSONObject;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.volvo.ai.analytic.center.entity.AiAnalysisRequestLogs;
 import com.volvo.ai.analytic.center.entity.TmAnalysisResult;
 import com.volvo.ai.analytic.center.enums.BusinessTypeEnum;
+import com.volvo.ai.analytic.center.mapper.AiAnalysisRequestLogsMapper;
 import com.volvo.ai.analytic.center.service.TmAnalysisResultService;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
@@ -23,6 +26,9 @@ public class CorpushIsLikeConsumer implements RocketMQListener<MessageExt>{
 
 	@Autowired
 	private TmAnalysisResultService tmAnalysisResultService;
+
+	@Autowired
+	private AiAnalysisRequestLogsMapper aiAnalysisRequestLogsMapper;
     @Override
     public void onMessage(MessageExt messageExt) {
 
@@ -36,10 +42,18 @@ public class CorpushIsLikeConsumer implements RocketMQListener<MessageExt>{
 				log.info(" 回调的aiAnalysisRequestId 为空：{} ", execDifyFlow);
 				return;
 			}
+			// 根据AIId查询log 获取 BusinessType
+			LambdaQueryWrapper<AiAnalysisRequestLogs> queryWrapper = new LambdaQueryWrapper<>();
+			queryWrapper.eq(AiAnalysisRequestLogs::getAiAnalysisRequestId, analysisRecordId);
+			AiAnalysisRequestLogs oldAiAnalysisRequestLogs= aiAnalysisRequestLogsMapper.selectOne(queryWrapper);
+			if(null == oldAiAnalysisRequestLogs){
+				log.info(" 回调的aiAnalysisResult的AiAnalysisRequestId查询为空：{} ", analysisRecordId);
+				return;
+			}
 			tmAnalysisResultService.saveTmCorpusReport(TmAnalysisResult.builder()
 					.aiAnalysisRequestId(execDifyFlow.getString("analysisRecordId"))
 					.analysisResult(execDifyFlow.toJSONString())
-					.analysisType(BusinessTypeEnum.SMART_ASSISTANT.getCode())
+					.analysisType(oldAiAnalysisRequestLogs.getAiAnalysisRequestType())
 					.updateBy("LTO")
 					.updateTime(new Date())
 					.build());