diff --git a/src/main/java/com/yupi/springbootinit/config/RabbitMQConfig.java b/src/main/java/com/yupi/springbootinit/config/RabbitMQConfig.java index 370afb0..e5e29cb 100644 --- a/src/main/java/com/yupi/springbootinit/config/RabbitMQConfig.java +++ b/src/main/java/com/yupi/springbootinit/config/RabbitMQConfig.java @@ -24,6 +24,9 @@ public class RabbitMQConfig { public static final String AI_CHAT_EXCHANGE_NAME = "ai.chat.headers.exchange"; public static final String BIG_BROTHER_EXCHANGE_NAME = "big.brother.headers.exchange"; + public static final String WEB_AI_EXCHANGE_NAME = "web.ai.headers.exchange"; + + //创建队列 //true:表示持久化 @@ -63,6 +66,13 @@ public class RabbitMQConfig { .build(); } + @Bean + public HeadersExchange webAiHeadersExchange() { + return ExchangeBuilder.headersExchange(WEB_AI_EXCHANGE_NAME) + .durable(true) + .build(); + } + @Bean public RabbitAdmin rabbitAdmin(ConnectionFactory cf) { diff --git a/src/main/java/com/yupi/springbootinit/rabbitMQ/MQSender.java b/src/main/java/com/yupi/springbootinit/rabbitMQ/MQSender.java index ed8da39..ee0446c 100644 --- a/src/main/java/com/yupi/springbootinit/rabbitMQ/MQSender.java +++ b/src/main/java/com/yupi/springbootinit/rabbitMQ/MQSender.java @@ -62,6 +62,7 @@ public class MQSender { return m; }); } + public void brotherSend(Long tenantId, Object payload) { // 发送消息,把 userId 放进 header rabbitTemplate.convertAndSend(RabbitMQConfig.BIG_BROTHER_EXCHANGE_NAME, "", payload, m -> { @@ -70,5 +71,13 @@ public class MQSender { }); } + public void webAISend(Long tenantId, Object payload) { + // 发送消息,把 userId 放进 header + rabbitTemplate.convertAndSend(RabbitMQConfig.WEB_AI_EXCHANGE_NAME, "", payload, m -> { + m.getMessageProperties().getHeaders().put("tenantId", tenantId); + return m; + }); + } + } \ No newline at end of file diff --git a/src/main/java/com/yupi/springbootinit/service/impl/HostInfoServiceImpl.java b/src/main/java/com/yupi/springbootinit/service/impl/HostInfoServiceImpl.java index 69b1308..0a08cb5 100644 --- a/src/main/java/com/yupi/springbootinit/service/impl/HostInfoServiceImpl.java +++ b/src/main/java/com/yupi/springbootinit/service/impl/HostInfoServiceImpl.java @@ -79,9 +79,16 @@ public class HostInfoServiceImpl extends ServiceImpl i newHosts.forEach(newHost -> { mqSender.send(newHost.getTenantId(),newHost); }); + log.info("发送消息到队列{}, 消息数量: {}", newHosts.get(0).getTenantId(), newHosts.size()); } + if (redisUtils.hasKeyByPrefix("webAI_login:"+ newHosts.get(0).getTenantId())) { + newHosts.forEach(newHost -> { + mqSender.webAISend(newHost.getTenantId(),newHost); + }); + log.info("发送消息到队列WebAI{}, 消息数量: {}", newHosts.get(0).getTenantId(), newHosts.size()); + } stopWatch.stop();