SpringBoot + WebSocket实战：从零手搓一个能实时收发消息和好友申请的聊天室（附完整源码）

SpringBoot WebSocket实战构建高可用实时聊天系统的架构设计与实现1. 现代实时通信系统的技术选型在当今互联网应用中实时交互功能已成为标配需求。传统HTTP协议的请求-响应模式难以满足即时消息、在线协作等场景而WebSocket协议凭借其全双工通信特性成为实时通信的首选方案。SpringBoot作为Java生态中最流行的微服务框架与WebSocket的结合能够快速构建高性能的实时应用系统。关键技术对比分析技术方案延迟性资源消耗开发复杂度适用场景HTTP轮询高高低简单通知Long Polling中中中低频更新Server-Sent Events中低中服务端单向推送WebSocket低低高实时双向通信提示选择WebSocket时需考虑浏览器兼容性现代主流浏览器均支持WebSocket协议但对于老旧系统可能需要降级方案2. 系统架构设计与核心组件2.1 整体架构分层本系统采用经典的三层架构设计各层职责明确表现层处理HTTP/WebSocket协议转换实现前后端数据交互业务逻辑层核心业务处理包括消息路由、用户管理等数据访问层负责数据持久化采用MySQL关系型数据库// WebSocket配置示例 Configuration EnableWebSocket public class WebSocketConfig implements WebSocketConfigurer { Autowired private MessageHandler messageHandler; Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { registry.addHandler(messageHandler, /ws) .setAllowedOrigins(*) .addInterceptors(new HttpSessionHandshakeInterceptor()); } }2.2 核心业务模型设计系统主要包含以下核心实体及其关系用户(User)系统使用者包含基础身份信息会话(Session)消息交互的上下文环境消息(Message)通信内容载体好友关系(Friend)用户间的社交关联-- 核心表结构设计 CREATE TABLE message ( id bigint NOT NULL AUTO_INCREMENT, sender_id bigint NOT NULL, session_id bigint NOT NULL, content text CHARACTER SET utf8mb4, created_at datetime NOT NULL DEFAULT CURRENT_TIMESTAMP, PRIMARY KEY (id), KEY idx_session (session_id,created_at) ) ENGINEInnoDB DEFAULT CHARSETutf8mb4;3. WebSocket深度集成与优化3.1 连接生命周期管理WebSocket连接需要处理完整的生命周期事件连接建立用户认证与会话初始化消息处理业务逻辑分发与执行异常处理网络波动与业务异常连接关闭资源清理与状态更新Component public class CustomWebSocketHandler extends TextWebSocketHandler { Override public void afterConnectionEstablished(WebSocketSession session) { // 连接建立逻辑 } Override protected void handleTextMessage(WebSocketSession session, TextMessage message) { // 消息处理逻辑 } Override public void handleTransportError(WebSocketSession session, Throwable exception) { // 异常处理逻辑 } Override public void afterConnectionClosed(WebSocketSession session, CloseStatus status) { // 连接关闭逻辑 } }3.2 性能优化策略连接保活定时心跳检测维持连接消息压缩对大型消息体进行压缩传输批量处理合并短时间内的多次操作异步处理耗时操作放入线程池执行// 心跳检测实现示例 Scheduled(fixedRate 30000) public void checkAliveConnections() { onlineUsers.keySet().forEach(userId - { WebSocketSession session onlineUsers.get(userId); try { session.sendMessage(new PingMessage()); } catch (IOException e) { cleanupSession(userId, session); } }); }4. 关键业务场景实现4.1 实时消息收发消息处理流程包含以下核心步骤消息内容校验与敏感词过滤消息持久化存储会话参与者查找消息实时推送未读消息计数更新public void handleMessage(Message message) { // 1. 内容校验 if (!messageValidator.validate(message)) { throw new InvalidMessageException(); } // 2. 持久化存储 messageRepository.save(message); // 3. 查找会话参与者 SetLong participantIds sessionService.getParticipantIds(message.getSessionId()); // 4. 实时推送 messagingTemplate.convertAndSendToUsers( participantIds, /queue/messages, new MessageDTO(message) ); // 5. 更新未读计数 unreadCounterService.incrementUnreadCount( message.getSessionId(), message.getSenderId(), participantIds ); }4.2 好友关系管理好友系统实现要点双向关系维护确保好友关系的对称性申请审批流程完整的申请-审批工作流黑名单机制防止骚扰消息关系状态同步实时更新好友状态Transactional public void approveFriendRequest(Long requestId, Long approverId) { FriendRequest request requestRepository.findById(requestId) .orElseThrow(() - new ResourceNotFoundException(Friend request not found)); if (!request.getReceiverId().equals(approverId)) { throw new PermissionDeniedException(); } // 创建双向好友关系 friendRelationRepository.save( new FriendRelation(request.getRequesterId(), request.getReceiverId()) ); friendRelationRepository.save( new FriendRelation(request.getReceiverId(), request.getRequesterId()) ); // 更新请求状态 request.approve(); requestRepository.save(request); // 发送通知 eventPublisher.publishEvent( new FriendRequestApprovedEvent(request) ); }5. 生产环境部署与监控5.1 Linux服务部署方案推荐采用systemd管理SpringBoot应用# /etc/systemd/system/chat.service [Unit] DescriptionChat Application Service Aftersyslog.target [Service] Userappuser ExecStart/usr/bin/java -jar /opt/chat/app.jar SuccessExitStatus143 [Install] WantedBymulti-user.target部署流程构建可执行JARmvn clean package服务器环境准备JDK、MySQL、防火墙配置应用部署与启动systemctl start chat日志配置与管理Logback ELK5.2 监控指标与健康检查核心监控指标包括连接数当前活跃WebSocket连接数量消息吞吐量单位时间内处理的消息数量响应延迟消息从接受到响应的耗时错误率异常请求占比RestController RequestMapping(/metrics) public class MetricsController { Autowired private WebSocketSessionRegistry sessionRegistry; GetMapping(/connections) public MapString, Object getConnectionMetrics() { return Map.of( activeConnections, sessionRegistry.getSessionCount(), userDistribution, sessionRegistry.getUserDistribution() ); } }6. 安全防护与最佳实践6.1 安全防护措施认证鉴权JWT WebSocket握手拦截消息加密TLS传输层加密防篡改消息签名验证限流防护防止连接滥用public class AuthHandshakeInterceptor implements HandshakeInterceptor { Override public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, MapString, Object attributes) { String token extractToken(request); return jwtValidator.validate(token); } // ... }6.2 性能调优经验在实际项目中遇到的性能瓶颈及解决方案连接数瓶颈采用分布式架构分担负载广播消息延迟使用Redis Pub/Sub优化数据库压力引入消息队列异步处理内存泄漏完善Session清理机制// 分布式会话管理示例 public class RedisSessionManager implements SessionManager { Autowired private RedisTemplateString, Object redisTemplate; Override public void registerSession(String userId, WebSocketSession session) { redisTemplate.opsForValue().set( session: userId, session.getId(), Duration.ofMinutes(30) ); } // ... }7. 扩展能力与未来演进系统可扩展方向多端同步移动端与Web端消息同步消息漫游历史消息多设备查看文件传输基于WebRTC的大文件传输智能机器人集成AI对话能力// 多端同步处理示例 public class MultiDeviceSyncHandler { public void handleMessage(Message message) { // 存储消息 messageRepository.save(message); // 查找用户所有活跃设备 SetWebSocketSession sessions sessionManager.getUserSessions(message.getSenderId()); // 同步到所有设备 sessions.forEach(session - { try { session.sendMessage(new TextMessage(message.getContent())); } catch (IOException e) { sessionManager.cleanupSession(session); } }); } }在实现实时聊天系统时特别需要注意异常场景的处理。比如网络抖动导致的连接中断需要完善重连机制消息投递的可靠性保证需要实现消息确认机制分布式环境下的状态同步需要设计合理的一致性方案。这些都是在实际开发中积累的宝贵经验。