websocket客户端实现（java）

小鱼儿 2023-10-09 17:22 40阅读 0赞

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();
        }
    }