redis 的发布订阅解决分布式下的websocket session 共享问题

1.项目背景 :使用websocket 和前端通信但是后端是分布式部署,session 无法共享,并且websocketSession 无法序列化,所以无法把session存到redis 中。

2.基础代码:

java 复制代码
@Configuration
@Slf4j
public class RedisConfig {

    @Bean
    public RedisMessageListenerContainer container(RedisConnectionFactory connectionFactory,
                                                   MessageListenerAdapter listenerAdapter) {
        RedisMessageListenerContainer container = new RedisMessageListenerContainer();
        container.setConnectionFactory(connectionFactory);
        container.addMessageListener(listenerAdapter, new PatternTopic(Constants.REDIS_CHANNEL));
        return container;
    }

    @Bean
    public MessageListenerAdapter listenerAdapter(RedisReceiver receiver) {
        return new MessageListenerAdapter(receiver, "onMessage");
    }


}
java 复制代码
@Slf4j
@Component
public class RedisReceiver implements MessageListener {

    @Autowired
    private TestHandler testHandler;

    @Override
    public void onMessage(Message message, byte[] pattern) {
        try {
            log.info("RedisReceiver onMessage:{}",new String(message.getBody(), StandardCharsets.UTF_8));
            String channel = new String(message.getChannel());
            String content = new String(message.getBody(), StandardCharsets.UTF_8);
            if (Constants.REDIS_CHANNEL.endsWith(channel)) {
                testHandler.sendMessageToClient(content);
            }
        } catch (Exception e) {
            log.error("RedisReceiver 消息处理失败",e);
        }
    }

}
java 复制代码
@Slf4j
@Component
public class TestHandler extends TextWebSocketHandler {

    private static final ConcurrentHashMap<String, WebSocketSession> SESSION_DISPATCH_HOLDER = new ConcurrentHashMap<>();


    @Resource
    private StringRedisTemplate stringRedisTemplate;



    @Override
    public void afterConnectionEstablished(@NonNull WebSocketSession session) throws Exception {
        super.afterConnectionEstablished(session);
        SESSION_DISPATCH_HOLDER.put(session.getId(), session);
        log.info("webSocket already connected:{}",session.getId());
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        super.afterConnectionClosed(session, status);
        SESSION_DISPATCH_HOLDER.remove(session.getId());
        log.info("webSocket disConnect:{}",session.getId());
    }

    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) {
        log.info("handleMessage enter message:{}", JsonUtils.objToJsonStr(message.getPayload()));
        TextMessage textMessage = new TextMessage("");
        try {
            session.sendMessage(textMessage);
        } catch (IOException e) {
            log.error("发送消息失败", e);
        }
    }

 

  
    public void sendMessageToClient(String contentList) {
        log.info("sendMessageToClient:{}", contentList);
        if (CollectionUtils.isEmpty(SESSION_DISPATCH_HOLDER)) {
            log.info("没有连接中的客户端");
            return;
        }
        SESSION_DISPATCH_HOLDER.values().parallelStream().forEach(session -> {
            try {
                TextMessage textMessage = new TextMessage(contentList);
                session.sendMessage(textMessage);
                log.info("websocket 消息推送成功:{}", contentList);
            } catch (Exception e) {
                log.error("发送消息失败", e);
            }
        });
    }



}
相关推荐
ANYOLY5 小时前
Redis 面试宝典
数据库·redis·面试
洲覆6 小时前
Redis 核心数据类型:从命令、结构到实战应用
服务器·数据库·redis·缓存
山海不说话7 小时前
Java后端面经(八股——Redis)
java·开发语言·redis
Insist7538 小时前
基于OpenEuler部署kafka消息队列
分布式·docker·kafka
muxin-始终如一8 小时前
MySQL与Redis面试问题详解
数据库·redis·mysql
在未来等你9 小时前
Elasticsearch面试精讲 Day 20:集群监控与性能评估
大数据·分布式·elasticsearch·搜索引擎·面试
luckys.one10 小时前
第12篇|[特殊字符] Freqtrade 交易所接入全解:API、WebSocket、限频配置详解
网络·ide·python·websocket·网络协议·flask·流量运营
沧澜sincerely10 小时前
Redis 键空间 & 五大类型
java·redis·docker
励志成为糕手11 小时前
Kafka选举机制深度解析:分布式系统中的民主与效率
分布式·kafka·linq·controller·isr机制
不剪发的Tony老师13 小时前
RedisFront:一款免费开源的跨平台Redis客户端工具
数据库·redis·redisfront