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{ @Autowired private TmAnalysisResultService tmAnalysisResultService; + + @Autowired + private AiAnalysisRequestLogsMapper aiAnalysisRequestLogsMapper; @Override public void onMessage(MessageExt messageExt) { @@ -36,10 +42,18 @@ public class CorpushIsLikeConsumer implements RocketMQListener{ log.info(" 回调的aiAnalysisRequestId 为空:{} ", execDifyFlow); return; } + // 根据AIId查询log 获取 BusinessType + LambdaQueryWrapper 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());