From 23a859ac97f793feb8f0a2285ec0c886c0636406 Mon Sep 17 00:00:00 2001 From: zren25 Date: Tue, 8 Apr 2025 19:18:01 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=9B=9E=E8=B0=83=E5=8F=96?= =?UTF-8?q?=E5=80=BC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../center/mq/CorpushIsLikeConsumer.java | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) 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{ } } + }