Spring Boot WebSocket客户端代码示例

简介：
本示例展示如何在Spring Boot应用程序中使用WebSocket进行实时通信。通过简洁的代码实现客户端连接、消息发送与接收功能。 Spring Boot WebSocket客户端的代码示例可以帮助开发者快速集成WebSocket功能到基于Spring Boot的应用程序中。下面是一个简单的例子来展示如何使用`spring-boot-starter-websocket`依赖项创建一个基本的WebSocket客户端。 首先，确保在项目中的pom.xml文件或build.gradle文件里添加了相应的WebSockets启动器作为依赖： ```xml org.springframework.boot spring-boot-starter-websocket ``` 接下来，在Spring Boot应用中配置WebSocket客户端。这通常涉及到创建一个`WebSocketHandler`来处理接收到的消息，以及使用`SimpMessagingTemplate`发送消息。 ```java import org.springframework.web.socket.client.WebSocketClient; import org.springframework.web.socket.client.standard.StandardWebSocketClient; import org.springframework.web.socket.sockjs.client.RestTemplateXhrTransport; import org.springframework.web.socket.sockjs.client.SockJsClient; import org.springframework.web.socket.sockjs.client.Transport; public class WebSocketConfig { public WebSocketClient getWebSocketClient() { List transports = new ArrayList<>(); transports.add(new RestTemplateXhrTransport()); SockJsClient sockJsClient = new SockJsClient(transports); return sockJsClient; } } ``` 然后创建一个自定义的`WebSocketHandler`来处理消息： ```java import org.springframework.web.socket.CloseStatus; import org.springframework.web.socket.TextMessage; import org.springframework.web.socket.WebSocketSession; public class MyWebSocketHandler implements WebSocketHandler { @Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { // 在连接建立后执行的代码，例如发送欢迎消息。 } @Override public void handleMessage(WebSocketSession session, Message message) throws Exception { if (message instanceof TextMessage) { String payload = ((TextMessage) message).getPayload(); System.out.println(Received: + payload); // 处理接收到的消息，例如更新UI或调用服务。 } } @Override public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception { // 错误处理逻辑。 } @Override public boolean supportsPartialMessages() { return false; } } ``` 最后，在应用启动类或配置类中使用`MyWebSocketHandler`来初始化客户端连接： ```java import org.springframework.context.annotation.Bean; public class ApplicationConfig { private final WebSocketClient webSocketClient = new StandardWebSocketClient(); @Bean public MyWebSocketHandler myWebSocketHandler() { return new MyWebSocketHandler(); } @Bean public void connectToServer(MyWebSocketHandler handler) throws Exception { this.webSocketClient.doHandshake(handler, ws://localhost:8080/websocket).get(1, TimeUnit.SECONDS); } } ``` 以上代码示例展示了如何使用Spring Boot的WebSockets客户端库来建立一个简单的连接，并处理接收到的消息。