admin/vlv-ai-analytic-center
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改ai回调mq处理

Browse Source
This commit is contained in:
zren25
2025-04-02 18:39:35 +08:00
parent 1cef8c6b5d
commit f52e94aa5c
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpushIsLikeConsumer.java
View File

@@ -1,8 +1,11 @@
package com.volvo.ai.analytic.center.mq; package com.volvo.ai.analytic.center.mq;
import com.alibaba.fastjson.JSONObject; 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.entity.TmAnalysisResult;
import com.volvo.ai.analytic.center.enums.BusinessTypeEnum; 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 com.volvo.ai.analytic.center.service.TmAnalysisResultService;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
@@ -23,6 +26,9 @@ public class CorpushIsLikeConsumer implements RocketMQListener<MessageExt>{
@Autowired @Autowired
private TmAnalysisResultService tmAnalysisResultService; private TmAnalysisResultService tmAnalysisResultService;
@Autowired
private AiAnalysisRequestLogsMapper aiAnalysisRequestLogsMapper;
@Override @Override
public void onMessage(MessageExt messageExt) { public void onMessage(MessageExt messageExt) {
@@ -36,10 +42,18 @@ public class CorpushIsLikeConsumer implements RocketMQListener<MessageExt>{
log.info(" 回调的aiAnalysisRequestId 为空：{} ", execDifyFlow); log.info(" 回调的aiAnalysisRequestId 为空：{} ", execDifyFlow);
return; 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() tmAnalysisResultService.saveTmCorpusReport(TmAnalysisResult.builder()
.aiAnalysisRequestId(execDifyFlow.getString("analysisRecordId")) .aiAnalysisRequestId(execDifyFlow.getString("analysisRecordId"))
.analysisResult(execDifyFlow.toJSONString()) .analysisResult(execDifyFlow.toJSONString())
.analysisType(BusinessTypeEnum.SMART_ASSISTANT.getCode()) .analysisType(oldAiAnalysisRequestLogs.getAiAnalysisRequestType())
.updateBy("LTO") .updateBy("LTO")
.updateTime(new Date()) .updateTime(new Date())
.build()); .build());