【Java】SpringBoot快速整合WebSocket实现客户端服务端相互推送信息

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.*;
 
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
 
    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/topic"); // 客户端订阅地址的前缀信息
        config.setApplicationDestinationPrefixes("/app"); // 客户端发送信息的前缀
    }
 
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS(); // 注册STOMP协议的节点，并映射指定的URL，并指定使用SockJS协议
    }
}

这段代码定义了一个WebSocket的配置类，实现了WebSocketMessageBrokerConfigurer接口，通过@EnableWebSocketMessageBroker注解启用了WebSocket消息代理。在configureMessageBroker方法中，我们定义了消息代理的简单中介功能，并指定了客户端订阅地址的前缀。在registerStompEndpoints方法中，我们注册了一个STOMP协议的端点，并指定了URL，同时配置了支持SockJS协议，以便在不支持WebSocket的环境中也能使用。