websocket springboot（手把手讲解）

💡一则或许对你有用的小广告

欢迎加入小哈的星球 ，你将获得：专属的项目实战 / 1v1 提问 / Java 学习路线 / 学习打卡 / 每月赠书 / 社群讨论

• 新项目:《从零手撸：仿小红书（微服务架构）》 正在持续爆肝中，基于 Spring Cloud Alibaba + Spring Boot 3.x + JDK 17...，点击查看项目介绍 ;
• 《从零手撸：前后端分离博客项目（全栈开发）》 2 期已完结，演示链接： http://116.62.199.48/ ;

截止目前， 星球 内专栏累计输出 82w+ 字，讲解图 3441+ 张，还在持续爆肝中.. 后续还会上新更多项目，目标是将 Java 领域典型的项目都整一波，如秒杀系统, 在线商城, IM 即时通讯，权限管理，Spring Cloud Alibaba 微服务等等，已有 2900+ 小伙伴加入学习 ，欢迎点击围观

前言：WebSocket 与 Spring Boot 的技术融合

在互联网技术快速发展的今天，实时通信已成为现代 Web 应用的核心需求。无论是即时聊天、在线游戏、股票行情推送，还是物联网设备的数据交互，传统基于 HTTP 的请求-响应模式都显得力不从心。此时，WebSocket Spring Boot 的组合便成为开发者们解决这一痛点的利器。本文将从技术原理到实战案例，循序渐进地讲解如何利用 Spring Boot 快速构建高性能的 WebSocket 应用，帮助编程初学者和中级开发者掌握这一关键技能。

WebSocket 的核心概念与优势

1.1 什么是 WebSocket？

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，允许服务器主动向客户端推送数据，无需客户端频繁发起请求。这就好比两个人通过一条双向车道对话，而不是每次说话都要先敲门再等待回应。

与 HTTP 的对比

1.2 WebSocket 的工作流程

1. 握手阶段：客户端通过 HTTP 升级请求（Upgrade: websocket）与服务器建立连接。
2. 消息传输：连接建立后，双方可随时发送文本或二进制数据。
3. 关闭连接：任意一方可通过发送关闭帧终止连接。

Spring Boot 集成 WebSocket 的实战步骤

2.1 项目环境搭建

在 Spring Boot 项目中集成 WebSocket 需要以下基础配置：

步骤 1：添加依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

步骤 2：配置 WebSocket 服务

创建配置类启用 WebSocket 支持：

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        // 注册端点并启用SockJS兼容性
        registry.addEndpoint("/ws")
                .setAllowedOriginPatterns("*")
                .withSockJS();
    }

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        // 配置消息代理前缀
        registry.setApplicationDestinationPrefixes("/app")
                .enableSimpleBroker("/topic", "/queue");
    }
}

2.2 核心注解与消息处理

2.2.1 消息接收与发送

• @MessageMapping：类似 @RequestMapping，用于标注消息处理器方法
• SimpMessagingTemplate：发送消息到指定目的地

@Component
public class ChatMessageHandler {
    @Autowired
    private SimpMessagingTemplate messagingTemplate;

    @MessageMapping("/chat")
    public void handleChatMessage(String message, 
                                  @Header("simpSessionId") String sessionId) {
        // 广播消息到/topic/chat
        messagingTemplate.convertAndSend("/topic/chat", 
            "Session " + sessionId + ": " + message);
    }
}

2.2.2 客户端连接管理

通过 WebSocketSession 可以实现用户身份验证、会话状态维护等功能：

@MessageMapping("/connect")
public void onConnect(Principal user, WebSocketSession session) {
    System.out.println("User " + user.getName() + " connected");
    // 可在此处执行登录验证或会话绑定操作
}

典型场景实战：构建实时聊天室

3.1 功能设计

• 用户通过网页连接 WebSocket 端点
• 输入消息后实时发送至服务器
• 服务器将消息广播给所有在线用户

3.2 服务端代码实现

@Controller
public class ChatController {
    @MessageMapping("/send")
    @SendTo("/topic/messages")
    public ChatMessage sendMessage(@Payload ChatMessage message) {
        System.out.println("Received message: " + message.getContent());
        return message;
    }
}

@Data
@AllArgsConstructor
@NoArgsConstructor
public class ChatMessage {
    private String content;
    private String sender;
}

3.3 前端 JavaScript 实现

const socket = new SockJS('/ws');
const client = Stomp.over(socket);

client.connect({}, function(frame) {
    console.log('Connected: ' + frame);
    client.subscribe('/topic/messages', function(message) {
        const msg = JSON.parse(message.body);
        appendMessage(msg.sender + ": " + msg.content);
    });
});

function sendMessage() {
    const input = document.getElementById('messageInput').value;
    client.send("/app/chat/send", {}, JSON.stringify({content: input, sender: 'User'}));
}

性能优化与高级特性

4.1 消息压缩与传输优化

@Configuration
public class WebSocketCompressionConfig implements WebSocketMessageBrokerConfigurer {
    @Override
    public void configureWebSocketTransport(WebSocketTransportRegistration registry) {
        registry.addDecoratorFactory(new WebSocketMessageDecoratorFactory() {
            @Override
            public WebSocketMessage<?> decorateMessage(WebSocketMessage<?> message) {
                if (message instanceof TextMessage) {
                    return new TextMessage(((TextMessage) message).getPayload().replaceAll("\\s+", ""));
                }
                return message;
            }
        });
    }
}

4.2 心跳检测与连接保活

spring.websocket.send-timeout=30000
spring.websocket.client.timeout=30000
spring.websocket.heartbeat-interval=10000

4.3 消息订阅权限控制

@Configuration
@EnableWebSocketMessageBroker
public class SecureWebSocketConfig implements WebSocketMessageBrokerConfigurer {
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/secure-ws")
                .withSockJS()
                .setHandshakeHandler(new DefaultHandshakeHandler() {
                    @Override
                    protected boolean allowedRequestOrigin(ServerHttpRequest request) {
                        return request.getHeaders().containsKey("Authorization");
                    }
                });
    }
}

常见问题与解决方案

5.1 跨域连接问题

在 registerStompEndpoints 方法中设置：

registry.addEndpoint("/ws")
        .setAllowedOriginPatterns("/**") // 允许所有域名
        .withSockJS();

5.2 消息丢失排查

1. 检查消息目的地路径是否正确
2. 确保消息代理配置生效（如 SimpleBroker）
3. 使用 @SendToUser 实现单用户消息

5.3 性能监控

@Configuration
public class WebSocketMetricsConfig {
    @Bean
    public ChannelInterceptor messageCounter() {
        return new ChannelInterceptor() {
            @Override
            public Message<?> preSend(Message<?> message, MessageChannel channel) {
                System.out.println("Message count: " + message.getHeaders().get("counter"));
                return message;
            }
        };
    }
}

结论：构建未来实时应用的基石

通过本文的讲解，读者应已掌握 WebSocket Spring Boot 的核心原理与实战技能。从基础配置到高级优化，我们构建了完整的实时通信解决方案。随着物联网、元宇宙等新兴领域的崛起，WebSocket 技术的应用场景将更加广泛。建议开发者在实践中持续探索以下方向：

• 结合 Spring Security 实现细粒度权限控制
• 与消息中间件（如 RabbitMQ）集成实现分布式 WebSocket
• 应用消息分片技术处理大数据量传输

掌握这一技术栈，开发者不仅能够应对当前的实时通信需求，更能为下一代实时交互应用奠定技术基础。