请用spring boot 写一个websocket客户端

当然，我可以帮你编写一个基于 Spring Boot 的 WebSocket 客户端。首先，你需要创建一个 Spring Boot 项目，并添加相关的依赖。在 pom.xml 文件中添加以下依赖：

<dependencies>
    <!-- Spring Boot WebSocket -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-websocket</artifactId>
    </dependency>
</dependencies>

接下来，创建一个 WebSocket 客户端类，示例如下：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.messaging.converter.MappingJackson2MessageConverter;
import org.springframework.messaging.converter.StringMessageConverter;
import org.springframework.messaging.simp.stomp.*;
import org.springframework.scheduling.concurrent.ConcurrentTaskScheduler;
import org.springframework.stereotype.Component;

import java.util.Scanner;

@SpringBootApplication
public class WebsocketClientApplication {

    public static void main(String[] args) {
        SpringApplication.run(WebsocketClientApplication.class, args);
    }

    @Component
    public static class WebSocketClient implements CommandLineRunner {

        private final StompSessionHandler sessionHandler;

        @Autowired
        public WebSocketClient(StompSessionHandler sessionHandler) {
            this.sessionHandler = sessionHandler;
        }

        @Override
        public void run(String... args) throws Exception {
            StompClient stompClient = new WebSocketStompClient(new SockJsClient(createTransportClient()));
            stompClient.setMessageConverter(new StringMessageConverter());

            StompSession stompSession = stompClient.connect("ws://localhost:8080/ws-endpoint", sessionHandler).get();
            System.out.println("Connected to WebSocket server.");

            Scanner scanner = new Scanner(System.in);
            while (true) {
                String message = scanner.nextLine();
                stompSession.send("/app/send-message", message.getBytes());
            }
        }

        private WebSocketTransport createTransportClient() {
            return new WebSocketTransport(new StandardWebSocketClient());
        }
    }

    @Component
    public static class CustomStompSessionHandler extends StompSessionHandlerAdapter {

        @Override
        public void afterConnected(StompSession session, StompHeaders connectedHeaders) {
            System.out.println("WebSocket session established.");
        }

        @Override
        public void handleTransportError(StompSession session, Throwable exception) {
            System.err.println("WebSocket transport error: " + exception.getMessage());
        }

        @Override
        public Type getPayloadType(StompHeaders headers) {
            return String.class;
        }

        @Override
        public void handleFrame(StompHeaders headers, Object payload) {
            System.out.println("Received message: " + payload);
        }
    }
}

这个示例中，我们创建了一个 CommandLineRunner 类型的 WebSocketClient，用于建立 WebSocket 连接并发送/接收消息。在 WebSocketClient 类的 run 方法中，我们使用 `StompClient` 连接到 WebSocket 服务器，并创建一个 `StompSession` 对象，用于发送消息。同时，我们还创建了一个 `Scanner` 对象，用于从控制台读取用户输入的消息，并通过 `stompSession.send` 方法发送给服务器。

在 CustomStompSessionHandler 类中，我们继承了 `StompSessionHandlerAdapter` 类，并重写了一些方法来处理连接建立、传输错误和收到消息等事件。

请确保在 `stompClient.connect` 方法的参数中，将 `ws://localhost:8080/ws-endpoint` 替换为实际的 WebSocket 服务器端点地址。

这只是一个简单的示例，你可以根据自己的需求进行扩展和修改。希望对你有所帮助！如果你还有其他问题，请随时提问。