如何在Java Websocket客户端中添加头部

14

我正在使用 javax.websocket 类在Java中连接到一个WebSocket服务器。

import javax.websocket.DeploymentException;
import javax.websocket.Session;
import javax.websocket.WebSocketContainer;
import java.io.IOException;
import java.net.URI;

public class WSClient {
    private WebSocketContainer webSocketContainer;

    public void sendMessage(URI endpointURI, String message) throws IOException, DeploymentException {
        Session session = webSocketContainer.connectToServer(MyClientEndpoint.class, endpointURI);
        session.getAsyncRemote().sendText(message);
    }
}

在初始的HTTP握手过程中,我想在客户端请求中添加额外的HTTP标头。

这个可行吗?

我知道使用ServerEndpointConfig.Configurator.modifyHandshake可以在服务器端实现此功能。是否有类似的解决方案可以在客户端端实现?

5个回答

14

ClientEndpointConfig.Configurator.beforeRequest(Map<String,List<String>> headers) 可能是可用的。

关于参数 headers 的JavaDoc如下所示:

实现将要发送以启动握手交互的握手请求头的可变映射。

那么,为什么不像下面这样覆盖 beforeRequest 方法呢?

@Override
public void beforeRequest(Map<String,List<String>> headers)
{
    List<String> values = new ArrayList<String>();
    values.add("My Value");

    headers.put("X-My-Custom-Header", values);
}

您可以将 ClientEndpointConfig 传递给 connectToServer(Class<? extends Endpoint> endpointClass, ClientEndpointConfig cec, URI path)


5
public class Config extends ClientEndpointConfig.Configurator{
    @Override
    public void beforeRequest(Map<String, List<String>> headers) {
        headers.put("Pragma", Arrays.asList("no-cache"));
        headers.put("Origin", Arrays.asList("https://www.bcex.ca"));
        headers.put("Accept-Encoding", Arrays.asList("gzip, deflate, br"));
        headers.put("Accept-Language", Arrays.asList("en-US,en;q=0.8,zh-CN;q=0.6,zh;q=0.4"));
        headers.put("User-Agent", Arrays.asList("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36"));
        headers.put("Upgrade", Arrays.asList("websocket"));
        headers.put("Cache-Control", Arrays.asList("no-cache"));
        headers.put("Connection", Arrays.asList("Upgrade"));
        headers.put("Sec-WebSocket-Version", Arrays.asList("13"));
    }

    @Override
    public void afterResponse(HandshakeResponse hr) {
        Map<String, List<String>> headers = hr.getHeaders();
        log.info("headers -> "+headers);
    }
}

2
Builder configBuilder = ClientEndpointConfig.Builder.create();
configBuilder.configurator(new Configurator() {
    @Override
    public void beforeRequest(Map<String, List<String>> headers) {
    headers.put("Cookie", Arrays.asList("JSESSIONID=" + sessionID));
    }
});
ClientEndpointConfig clientConfig = configBuilder.build();
webSocketContainer.connectToServer(MyClientEndpoint.class, clientConfig, new URI(uri));

1

更新于2020年。这是IntelliJ社区提出的方案:

  public WsClient(String uri) throws Exception {

    ClientEndpointConfig.Builder configBuilder = ClientEndpointConfig.Builder.create();

    configBuilder.configurator(new ClientEndpointConfig.Configurator() {
        public void beforeRequest(Map<String, List<String>> headers) {
            headers.put("FriendlyName", Arrays.asList("TakaTurautin"));
        }
    });
    ClientEndpointConfig clientConfig = configBuilder.build();
    WebSocketContainer container = ContainerProvider.getWebSocketContainer();

    container.connectToServer(this, clientConfig, URI.create(uri));

}

0

我尝试了下面类似的代码

webSocketContainer.connectToServer(MyClientEndpoint.class, clientConfig, new URI(uri));

这里的"MyClientEndpoint"继承了Endpoint类(按照方法规范应该是这样的)。 但是,在Endpoint类中没有可用的@OnMessage事件监听器。如果我简单地使用webSocketContainer.connectToServer(this, new URI(uri))而不进行客户端配置,我会遇到授权问题。请指教。


1
请不要在另一个问题的答案下面提问。 - Garet Jax

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接