基于 Spring Boot 和 WebSocket 的实时聊天室系统实现

系统架构与数据流

本项目最核心的部分是 Spring Boot + WebSocket 实现的实时通信机制。

架构示意图

本项目采用经典的三层架构（Controller/Service/Dao）配合 WebSocket 的通信模型。

核心通信流程：WebSocket

1. 用户通过 HTTP /login 登录成功后，获取会话信息。
2. 用户发起 WebSocket 连接请求到 /WebSocketMessage。
3. 服务器将用户 ID 和对应的 WebSocket 会话 (Session) 绑定存储。
4. 用户 A 发送消息到服务器（通过 WebSocket）。
5. 服务器根据消息中的目标用户 ID 查找对应的 WebSocket Session。
6. 服务器通过目标 Session 将消息实时推送给用户 B。

项目结构概览

清晰的模块化结构，便于开发者理解和维护：

java_chatroom/
├── src/
│   ├── main/
│   │   ├── java/com/example/java_chatroom/
│   │   │   ├── api/ # Controller 层
│   │   │   ├── component/ # 组件类
│   │   │   ├── config/ # 配置类
│   │   │   └── model/ # 数据模型
│   │   ├── resources/
│   │   │   ├── mapper/ # MyBatis XML 映射文件
│   │   │   └── static/ # 前端静态资源 (HTML/CSS/JS)
│   │   └── db.sql # 数据库初始化脚本
├── pom.xml # Maven 依赖配置
└── README.md # 项目说明文档

数据库设计 (MySQL)

系统采用 5 个核心数据表来实现用户关系和消息存储。

关键数据表

核心功能模块与 API

项目主要通过 RESTful API 和 WebSocket 端点实现功能。

认证与用户信息

好友与会话管理

消息服务

环境与运行指南

运行环境要求

• Java Development Kit (JDK): 1.8 或更高版本
• MySQL Server: 5.7 或更高版本
• Maven: 3.6 或更高版本

步骤

1. 初始化数据库:
   • 确保 MySQL 服务运行，并创建一个名为 java_chatroom 的数据库。
   • 打开 src/main/resources/application.yml 文件。
   • 修改 username 和 password 为您的 MySQL 账户信息。
2. 访问应用: 打开浏览器访问：http://localhost:8080/login.html

运行项目:

mvn clean install
mvn spring-boot:run

配置数据库连接:

spring:
  datasource:
    url: jdbc:mysql://127.0.0.1:3306/java_chatroom?...
    username: root
    password: your_password

执行 SQL 脚本：

mysql -u root -p < src/main/java/db.sql

克隆项目:

git clone <repository-url>

功能特性总结

1. 用户认证体系:
   • 支持用户注册新账号。
   • 实现基于会话的登录校验。
2. 高效实时通信:
   • 利用 WebSocket 实现消息的毫秒级推送。
   • 支持一对一私密聊天。
3. 完善的消息与会话管理:
   • 自动管理私聊会话的创建与激活。
   • 持久化存储消息记录，支持查看历史消息。
4. 基础的好友关系:
   • 展示当前用户的好友列表。

未来展望

本项目可进一步扩展以实现更丰富的功能：

• 好友请求功能: 实现用户搜索、发送/接受好友请求的完整流程。
• 个性化展示: 用户添加头像，个性签名等。
• 群聊支持: 扩展会话模型，支持多人聊天室和群组管理。
• 消息类型扩展: 支持发送图片、文件和表情包。
• 用户状态管理: 实时显示用户的在线/离线状态和最后活跃时间。
• UI/UX 优化: 引入更现代的前端框架或库，实现响应式设计。