SpringBoot用WebSocket(Stomp)实现单点连接

本文介绍如何使用Spring Boot实现WebSocket的单点连接,即确保每个用户只维持一个有效的WebSocket连接。核心思路是将连接保存到Map中,并根据用户信息进行管理。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

WebSocket在聊天室和在线游戏,以及对数据实时性要求高的Web页面中都有应用

怎么通过SpringBoot创建WebSocket服务的文章很多,

类似于单点登录,一个账号只能在一个地方登录,我们的长连接也经常会需要限制一个用户只能保持一个长连接

这里就讲一下实现单点连接的一种思路

/**
 * WebSocket 配置类
 */
@Configuration
@EnableWebSocketMessageBroker
@Log4j
public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer {
 
    // 用于缓存连接的Map
    private static Map<String,WebSocketSession> userMap=new ConcurrentMap<>();
 
    @Override
    public void registerStompEndpoints(StompEndpointRegistry stompEndpointRegistry) {
        // 客户端与服务器端建立连接的点
        stompEndpointRegistry.addEndpoint("/endpoint").withSockJS();
    }
 
    @Override
    public void configureMessageBroker(MessageBrokerRegistry messageBrokerRegistry) {
        // 配置客户端发送信息的路径的前缀
        messageBrokerRegistry.setApplicationDestinationPrefixes("/chat");
        messageBrokerRegistry.enableSimpleBroker("/chat", "/topic");
    }
 
    @Override
    public void configureWebSocketTransport(final WebSocketTransportRegistration registration) {
        registration.addDecoratorFactory(new WebSocketHandlerDecoratorFactory() {
            @Override
            public WebSocketHandler decorate(final WebSocketHandler handler) {
                return new WebSocketHandlerDecorator(handler) {
                    @Override
                    public void afterConnectionEstablished(final WebSocketSession session) throws Exception {
                        // 读取连接中的用户信息
                        String username = session.getPrincipal().getName();
                        // 查找该用户是否已经存在长连接
                        WebSocketSession oldSessioin=userMap.get(username);
                        if(oldSession!=null){ 
                            // 如果已经存在连接,则断开旧连接
                            oldSession.close();
                            // 也可以保留旧连接,断开新连接,看业务需求
                            // session.close();
                        }
                        userMap.put(username,session);//保存新连接
                        super.afterConnectionEstablished(session);
                    }
 
                    @Override
                    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus)
                            throws Exception {
                        // 
                        String username = session.getPrincipal().getName();
                        userMap.remove(username);//清除该用户下的连接
                        super.afterConnectionClosed(session, closeStatus);
                    }
                };
            }
        });
        super.configureWebSocketTransport(registration);
    }
}

主要的核心就是要把连接保存到一个Map中,后续再根据用户来查找连接,限制保留一个连接,这样就实现了单点连接的需求。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值