Spring Boot + WebSocket：从零到一，手把手教你打造一个能记住用户的在线聊天室（附完整源码）
Spring Boot + WebSocket：构建带用户识别的智能聊天室实战指南
想象一下，你正在开发一个在线教育平台，需要实现师生实时互动功能。当学生A进入虚拟教室时，系统需要自动广播"学生A已上线"；当老师发送私信给学生B时，消息必须精准送达且显示发送者身份。这种场景下，简单的消息收发远远不够——我们需要一个能识别用户身份、管理在线状态的智能聊天系统。
传统WebSocket教程往往止步于基础连接和消息转发，而真实业务场景要求我们解决三个核心问题：如何绑定用户身份？如何维护在线列表？如何实现定向消息推送？本文将手把手带你用Spring Boot和WebSocket实现这些功能，重点突破HttpSession与WebSocket的整合难题。
1. 环境准备与架构设计
1.1 技术栈选型
我们选择以下技术组合：
- Spring Boot 2.7+：简化配置和依赖管理
- Java WebSocket API：标准化的全双工通信实现
- Tomcat嵌入式容器：内置WebSocket支持
- ConcurrentHashMap：线程安全的在线用户管理
<!-- pom.xml关键依赖 --> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency>1.2 系统架构设计
整个系统分为三个核心模块：
| 模块 | 职责 | 实现要点 |
|---|---|---|
| 身份认证层 | 绑定HTTP会话与WebSocket连接 | 自定义Configurator获取HttpSession |
| 连接管理层 | 维护在线用户状态 | 线程安全的ConcurrentHashMap |
| 消息路由层 | 处理广播与私聊逻辑 | @OnMessage方法中的路由判断 |
2. 用户身份绑定实战
2.1 创建Session配置器
关键突破点在于如何将HTTP协议的会话信息传递到WebSocket协议中。我们通过继承ServerEndpointConfig.Configurator实现：
public class HttpSessionConfigurator extends ServerEndpointConfig.Configurator { @Override public void modifyHandshake(ServerEndpointConfig config, HandshakeRequest request, HandshakeResponse response) { HttpSession httpSession = (HttpSession)request.getHttpSession(); config.getUserProperties().put(HttpSession.class.getName(), httpSession); } }注意：这里需要确保WebSocket连接建立前用户已经完成登录，HttpSession中需提前设置user属性
2.2 端点类配置
在WebSocket端点类中应用配置器并声明用户管理容器：
@ServerEndpoint(value = "/chat", configurator = HttpSessionConfigurator.class) @Component public class ChatEndpoint { private static final Map<String, Session> onlineUsers = new ConcurrentHashMap<>(); private HttpSession httpSession; // 其他方法将在后续章节实现 }3. 连接生命周期管理
3.1 连接建立处理
当新连接建立时，我们需要完成三项操作：
- 从HttpSession提取用户信息
- 更新在线用户列表
- 广播上线通知
@OnOpen public void onOpen(Session session, EndpointConfig config) { this.httpSession = (HttpSession) config.getUserProperties() .get(HttpSession.class.getName()); String username = (String) httpSession.getAttribute("user"); onlineUsers.put(username, session); broadcast(SystemMessage.userJoin(username)); }3.2 连接关闭处理
连接断开时需要执行逆向操作：
- 从在线列表移除用户
- 广播下线通知
@OnClose public void onClose(Session session) { String username = (String) httpSession.getAttribute("user"); onlineUsers.remove(username); broadcast(SystemMessage.userLeave(username)); }4. 消息路由与业务逻辑
4.1 消息协议设计
采用JSON格式区分系统消息与用户消息：
// 系统消息格式 { "system": true, "content": "用户A已上线" } // 用户消息格式 { "system": false, "from": "用户A", "to": "用户B", "content": "你好" }4.2 消息分发实现
根据消息类型实现不同的路由逻辑：
@OnMessage public void onMessage(String jsonMessage) { Message message = MessageParser.parse(jsonMessage); if(message.isSystemMessage()) { handleSystemMessage(message); } else { if(message.isPrivate()) { sendPrivateMessage(message); } else { broadcast(message); } } } private void sendPrivateMessage(Message message) { Session targetSession = onlineUsers.get(message.getTo()); if(targetSession != null) { targetSession.getAsyncRemote().sendText(message.toString()); } }5. 高级功能扩展
5.1 心跳检测机制
防止死连接占用资源，添加心跳检测：
// 在端点类中添加 private static final long HEARTBEAT_TIMEOUT = 300000; // 5分钟 @OnOpen public void onOpen(Session session) { session.setMaxIdleTimeout(HEARTBEAT_TIMEOUT); // ...原有逻辑 }5.2 消息持久化方案
结合Spring Data实现消息存储：
@Service public class MessageStorageService { @Autowired private MessageRepository repository; public void save(Message message) { if(!message.isSystemMessage()) { repository.save(MessageEntity.from(message)); } } }6. 性能优化与安全
6.1 并发发送优化
使用异步发送避免阻塞：
private void broadcast(Message message) { onlineUsers.values().parallelStream().forEach(session -> { if(session.isOpen()) { session.getAsyncRemote().sendText(message.toString()); } }); }6.2 安全防护措施
添加基础安全校验：
@OnMessage public void onMessage(String message) { if(!isValidUser(httpSession) || message.length() > 1000) { session.close(new CloseReason(CloseReason.CloseCodes.VIOLATED_POLICY, "Invalid request")); return; } // ...正常处理逻辑 }在实际项目中部署这套系统时，建议配合Nginx做负载均衡，每个实例维护独立的用户列表时需要通过Redis实现状态共享。我曾在一个在线医疗咨询系统中采用这种架构，成功支撑了500+并发用户的实时交流需求。