diff --git a/src/main/java/io/github/yubincloud/fairywiki/rocketmq/VoteTopicConsumer.java b/src/main/java/io/github/yubincloud/fairywiki/rocketmq/VoteTopicConsumer.java index e542f4b479cdb016a7584f73303fb0209b662d0b..83855e58959a31e041efa88b0784c541e8eb6527 100644 --- a/src/main/java/io/github/yubincloud/fairywiki/rocketmq/VoteTopicConsumer.java +++ b/src/main/java/io/github/yubincloud/fairywiki/rocketmq/VoteTopicConsumer.java @@ -1,5 +1,6 @@ package io.github.yubincloud.fairywiki.rocketmq; +import io.github.yubincloud.fairywiki.websocket.WebSocketServer; import org.apache.rocketmq.common.message.MessageExt; import org.apache.rocketmq.spring.annotation.RocketMQMessageListener; import org.apache.rocketmq.spring.core.RocketMQListener; @@ -7,15 +8,21 @@ import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.stereotype.Service; +import javax.annotation.Resource; + @Service @RocketMQMessageListener(consumerGroup = "default", topic = "VOTE_TOPIC") public class VoteTopicConsumer implements RocketMQListener { private static final Logger LOG = LoggerFactory.getLogger(VoteTopicConsumer.class); + @Resource + private WebSocketServer webSocketServer; + @Override public void onMessage(MessageExt messageExt) { byte[] body = messageExt.getBody(); LOG.info("ROCKETMQ 收到消息:{}", new String(body)); + webSocketServer.sendInfo(new String(body)); } } \ No newline at end of file