單機節(jié)點下，WebSocket連接成功后，可以直接發(fā)送消息。而多節(jié)點下，連接時通過nginx會代理到不同節(jié)點。

假設一開始用戶連接了node1的socket服務。觸發(fā)消息發(fā)送的條件的時候也通過nginx進行代理，假如代理轉到了node2節(jié)點上，那么node2節(jié)點的socket服務就發(fā)送不了消息，因為一開始用戶注冊的是node1節(jié)點。這就導致了消息發(fā)送失敗。

為了解決這一方案，消息發(fā)送時，就需要一個中間件來記錄，這樣，三個節(jié)點都可以獲取消息，然后在根據(jù)條件進行消息推送。

1、依賴

<!-- redis --> <dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-data-redis</artifactId></dependency><!-- websocket --> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId></dependency>

2、創(chuàng)建業(yè)務處理類 Demo.class，該類可以實現(xiàn)MessageListener接口后重寫onMessage方法，也可以不實現(xiàn)，自己寫方法。

import com.alibaba.fastjson.JSON;import com.dy.service.impl.OrdersServiceImpl;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.data.redis.connection.Message;import org.springframework.data.redis.connection.MessageListener;import org.springframework.stereotype.Component; import java.util.HashMap; /** * @program: * @description: redis消息訂閱-業(yè)務處理 * @author: zhang yi * @create: 2021-01-25 16:46 */@Componentpublic class Demo implements MessageListener { Logger logger = LoggerFactory.getLogger(this.getClass()); @Override public void onMessage(Message message, byte[] pattern) { logger.info('消息訂閱成功---------'); logger.info('內容：'+message.getBody()); logger.info('交換機：'+message.getChannel()); }}

3、創(chuàng)建PubSubConfig配置類

import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.cache.annotation.EnableCaching;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.data.redis.connection.RedisConnectionFactory;import org.springframework.data.redis.core.StringRedisTemplate;import org.springframework.data.redis.listener.PatternTopic;import org.springframework.data.redis.listener.RedisMessageListenerContainer;import org.springframework.data.redis.listener.adapter.MessageListenerAdapter; /** * @program: * @description: redis發(fā)布訂閱配置 * @author: zhang yi * @create: 2021-01-25 16:49 */@Configuration@EnableCachingpublic class PubSubConfig { Logger logger = LoggerFactory.getLogger(this.getClass()); //如果是多個交換機，則參數(shù)為(RedisConnectionFactory connectionFactory, // MessageListenerAdapter listenerAdapter, // MessageListenerAdapter listenerAdapter2) @Bean RedisMessageListenerContainer container(RedisConnectionFactory connectionFactory, MessageListenerAdapter listenerAdapter) { RedisMessageListenerContainer container = new RedisMessageListenerContainer(); container.setConnectionFactory(connectionFactory); // 可以添加多個 messageListener，配置不同的交換機 container.addMessageListener(listenerAdapter, new PatternTopic('channel:demo')); //container.addMessageListener(listenerAdapter2, new PatternTopic('channel:demo2')); return container; } /** * 消息監(jiān)聽器適配器，綁定消息處理器，利用反射技術調用消息處理器的業(yè)務方法 * @param demo 第一步的業(yè)務處理類 * @return */ @Bean MessageListenerAdapter listenerAdapter(Demo demo) { logger.info('----------------消息監(jiān)聽器加載成功----------------'); // onMessage 就是方法名，基于反射調用 return new MessageListenerAdapter(demo, 'onMessage'); } /** * 多個交換機就多寫一個 * @param subCheckOrder * @return */ //@Bean //MessageListenerAdapter listenerAdapter2(SubCheckOrder subCheckOrder) { // logger.info('----------------消息監(jiān)聽器加載成功----------------'); // return new MessageListenerAdapter(subCheckOrder, 'onMessage'); //} @Bean StringRedisTemplate template(RedisConnectionFactory connectionFactory) { return new StringRedisTemplate(connectionFactory); }}

4、消息發(fā)布

@Autowiredprivate RedisTemplate<String, Object> redisTemplate; redisTemplate.convertAndSend('channel:demo', '我是內容');三、具體用法 socket連接成功。 socket消息推送時，把信息發(fā)布到redis中。socket服務訂閱redis的消息，訂閱成功后進行推送。集群下的socket都能訂閱到消息，但是只有之前連接成功的節(jié)點能推送成功，其余的無法推送。