diff --git a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpusProcessKafkaConsumer.java b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpusProcessKafkaProducer.java similarity index 96% rename from ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpusProcessKafkaConsumer.java rename to ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpusProcessKafkaProducer.java index 5356efd..1b22bfc 100644 --- a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpusProcessKafkaConsumer.java +++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpusProcessKafkaProducer.java @@ -25,7 +25,7 @@ import java.time.LocalDateTime; @Slf4j @Component @RestController -public class CorpusProcessKafkaConsumer { +public class CorpusProcessKafkaProducer { @Autowired private TmTelephoneCorpusService tmTelephoneCorpusService; @@ -46,6 +46,9 @@ public class CorpusProcessKafkaConsumer { BeanUtils.copyProperties(aicorpusTelephone, tmTelephoneCorpus); tmTelephoneCorpus.setCreateBy("kafka"); tmTelephoneCorpus.setCreateTime(LocalDateTime.now()); + + // TODO 只处理dcc的 + tmTelephoneCorpusService.saveTelephoneCorpus(tmTelephoneCorpus); tmTelephoneCorpusService.runTelephoneCorpusDify(aicorpusTelephone); 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 new file mode 100644 index 0000000..17df53a --- /dev/null +++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mq/CorpushIsLikeConsumer.java @@ -0,0 +1,21 @@ +package com.volvo.ai.analytic.center.mq; + +import lombok.extern.slf4j.Slf4j; +import org.apache.rocketmq.common.message.MessageExt; +import org.apache.rocketmq.spring.annotation.RocketMQMessageListener; +import org.apache.rocketmq.spring.core.RocketMQListener; +import org.springframework.stereotype.Component; + +@Slf4j +@Component +@RocketMQMessageListener(consumerGroup = "${rocketmq.consumer.corpus.isLikeTpoicGroup}", + topic = "${rocketmq.consumer.corpus.isLikeTpoic}", + enableMsgTrace = true) +public class CorpushIsLikeConsumer implements RocketMQListener{ + + + @Override + public void onMessage(MessageExt messageExt) { + log.info("CorpushIsLikeConsumer message: " + messageExt); + } +} diff --git a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/TmTelephoneCorpusServiceImpl.java b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/TmTelephoneCorpusServiceImpl.java index 39d55db..252a898 100644 --- a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/TmTelephoneCorpusServiceImpl.java +++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/TmTelephoneCorpusServiceImpl.java @@ -20,6 +20,7 @@ import com.volvo.ai.analytic.center.mapper.TmTelephoneCorpusMapper; import com.volvo.ai.analytic.center.service.DataMaskingRuleService; import com.volvo.ai.analytic.center.service.DiFyService; import com.volvo.ai.analytic.center.service.TmTelephoneCorpusService; +import com.volvo.ai.analytic.center.utils.ConstantStr; import com.volvo.ai.analytic.center.utils.FlowResultSplitUtil; import lombok.extern.slf4j.Slf4j; import org.apache.commons.collections.CollectionUtils; @@ -28,6 +29,7 @@ import org.apache.rocketmq.client.producer.SendResult; import org.apache.rocketmq.spring.core.RocketMQTemplate; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; +import org.springframework.cloud.context.config.annotation.RefreshScope; import org.springframework.messaging.support.MessageBuilder; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; @@ -42,6 +44,7 @@ import java.util.stream.Collectors; * @author rz * @date 2025-03-04 */ +@RefreshScope @Slf4j @Service public class TmTelephoneCorpusServiceImpl extends ServiceImpl implements TmTelephoneCorpusService { @@ -53,9 +56,10 @@ public class TmTelephoneCorpusServiceImpl extends ServiceImpl inputMap = new HashMap(); DiFyReq diFyImageReq = new DiFyReq(); - diFyImageReq.setUser("11111"); - diFyImageReq.setFlowId("app-peJXSjHjVKdkYxjdOUuPnZ5b"); + diFyImageReq.setUser(ConstantStr.corpus_user); + diFyImageReq.setFlowId(telephoneToken); JSONObject jsonObject = JSONObject.parseObject( aicorpusTelephone.getDisplay());