WebSocket 两个实现方式 (Spring Boot)

在Spring Boot中，有两种常见的WebSocket实现方式：使用Spring的@EnableWebSocketMessageBroker注解和使用WebSocketServletFactory。

1. 使用@EnableWebSocketMessageBroker注解

import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.*;
 
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS();
    }
 
    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        registry.enableSimpleBroker("/topic");
        registry.setApplicationDestinationPrefixes("/app");
    }
}

2. 使用WebSocketServletFactory

import org.eclipse.jetty.websocket.api.WebSocketServlet;
import org.eclipse.jetty.websocket.api.WebSocketServletFactory;
import org.eclipse.jetty.websocket.servlet.WebSocketServletFactory;
 
public class WebSocketEndpoint extends WebSocketServlet {
 
    @Override
    public void configure(WebSocketServletFactory factory) {
        factory.register(MyWebSocket.class);
    }
}

在这个例子中，MyWebSocket是一个继承自WebSocket的类，用于处理WebSocket连接的打开、关闭和消息接收。

注意：第二种方式使用的是org.eclipse.jetty.websocket.api包中的WebSocket API，这是Jetty项目提供的WebSocket实现。如果你使用的是Tomcat作为你的Servlet容器，那么你可能需要使用Spring的@EnableWebSocketMessageBroker注解方式。