Okhttp WebSocket连接后如何获取默认SSLSocketFactory创建的Socket实例

公开API支持情况 #

OkHttp 未提供任何公开API用于直接获取WebSocket绑定的底层Socket实例。对外暴露的WebSocket接口仅封装了消息收发、连接关闭、请求参数查询这类上层能力，底层Socket、连接管理相关的逻辑全部被收在内部实现类中，不对外开放访问。

如果你的需求是配置Socket参数（比如设置TCP_NODELAY、调整读写超时、自定义Socket工厂），不需要拿Socket实例，直接通过OkHttpClient.Builder提供的公开配置项实现即可，这是最稳定的方案。

反射获取实现方式 #

如果确实需要拿到Socket实例做自定义操作，可以通过反射访问内部字段实现，注意必须在WebSocketListener的onOpen回调中执行反射逻辑，此时连接才完成初始化，内部字段不会为空。

以下是适配OkHttp 3.x、4.x主流版本的反射代码：

@Override
public void onOpen(@NotNull WebSocket webSocket, @NotNull Response response) {
    super.onOpen(webSocket, response);
    Socket targetSocket = null;
    try {
        // 第一步：获取RealWebSocket（WebSocket的内部实现类）持有的WebSocketConnection引用
        Class<?> realWsClass = Class.forName("okhttp3.internal.ws.RealWebSocket");
        Field connectionField = realWsClass.getDeclaredField("connection");
        connectionField.setAccessible(true);
        Object wsConnection = connectionField.get(webSocket);

        // 第二步：从WebSocketConnection中获取持有的RealConnection引用
        Field realConnField = wsConnection.getClass().getDeclaredField("connection");
        realConnField.setAccessible(true);
        Object realConnection = realConnField.get(wsConnection);

        // 第三步：从RealConnection中获取原始Socket实例
        Field rawSocketField = realConnection.getClass().getDeclaredField("rawSocket");
        rawSocketField.setAccessible(true);
        targetSocket = (Socket) rawSocketField.get(realConnection);

        // 如果是wss协议要拿SSLSocket实例，替换上一步的字段名为"sslSocket"即可
    } catch (Exception e) {
        e.printStackTrace();
    }

    // targetSocket就是你需要的底层Socket实例
}

反射方案注意事项 #

• 兼容性无保障：OkHttp的内部类、字段名不属于公开API约定，版本迭代时随时可能修改，升级OkHttp版本后需要重新校验反射逻辑是否匹配当前版本的实现。
• 禁止在连接未建立阶段（刚调用newWebSocket未收到onOpen回调时）执行反射，此时内部连接对象还未初始化，会拿到null或者抛出异常。
• 不要手动修改拿到的Socket的核心连接参数、不要主动关闭Socket，否则会导致WebSocket连接状态异常，出现意料之外的断连、消息收发失败问题。

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