springBoot快速搭建WebSocket
- 添加依赖
在pom.xml中加入WebSocket相关依赖：
<dependencies><!-- websocket --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-websocket</artifactId></dependency><!-- websocket -->
</dependencies>- 创建SocketHandler（我们socket的消息类型有两种：TextMessage 类型用于处理文本帧， BinaryMessage 类型用于处理二进制帧）
创建MyWebSocketHandler处理文本帧
import org.springframework.stereotype.Component;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;@Component
public class MyWebSocketHandler extends TextWebSocketHandler {@Overrideprotected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {String payload = message.getPayload();//接收到的消息System.out.println("Received: " + payload);session.sendMessage(new TextMessage(payload));}
}
创建MyBinaryWebSocketHandler处理二进制帧
import org.springframework.stereotype.Component;
import org.springframework.web.socket.BinaryMessage;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.BinaryWebSocketHandler;import java.nio.ByteBuffer;@Component
public class MyBinaryWebSocketHandler extends BinaryWebSocketHandler {@Overridepublic void handleBinaryMessage(WebSocketSession session, BinaryMessage message) throws Exception {// 从BinaryMessage中获取二进制数据ByteBuffer byteBuffer = message.getPayload();// 这里可以处理接收到的二进制数据while (byteBuffer.hasRemaining()) {byte b = byteBuffer.get();// 处理每一个字节System.out.println("Received byte: " + b);}//发送二进制响应//session.sendMessage(new BinaryMessage(byteBuffer));//或发送文本响应session.sendMessage(new TextMessage("111111"));}
}
- 创建WebSocketConfig
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
public class WebSocketConfig implements WebSocketConfigurer {private final MyWebSocketHandler myWebSocketHandler;private final MyBinaryWebSocketHandler myBinaryWebSocketHandler;public WebSocketConfig(MyWebSocketHandler myWebSocketHandler, MyBinaryWebSocketHandler myBinaryWebSocketHandler) {this.myWebSocketHandler = myWebSocketHandler;this.myBinaryWebSocketHandler = myBinaryWebSocketHandler;}@Overridepublic void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {registry.addHandler(myWebSocketHandler, "/TextMessage").setAllowedOrigins("*");//接收/TextMessage结尾的文本消息registry.addHandler(myBinaryWebSocketHandler, "/BinaryMessage").setAllowedOrigins("*");//接收/BinaryMessage结尾的二进制帧消息}
}
- 创建前端页面index.html 文件，包含一个简单的 WebSocket 客户端
<!DOCTYPE html>
<html lang="en">
<head><meta charset="UTF-8"><title>WebSocket Demo</title>
</head>
<body><h1>WebSocket Demo</h1><button onclick="connect()">Connect</button><button onclick="disconnect()">Disconnect</button><br><input type="text" id="message" placeholder="Enter message"><button onclick="sendMessage()">Send</button><ul id="messages"></ul><script>var webSocket;function connect() {webSocket = new WebSocket('ws://localhost:8080/TextMessage');webSocket.onmessage = function(event) {var messages = document.getElementById('messages');var message = document.createElement('li');message.textContent = event.data;messages.appendChild(message);};webSocket.onopen = function(event) {console.log('Connected to WebSocket');};webSocket.onclose = function(event) {console.log('Disconnected from WebSocket');};}function disconnect() {if (webSocket) {webSocket.close();}}function sendMessage() {var messageInput = document.getElementById('message');var message = messageInput.value;webSocket.send(message);messageInput.value = '';}</script>
</body>
</html>
- 运行应用程序
运行 Spring Boot 应用程序，然后浏览器打开 index.html。你应该会看到一个简单的 WebSocket 客户端界面。
点击 “Connect” 按钮连接到 WebSocket 服务器，输入消息并点击 “Send” 按钮发送消息。你应该会在页面上看到服务器返回的 “Echo” 消息。