在Spring Boot中，你可以使用spring-boot-starter-data-redis依赖来集成Redis，并结合spring-boot-starter-websocket来实现WebSocket。以下是一个简化的例子：
- 添加依赖到你的
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>
- 配置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();
}
}
- 创建WebSocket服务：
@Service
public class WebSocketService {
@Autowired
private SimpMessagingTemplate simpMessagingTemplate;
public void sendMessageToUser(String user, String message) {
simpMessagingTemplate.convertAndSendToUser(user, "/queue/messages", message);
}
}
- 在你的控制器中使用WebSocket服务：
@Controller
public class WebSocketController {
@Autowired
private WebSocketService webSocketService;
@MessageMapping("/chat")
public void sendMessage(Principal principal, String message) {
webSocketService.sendMessageToUser(principal.getName(), message);
}
}
- 在你的客户端，你可以使用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时，他们的消息会被发送到特定用户的队列中，并且只有该用户可以接收这些消息。