Java WebSocket案例

wen java案例 2026-08-03 1

本文目录导读：

Java WebSocket案例

目录导读
WebSocket核心原理与Java生态选型
项目环境搭建与依赖配置
服务端核心代码实现（@ServerEndpoint）
客户端连接与消息收发完整案例
连接管理与心跳检测机制
常见问题排查与性能优化建议
面试高频问答环节

Java WebSocket实战案例：从零搭建实时消息推送系统（附完整代码）

目录导读

WebSocket核心原理与Java生态选型
项目环境搭建与依赖配置
服务端核心代码实现（@ServerEndpoint）
客户端连接与消息收发完整案例
连接管理与心跳检测机制
常见问题排查与性能优化建议
面试高频问答环节

WebSocket核心原理与Java生态选型

WebSocket是一种在单个TCP连接上进行全双工通信的协议,它解决了HTTP协议只能由客户端发起请求的局限性，在Java生态中，主流的实现方案包括JSR 356标准API（javax.websocket）、Spring WebSocket模块以及Netty高性能框架。

为什么选择JSR 356？
对于大多数企业级应用，无需引入额外框架，原生API足够轻量，本文案例基于Servlet 3.1+容器（如Tomcat 9+）实现，兼容性最好。

项目环境搭建与依赖配置

Maven依赖（pom.xml核心部分）：

<dependency>
    <groupId>javax.websocket</groupId>
    <artifactId>javax.websocket-api</artifactId>
    <version>1.1</version>
    <scope>provided</scope>
</dependency>
<dependency>
    <groupId>javax.servlet</groupId>
    <artifactId>javax.servlet-api</artifactId>
    <version>4.0.1</version>
    <scope>provided</scope>
</dependency>

关键点： provided作用域确保容器（Tomcat）提供实现类，避免冲突。

服务端核心代码实现（@ServerEndpoint）

这是一个完整可运行的WebSocket服务端案例，包含连接管理、消息群发、异常处理：

import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.concurrent.CopyOnWriteArraySet;
@ServerEndpoint("/chat")
public class ChatEndpoint {
    // 线程安全的连接集合
    private static final CopyOnWriteArraySet<ChatEndpoint> CONNECTIONS = new CopyOnWriteArraySet<>();
    private Session session;
    @OnOpen
    public void onOpen(Session session) {
        this.session = session;
        CONNECTIONS.add(this);
        broadcast("用户上线，当前在线人数：" + CONNECTIONS.size());
    }
    @OnMessage
    public void onMessage(String message, Session session) {
        broadcast("用户消息：" + message);
    }
    @OnClose
    public void onClose() {
        CONNECTIONS.remove(this);
        broadcast("用户下线，当前在线人数：" + CONNECTIONS.size());
    }
    @OnError
    public void onError(Throwable error) {
        error.printStackTrace();
    }
    private static void broadcast(String msg) {
        for (ChatEndpoint client : CONNECTIONS) {
            try {
                client.session.getBasicRemote().sendText(msg);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

设计要诀： 使用CopyOnWriteArraySet保证高并发下的读写安全，广播方法遍历所有会话发送消息。

客户端连接与消息收发完整案例

HTML5 JavaScript客户端（原生实现）：

let ws = new WebSocket("ws://your-server.com/websocket/chat");
ws.onopen = function() {
    console.log("连接已建立");
    ws.send("你好，服务器！");
};
ws.onmessage = function(event) {
    console.log("收到消息：" + event.data);
    // 实时更新UI
    document.getElementById("messages").innerHTML += `<p>${event.data}</p>`;
};
ws.onclose = function() {
    console.log("连接关闭");
};
// 发送消息函数
function sendMsg() {
    const input = document.getElementById("input");
    ws.send(input.value);
    input.value = "";
}

测试步骤： 启动Tomcat后，打开两个浏览器标签页访问同页面，即可看到双向实时通信效果。

连接管理与心跳检测机制

为什么需要心跳？ 长连接中，代理服务器（如Nginx）会主动断开空闲连接，导致僵尸连接占用资源。

服务端心跳方案（配合定时任务）：

// 定期发送ping包，服务器端只需在容器层处理
// 客户端需实现：
ws.onclose = function() {
    // 自动重连，指数退避算法
    setTimeout(() => { connect(); }, 3000);
};

最佳实践： 建议设置session.setMaxIdleTimeout(30000)，并配合客户端30秒自动重连逻辑。

常见问题排查与性能优化建议

高频问题：

连接失败： 检查ws://和wss://协议是否与页面HTTPS匹配
消息乱码： 服务端统一使用UTF-8编码
内存泄漏： 确保onClose中移除无用引用

性能优化四板斧：

使用消息压缩（如permessage-deflate）
设置合理的最大消息大小（默认8K，可调）
通过异步发送（sendMessage替代sendText）
多节点部署时使用Redis发布订阅实现跨节点广播

面试高频问答环节

Q1：WebSocket和HTTP长轮询有什么区别？
A：WebSocket只需一次握手建立持久连接，服务端可主动推送；长轮询每次请求都携带完整HTTP头，且推延时高于WebSocket，线连接数利用率完全不在一个量级。

Q2：如何处理在线用户列表的一致性？
A：常规做法是维护ConcurrentHashMap映射userId到Session，但集群环境下必须引入分布式存储（如Redis）记录在线状态，并通过广播通知所有节点更新。

Q3：什么是WebSocket的子协议（subprotocol）？
A：子协议是应用层协议标识，如STOMP，Java端可用@ServerEndpoint(value="/chat", subprotocols={"stomp"})声明，用于消息路由和订阅管理。

Q4：如何防御WebSocket的跨站点劫持攻击？
A：校验Origin请求头，拒绝非白名单来源；正确实现鉴权（握手阶段传递token）；使用WSS加密传输。

本文从一个可运行的案例出发，完整讲解了Java WebSocket的编码规范、生命周期管理、生产级优化方向，掌握这些要点后，即可构建出支撑千万级并发的实时交互系统，建议在真实项目中，结合Spring的WebSocketHandler进行更细粒度的消息拦截，或使用Netty获得极致性能。