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 f6cecbd..8b6726b 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,6 @@ 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.mapper.AiAnalysisRequestLogsMapper; import com.volvo.ai.analytic.center.service.TmAnalysisResultService; @@ -41,19 +39,17 @@ 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; + String aiAnalysisRequestType = ""; + if(execDifyFlow.containsKey("aiAnalysisRequestType")){ + aiAnalysisRequestType = execDifyFlow.getString("aiAnalysisRequestType"); + } + if(StringUtils.isEmpty(aiAnalysisRequestType) && execDifyFlow.containsKey("isLike")){ + aiAnalysisRequestType= analysisRecordId.substring(0,analysisRecordId.indexOf("-")); } tmAnalysisResultService.saveTmCorpusReport(TmAnalysisResult.builder() .aiAnalysisRequestId(execDifyFlow.getString("analysisRecordId")) .analysisResult(execDifyFlow.toJSONString()) - .analysisType(oldAiAnalysisRequestLogs.getAiAnalysisRequestType()) - .updateBy("LTO") + .analysisType(aiAnalysisRequestType) .updateTime(new Date()) .build()); } catch (Exception e) { @@ -61,4 +57,5 @@ public class CorpushIsLikeConsumer implements RocketMQListener{ } } + }