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

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中，后续再根据用户来查找连接，限制保留一个连接，这样就实现了单点连接的需求。