okhttp的WebSocket心跳实现原理

okhttp的WebSocket实现心跳包需要服务端新增协议吗

‌不需要。‌

OkHttp的WebSocket实现已经内置了心跳包机制，通过PING/PONG帧来维持连接保活。这意味着，OkHttp的WebSocket客户端和服务端在通信过程中，会自动发送PING/PONG帧来检测连接的活跃状态，无需服务端新增任何协议或特别处理。这种机制确保了即使在没有数据传输的情况下，连接也能保持活跃，防止因为长时间没有通信而导致连接被意外关闭。因此，使用OkHttp进行WebSocket通信时，服务端不需要做任何特别的处理来实现心跳包功能‌1。

此外，WebSocket协议本身已经设计了心跳机制，这个功能用于检测链接是否可用。虽然心跳功能不一定支持携带数据，具体实现可能会有所不同，但基本的心跳检测功能已经包含在WebSocket协议中，无需服务端额外实现‌

WebSocket协议 的PING/PONG机制

WebSocket 的 PING/PONG 是一种保持连接活跃的机制。当服务器需要检查客户端是否仍然在线，或者客户端需要检查服务器是否在线时，它们可以相互发送 PING 消息。对方需要回复 PONG 消息以证明自己在线。

PING 是一个控制帧，用于检测对端是否在线，PONG 是对 PING 的响应。PING 可以携带数据，PONG 必须不携带数据

okhttp的WebSocket添加心跳

首先，我来简单梳理一下okhttp中心跳的实现原理，如果只是想要解决方法的朋友可以直接跳过这一部分。

在okhttp中，实现心跳的方式非常简单，只需要在OkHttpClient创建时添加相应的配置即可：

  OkHttpClient.Builder().pingInterval(HEART_BEAT_RATE, TimeUnit.SECONDS).build()

okhttp的WebSocket的心跳实现原理

    //OkHttpClient.java@Override public WebSocket newWebSocket(Request request, WebSocketListener listener) {RealWebSocket webSocket = new RealWebSocket(request, listener, new Random(), pingInterval);webSocket.connect(this);return webSocket;}

   //RealWebSocket.javapublic RealWebSocket(Request request, WebSocketListener listener, Random random,long pingIntervalMillis) {//...this.pingIntervalMillis = pingIntervalMillis;//...}public void initReaderAndWriter(String name, Streams streams) throws IOException {synchronized (this) {//...this.executor = new ScheduledThreadPoolExecutor(1, Util.threadFactory(name, false));if (pingIntervalMillis != 0) {executor.scheduleAtFixedRate(new PingRunnable(), pingIntervalMillis, pingIntervalMillis, MILLISECONDS);}//...}}private final class PingRunnable implements Runnable {@Override public void run() {writePingFrame();}}void writePingFrame() {//...try {writer.writePing(ByteString.EMPTY);} catch (IOException e) {failWebSocket(e, null);}//...}//WebSocketWriter.javavoid writePing(ByteString payload) throws IOException {writeControlFrame(OPCODE_CONTROL_PING, payload);}

上面的代码就是ping的主要发送逻辑了，简单总结一下就是如果pingInterval不为0，那就开启一个的循环任务，定时的去发送代表ping的ControlFrame。

其中值得一提的就是ControlFrame这个概念，在WebSocket中的frame分为两类，一类叫做MessageFrame，也就是平时客户端与服务端互相通信的部分。另一类叫做ControlFrame，其中包括CONTROL_PING，CONTROL_PONG，CONTROL_CLOSE，可以看出这一类更偏重与功能性的方面。具体为哪一类的Frame可以在Header中进行区分。

上面已经介绍了心跳的发送逻辑，那么下面就轮到接收的逻辑了，还是先来看看代码：

    //RealWebSocket.javapublic void loopReader() throws IOException {while (receivedCloseCode == -1) {// This method call results in one or more onRead* methods being called on this thread.reader.processNextFrame();}}//WebSocketReader.javavoid processNextFrame() throws IOException {readHeader();if (isControlFrame) {readControlFrame();} else {readMessageFrame();}}private void readControlFrame() throws IOException {//...switch (opcode) {case OPCODE_CONTROL_PING:frameCallback.onReadPing(controlFrameBuffer.readByteString());break;case OPCODE_CONTROL_PONG:frameCallback.onReadPong(controlFrameBuffer.readByteString());break;case OPCODE_CONTROL_CLOSE://...default:throw new ProtocolException("Unknown control opcode: " + toHexString(opcode));}}

可以看到，接收的部分逻辑也很简单，就是通过一个循环去读取，如果接收到了消息，那就先通过header确定frame的类型，然后再分类进行处理。

而且值得注意的是，上面代码中出现了一个frameCallback的对象，而这个对象是WebSocketReader.FrameCallback这个接口的实现，而里面的onReadPing和onReadPong就是我们之后能够做文章的地方了。

    WebSocketReader.FrameCallbackpublic interface FrameCallback {void onReadMessage(String text) throws IOException;void onReadMessage(ByteString bytes) throws IOException;void onReadPing(ByteString buffer);void onReadPong(ByteString buffer);void onReadClose(int code, String reason);}

okhttp的WebSocket的心跳超时会自动重连吗

‌OkHttp的WebSocket实现支持心跳超时自动重连。‌

OkHttp的WebSocket实现通过设置心跳间隔来检测连接状态。如果客户端在一段时间内没有收到服务器的响应，即心跳超时，它会认为连接断开，并触发断线重连机制。这个过程确保了与服务器的持续通信。
具体来说，OkHttp允许用户设置心跳间隔，例如，通过:[OkHttpClient.Builder].pingInterval()方法设置心跳间隔，这个方法允许用户指定心跳消息发送的频率。如果服务器在规定的时间内没有响应心跳消息，客户端会认为连接已断开，并尝试重新建立连接。这种机制对于保持长时间运行的WebSocket连接特别重要，因为它能够自动处理网络不稳定或临时中断的情况，从而保持应用的实时性和响应性。

此外，OkHttp还支持通过配置来设置重连间隔，例如，通过:[RxWebSocket]的配置来设置重连间隔，这为用户提供了更大的灵活性，可以根据应用的需求调整重连的策略。这种自动重连的功能对于提高应用的健壮性和用户体验非常重要，尤其是在网络条件不稳定的环境中‌。

相关内容:
为okhttp的WebSocket添加心跳回调