Java上的WebSocket：高级实时应用

Java中的WebSocket：创建高级实时应用程序

现代Web应用程序越来越需要客户端和服务器之间的即时数据传输。基于请求-响应的传统HTTP对于延迟至关重要的任务（如在线聊天、股票报价、协作编辑文档或游戏服务器）效率低下。WebSocket协议通过建立持久的双向连接解决了这个问题。在本文中，我们将探讨Java中超出基础示例范围的WebSocket高级技术。

1. Java中高级WebSocket应用程序的架构

使用@ServerEndpoint注解的Java WebSocket基础示例适合学习，但对于生产级解决方案，需要更可靠的架构。让我们探讨关键组件。

1.1. 技术栈选择

对于企业级应用程序，事实上的标准是Spring Boot及其模块spring-boot-starter-websocket。它提供了与STOMP（简单文本导向消息协议）的集成——这是WebSocket之上的一个子协议，增加了消息路由、队列和订阅功能。

1.2. Spring Boot中的WebSocket基本配置

从配置类开始：

@Configuration@EnableWebSocketMessageBrokerpublic class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override    public void configureMessageBroker(MessageBrokerRegistry config) {        // 用于/topic和/queue队列的内置代理        config.enableSimpleBroker("/topic", "/queue");        // 从客户端到服务器的消息前缀        config.setApplicationDestinationPrefixes("/app");        // 用户队列的前缀（点对点）        config.setUserDestinationPrefix("/user");    }

    @Override    public void registerStompEndpoints(StompEndpointRegistry registry) {        // 通过SockJS连接的端点（旧版浏览器的回退方案）        registry.addEndpoint("/ws")                .setAllowedOrigins("*")                .withSockJS();    }}

这段代码配置了连接点、路由前缀和内置消息代理。使用STOMP大大简化了订阅和队列的处理。

2. 高级消息处理：从聊天到股票数据

现在让我们实现具体的场景。考虑三种典型用例：群组聊天、个人通知和数据流。

2.1. 带消息历史的群组聊天

简单的聊天是将消息广播到主题。让我们增加难度：添加历史记录保存和按房间过滤。

@Controllerpublic class ChatController {

    @Autowired    private ChatMessageRepository repository; // JPA仓库

    @MessageMapping("/chat.sendMessage")    @SendTo("/topic/chat/{roomId}")    public ChatMessage sendMessage(@Payload ChatMessage message,                                   @DestinationVariable String roomId,                                   SimpMessageHeaderAccessor headerAccessor) {        // 保存到数据库        message.setTimestamp(Instant.now());        repository.save(message);        // 从会话中添加发送者信息        String username = (String) headerAccessor.getSessionAttributes().get("username");        message.setSender(username);        return message;    }

    @MessageMapping("/chat.addUser")    @SendTo("/topic/chat/{roomId}")    public ChatMessage addUser(@Payload ChatMessage message,                               @DestinationVariable String roomId,                               SimpMessageHeaderAccessor headerAccessor) {        headerAccessor.getSessionAttributes().put("username", message.getSender());        return message;    }}

这里使用了