websocket客户端实现（java）

import java.net.URI;
import java.util.HashMap;
import java.util.Map;
import javax.websocket.ClientEndpoint;
import javax.websocket.ContainerProvider;
import javax.websocket.OnMessage;
import javax.websocket.Session;
import javax.websocket.WebSocketContainer;
import javax.websocket.ClientEndpointConfig;
import javax.websocket.HandshakeResponse;

@ClientEndpoint
public class WebSocketClient {
    private Session session;

    public WebSocketClient(String uri, Map<String, String> headers) {
        try {
            WebSocketContainer container = ContainerProvider.getWebSocketContainer();
            ClientEndpointConfig.Configurator configurator = new ClientEndpointConfig.Configurator() {
                @Override
                public void beforeRequest(Map<String, List<String>> headers) {
                    for (Map.Entry<String, String> entry : headers.entrySet()) {
                        headers.put(entry.getKey(), Arrays.asList(entry.getValue()));
                    }
                }

                @Override
                public void afterResponse(HandshakeResponse response) {
                    // 可以在这里获取响应头
                }
            };
            ClientEndpointConfig config = ClientEndpointConfig.Builder.create()
                    .configurator(configurator)
                    .build();
            config.getUserProperties().put(ClientEndpointConfig.HEADERS, headers);
            container.connectToServer(this, config, new URI(uri));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @OnMessage
    public void onMessage(String message) {
        // 处理收到的消息
    }

    public void sendMessage(String message) {
        session.getAsyncRemote().sendText(message);
    }

    public void close() {
        try {
            session.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

其中，headers 参数是一个键值对，表示需要设置的请求头。在构造函数中，我们首先创建了一个 ClientEndpointConfig.Configurator 对象，重写了其中的 beforeRequest() 方法，用于在请求之前设置请求头。然后，我们使用 ClientEndpointConfig.Builder.create() 方法创建一个 ClientEndpointConfig 对象，并将 headers 设置为 UserProperties 的一个属性。最后，我们调用 container.connectToServer() 方法时，将 ClientEndpointConfig 对象作为参数传入，用于配置 WebSocket 客户端连接时的请求头。

需要注意的是，不同的 WebSocket 客户端实现可能需要使用不同的 API 来设置请求头，具体请根据实际情况进行修改。

调用示例如下：

import java.util.HashMap;
import java.util.Map;

public class WebSocketClientDemo {
    public static void main(String[] args) {
        Map<String, String> headers = new HashMap<>();
        headers.put("Authorization", "Bearer 123456");
        WebSocketClient client = new WebSocketClient("ws://localhost:8080/websocket", headers);
        client.sendMessage("Hello, WebSocket!");
        client.close();
    }
}