如何为Spring WebSockets的SockJsClient配置程序化代理

由于底层依赖的org.apache.tomcat.websocket.WsWebSocketContainer硬编码使用ProxySelector.getDefault()获取代理，我们可以通过线程局部临时替换默认ProxySelector的方式，为特定STOMP连接单独配置代理，避免影响JVM内其他网络请求。

步骤1：封装代理切换工具类 #

创建工具类管理线程内的代理配置切换，确保操作完成后恢复原设置：

import java.net.ProxySelector;
import java.net.InetSocketAddress;

public class ProxyContext {
    private static final ThreadLocal<ProxySelector> ORIGINAL_SELECTOR = new ThreadLocal<>();

    /**
     * 为当前线程设置自定义代理
     */
    public static void setCustomProxy(String proxyHost, int proxyPort) {
        ORIGINAL_SELECTOR.set(ProxySelector.getDefault());
        ProxySelector customSelector = ProxySelector.of(new InetSocketAddress(proxyHost, proxyPort));
        ProxySelector.setDefault(customSelector);
    }

    /**
     * 恢复当前线程的默认代理配置
     */
    public static void restoreProxy() {
        ProxySelector original = ORIGINAL_SELECTOR.get();
        if (original != null) {
            ProxySelector.setDefault(original);
            ORIGINAL_SELECTOR.remove();
        }
    }
}

步骤2：在STOMP连接流程中切换代理 #

在初始化WebSocketStompClient并建立连接的代码块中，先设置自定义代理，连接完成后立即恢复默认配置：

import org.springframework.messaging.simp.stomp.StompSession;
import org.springframework.web.socket.client.standard.StandardWebSocketClient;
import org.springframework.web.socket.messaging.WebSocketStompClient;

// 初始化WebSocketStompClient
WebSocketStompClient stompClient = new WebSocketStompClient(new StandardWebSocketClient());

// 自定义代理参数
String proxyHost = "127.0.0.1";
int proxyPort = 8081;

try {
    // 临时替换代理配置
    ProxyContext.setCustomProxy(proxyHost, proxyPort);
    
    // 建立STOMP连接（根据实际业务调整连接逻辑，比如使用回调或Future）
    StompSession session = stompClient.connect("ws://your-spring-boot-endpoint/ws", null)
            .get();
    
    // 后续STOMP消息订阅/发送操作...
} catch (Exception e) {
    // 异常处理逻辑
} finally {
    // 必须恢复原代理配置，避免影响其他线程的网络请求
    ProxyContext.restoreProxy();
}

扩展：代理认证支持 #

如果代理需要身份验证，可以扩展工具类，同步切换Authenticator：

import java.net.Authenticator;
import java.net.PasswordAuthentication;

public class ProxyContext {
    private static final ThreadLocal<ProxySelector> ORIGINAL_SELECTOR = new ThreadLocal<>();
    private static final ThreadLocal<Authenticator> ORIGINAL_AUTH = new ThreadLocal<>();

    /**
     * 设置带认证的自定义代理
     */
    public static void setCustomProxyWithAuth(String proxyHost, int proxyPort, String proxyUser, String proxyPass) {
        // 保存原配置
        ORIGINAL_SELECTOR.set(ProxySelector.getDefault());
        ORIGINAL_AUTH.set(Authenticator.getDefault());
        
        // 设置代理认证
        Authenticator.setDefault(new Authenticator() {
            @Override
            protected PasswordAuthentication getPasswordAuthentication() {
                if (getRequestorType() == RequestorType.PROXY) {
                    return new PasswordAuthentication(proxyUser, proxyPass.toCharArray());
                }
                return super.getPasswordAuthentication();
            }
        });
        
        // 设置自定义代理选择器
        ProxySelector customSelector = ProxySelector.of(new InetSocketAddress(proxyHost, proxyPort));
        ProxySelector.setDefault(customSelector);
    }

    /**
     * 恢复所有原配置
     */
    public static void restoreProxy() {
        // 恢复ProxySelector
        ProxySelector originalSelector = ORIGINAL_SELECTOR.get();
        if (originalSelector != null) {
            ProxySelector.setDefault(originalSelector);
            ORIGINAL_SELECTOR.remove();
        }
        
        // 恢复Authenticator
        Authenticator originalAuth = ORIGINAL_AUTH.get();
        if (originalAuth != null) {
            Authenticator.setDefault(originalAuth);
            ORIGINAL_AUTH.remove();
        }
    }
}

原理说明 #

WsWebSocketContainer在创建WebSocket连接时会调用ProxySelector.getDefault()，而ProxySelector.setDefault()是全局设置，但我们通过ThreadLocal保存线程原配置，仅在当前连接线程中临时替换代理，操作完成后立即恢复，既实现了单个连接的自定义代理，又不会影响JVM内其他线程的网络请求，完全适配企业级多应用不同代理的需求。

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