Java WebSocket 原理、实现与核心特性解析

方式 2：Spring WebSocket（Spring Boot 整合，更常用）

适用于 Spring Boot 项目，封装 JSR 356，易整合 Spring 生态（如 Spring Security），开发效率更高。

第一步：添加依赖（pom.xml）

<dependencies>
    <!-- Spring Boot Web 核心依赖 -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <!-- Spring WebSocket 依赖 -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-websocket</artifactId>
    </dependency>
</dependencies>

第二步：编写配置类

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

@Configuration
@EnableWebSocket // 开启 WebSocket 支持
public class WebSocketConfig implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        // 注册处理器，指定端点地址，允许跨域（生产环境需指定具体域名）
        registry.addHandler(new MyWebSocketHandler(), "/ws/demo").setAllowedOrigins("*");
    }
}

第三步：编写处理器

import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

public class MyWebSocketHandler extends TextWebSocketHandler {
    // 连接建立时触发
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        System.out.println("客户端连接成功，会话 ID：" + session.getId());
        // 服务端主动推送欢迎消息（实时推送体现）
        session.sendMessage(new TextMessage("欢迎连接 Spring WebSocket！"));
    }

    // 收到客户端文本消息时触发
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        String msg = message.getPayload();
        System.out.println("收到客户端消息：" + msg);
        // 服务端实时响应
        session.sendMessage(new TextMessage("Spring 服务端已收到：" + msg));
    }
}

四、核心区别：WebSocket vs 普通 HTTP 通信

WebSocket 与 HTTP 并非替代关系，而是互补关系，核心区别汇总如下（清晰易懂，快速选型）：

五、底层拆解：WebSocket 四大核心特性如何实现？

结合前面的原理和实现，拆解 WebSocket 四大核心特性的底层逻辑，搞懂'它为什么能做到'：

1. 双向对话（全双工）：基于 TCP 特性，打破 HTTP 约束

底层基础：TCP 连接本身就是全双工（类似双向车道），HTTP 的局限是协议层加了'请求 - 响应'单向规则；

WebSocket 突破：握手升级后，抛弃 HTTP 的'请求 - 响应'规则，直接利用 TCP 的双向字节流，双方可随时主动发数据，无需等待对方。

2. 长期占用连接（长连接）：协议不主动断开 + 心跳保活

协议层：WebSocket 无'请求完成即断开'规则，只要不主动发 Close 帧，TCP 连接就持续保留；

应用层：通过 Ping/Pong 心跳帧，防止中间网络设备断开'长时间无数据'的连接，确保连接稳定。

3. 提升传输效率：极简帧结构 + 无重复开销

帧结构优化：用 2-10 字节极简帧头，替代 HTTP 大体积请求头；

无重复开销：仅握手时携带一次 HTTP 头，后续传输只有帧头 + 业务数据；

额外优化：支持二进制传输（无需 Base64 编码，减少 33% 体积）、数据分片（避免大消息阻塞）。

4. 原生实时通讯：服务端持有连接 + 主动推送

核心前提：服务器需存储 WebSocket Session（连接会话），用于定位在线客户端，具体存储逻辑：

• 存储内容：Session 对象（会话 ID、客户端地址、连接状态、通信通道等）；
• 存储方式：线程安全集合（如 ConcurrentHashMap），键为会话 ID，值为 Session 对象，适配多线程场景；
• 生命周期：客户端连接建立时添加，关闭/失效时移除，避免内存泄漏。

实现逻辑：服务端产生新数据后，检测目标客户端 Session 是否在线，在线则直接推送；离线则暂存消息，待重连后推送，客户端通过 onmessage 事件实时接收并渲染。

六、进阶拓展：离线消息存储与跨服务器通讯

实际开发中，常会遇到'一方不在线''跨服务器通讯'的场景，结合数据库等中间件，汇总落地解决方案：

1. 离线消息存储（一方不在线时）

核心逻辑：离线消息与在线 Session 存储独立，以服务端存储为主，分场景选择存储方式，兼顾内存占用与消息可靠性。

• 存储位置：
  • 临时存储（短期非重要消息）：Redis/ConcurrentHashMap，设置过期时间，避免内存堆积；
  • 持久化存储（长期重要消息）：MySQL/MongoDB，持久化到磁盘，基本不占用服务端内存。
• 内存占用说明：
  • 内存缓存（Redis/HashMap）会占用内存，需设置过期时间和消息上限；
  • 数据库存储仅查询时占用少量内存，核心占用磁盘空间，不影响服务端运行。
• 落地方案：
  • 方案 1（临时消息）：Redis+ 过期清理，客户端重连后推送并删除缓存；
  • 方案 2（持久化消息）：数据库+Redis 辅助，近期消息缓存提速，远期消息查数据库；
  • 方案 3（高并发）：消息队列 + 数据库/Redis，缓冲消息，异步写入，避免高并发堵塞。
• 注意点：设置消息保留期限和未读上限，定期清理过期/已读消息，减少存储占用。

2. 跨服务器通讯（引入数据库后）

核心痛点：跨服务器 Session 不共享、消息不同步，基于'数据库 + 中间件'实现拓展，分业务规模选择方案：

• 方案 1（小型分布式）：数据库共享+Session 同步
  • 核心：多台服务器共用数据库，Session 信息同步存入数据库/Redis，通过 HTTP 接口转发消息；
  • 适用：2-3 台服务器，千级以下客户端，开发成本低。
• 方案 2（中型分布式）：数据库+Redis 订阅发布
  • 核心：Redis 存储 Session+ 订阅发布消息，实时路由，数据库持久化离线消息；
  • 适用：3-10 台服务器，千级 - 万级客户端，兼顾性能与可靠性。
• 方案 3（大型高并发）：数据库 + 消息队列 + 分布式框架
  • 核心：消息队列缓冲消息，分布式 WebSocket 框架（Netty-websocket 等）管理 Session 和路由；
  • 适用：10 台以上服务器，万级以上客户端，如直播、大型聊天平台。

核心注意事项：确保数据一致性（事务/Redis 原子操作）、避免消息重复推送（唯一标识）、优化性能（减少数据库查询）、增加容错重试机制。

七、总结与选型建议

WebSocket 的核心价值，是在现有 TCP/IP 和 HTTP 基础上，通过'协议升级'和'极简设计'，实现高效、实时的双向通信。它没有创造新的通信能力，而是把 TCP 的特性用到极致，同时砍掉 HTTP 的冗余开销，成为实时通信场景的首选。

选型建议（快速匹配业务场景）：

• 优先用 HTTP：低频、非实时、一次性请求（用户登录、表单提交、网页加载、文件下载）；
• 优先用 WebSocket：高频、实时、双向交互（在线聊天、客服对话、直播弹幕、股票行情推送、物联网数据上报、多人协作编辑）。