如何为Spring StandardWebSocketClient配置代理解决连接报错？

好问题！我之前也碰到过类似的场景——Spring的StandardWebSocketClient默认确实没有直接暴露代理配置的入口，但我们可以通过它底层依赖的JDK WebSocket客户端来实现代理设置，和你配置REST代理的思路是相通的。下面分享两种可行的方案：

方案一：通过JDK系统属性全局配置代理 #

StandardWebSocketClient基于JDK自带的WebSocketContainer实现，我们可以通过设置JDK的系统属性来为其配置代理，这种方式简单直接，适合全局代理场景：

// 1. 先配置代理相关的系统属性
// 针对ws://协议的代理
System.setProperty("http.proxyHost", "你的代理IP或域名");
System.setProperty("http.proxyPort", "代理端口");
// 针对wss://协议的代理（如果你的WebSocket是加密的）
System.setProperty("https.proxyHost", "你的代理IP或域名");
System.setProperty("https.proxyPort", "代理端口");

// 如果代理需要用户名密码认证，补充以下属性
System.setProperty("http.proxyUser", "代理用户名");
System.setProperty("http.proxyPassword", "代理密码");
System.setProperty("https.proxyUser", "代理用户名");
System.setProperty("https.proxyPassword", "代理密码");

// 2. 创建配置好代理的WebSocketContainer
WebSocketContainer container = ContainerProvider.getWebSocketContainer();

// 3. 传入StandardWebSocketClient并创建连接管理器
StandardWebSocketClient aStandardWebSocketClient = new StandardWebSocketClient(container);
WebSocketConnectionManager manager = new WebSocketConnectionManager(aStandardWebSocketClient, new WebSockethandler(), aWebSockURL);

方案二：自定义ProxySelector实现局部代理 #

如果你不想让代理设置影响全局其他请求，可以自定义ProxySelector，只针对目标WebSocket URL应用代理规则：

// 1. 实现自定义ProxySelector
ProxySelector customProxySelector = new ProxySelector() {
    @Override
    public List<Proxy> select(URI uri) {
        // 仅对目标WebSocket URL应用代理
        if (aWebSockURL.equals(uri.toString())) {
            Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress("你的代理IP或域名", 代理端口));
            return Collections.singletonList(proxy);
        }
        // 其他URL使用默认代理逻辑
        return ProxySelector.getDefault().select(uri);
    }

    @Override
    public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
        // 可在这里添加代理连接失败的处理逻辑
        ioe.printStackTrace();
    }
};

// 2. 设置全局ProxySelector
ProxySelector.setDefault(customProxySelector);

// 3. 正常创建WebSocket客户端和连接管理器
StandardWebSocketClient aStandardWebSocketClient = new StandardWebSocketClient();
WebSocketConnectionManager manager = new WebSocketConnectionManager(aStandardWebSocketClient, new WebSockethandler(), aWebSockURL);

额外注意事项 #

• 如果你的WebSocket是wss://协议，确保代理服务器支持HTTPS代理转发，否则可能无法完成握手。
• 若代理需要认证，除了设置系统属性，还可以通过自定义Authenticator来处理：

Authenticator.setDefault(new Authenticator() {
    @Override
    protected PasswordAuthentication getPasswordAuthentication() {
        return new PasswordAuthentication("代理用户名", "代理密码".toCharArray());
    }
});

内容的提问来源于stack exchange，提问作者Gangadhar