一、基础配置

依赖引入

<dependency>

<groupId>org.springframework.boot</groupId>

<artifactId>spring-boot-starter-websocket</artifactId>

</dependency>

需同步引入JSON处理库（如Jackson）用于消息序列化。

启用WebSocket支持

@Configuration

@EnableWebSocketMessageBroker

public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

@Override

public void registerStompEndpoints(StompEndpointRegistry registry) {

registry.addEndpoint("/ws").setAllowedOrigins("*").withSockJS();

}

@Override

public void configureMessageBroker(MessageBrokerRegistry registry) {

registry.enableSimpleBroker("/topic");

registry.setApplicationDestinationPrefixes("/app");

}

}

说明：

• /ws为连接端点，/topic为广播前缀，/app为应用消息前缀。 withSockJS()启用降级兼容（如HTTP长轮询）。

二、消息处理

群发消息控制器

@Controller

public class ChatController {

@MessageMapping("/group")

@SendTo("/topic/messages")

public String broadcast(String message) {

return "Server: " + message;

}

}

客户端需订阅/topic/messages接收广播消息。

点对点消息推送

@MessageMapping("/private")

@SendToUser("/queue/notify")

public String privateMsg(String msg, Principal user) {

return user.getName() + ": " + msg;

}

需配合Principal认证，客户端订阅/user/queue/notify。

三、安全与扩展

握手拦截器

public class AuthInterceptor implements HandshakeInterceptor {

@Override

public boolean beforeHandshake(/*...*/) {

String token = request.getParameter("token");

return isValidToken(token); // 自定义校验逻辑

}

}

注册到WebSocketConfig中实现连接认证。

集群部署方案

• 集成Redis的Pub/Sub解决跨节点消息同步问题。 使用Nginx负载均衡时需配置proxy_set_header Upgrade $http_upgrade。

四、前端连接示例

const socket = new SockJS('/ws');

const stompClient = Stomp.over(socket);

stompClient.connect({}, () => {

stompClient.subscribe('/topic/messages', (res) => {

console.log(res.body);

});

});

需引入sockjs-client和stompjs库。

五、常见问题

跨域问题

• 生产环境禁用setAllowedOrigins("*")，改用精确域名白名单。

心跳断开

• 默认10秒无活动断开，可通过@Configuration调整心跳间隔。

序列化异常

• 确保消息类有无参构造器和getter/setter。