SpringBoot+Redis实现分布式WebSocket

在Spring Boot中，你可以使用spring-boot-starter-data-redis依赖来集成Redis，并结合spring-boot-starter-websocket来实现WebSocket。以下是一个简化的例子：

1. 添加依赖到你的pom.xml：

<dependencies>
    <!-- Redis -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-data-redis</artifactId>
    </dependency>
    <!-- WebSocket -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-websocket</artifactId>
    </dependency>
</dependencies>

2. 配置Redis和WebSocket：

@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();
    }
}

3. 创建WebSocket服务：

@Service
public class WebSocketService {
    @Autowired
    private SimpMessagingTemplate simpMessagingTemplate;
 
    public void sendMessageToUser(String user, String message) {
        simpMessagingTemplate.convertAndSendToUser(user, "/queue/messages", message);
    }
}

4. 在你的控制器中使用WebSocket服务：

@Controller
public class WebSocketController {
    @Autowired
    private WebSocketService webSocketService;
 
    @MessageMapping("/chat")
    public void sendMessage(Principal principal, String message) {
        webSocketService.sendMessageToUser(principal.getName(), message);
    }
}

5. 在你的客户端，你可以使用STOMP over SockJS来连接到WebSocket端点：

var socket = new SockJS('/ws');
var stompClient = Stomp.over(socket);
stompClient.connect({}, function(frame) {
    stompClient.subscribe('/user/queue/messages', function(message) {
        // Handle message
    });
});

以上代码提供了一个基本的WebSocket服务，它使用Redis作为消息代理，以此来实现在多个节点上的WebSocket连接的可伸缩性。当用户连接到WebSocket时，他们的消息会被发送到特定用户的队列中，并且只有该用户可以接收这些消息。