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 groupIdorg.springframework.boot/groupId artifactIdspring-boot-starter-websocket/artifactId /dependency1.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 MapString, 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并发用户的实时交流需求。