Apollo Kotlin 项目中 WebSocket 重连问题的分析与解决

在基于 Apollo Kotlin 构建的多平台应用开发中，WebSocket 连接的重连机制是一个关键功能点。本文将以一个实际案例为背景，深入分析在 iOS 和 Android 平台上 WebSocket 重连失效的问题，并探讨最终的解决方案。

问题背景

开发者在构建使用 AWS AppSync 和 WebSocket 通信的多平台应用时，遇到了 WebSocket 连接在临时网络中断后无法自动重连的问题。该应用同时面向 iOS 和 Android 平台，使用 Kotlin Multiplatform Mobile (KMM) 技术栈。

技术实现细节

项目中使用了 Apollo Kotlin 的 WebSocketNetworkTransport 来管理 WebSocket 连接，并配置了 reopenWhen 和 retryWhen 两个关键的重连策略：

val webSocketNetworkTransport = WebSocketNetworkTransport.Builder()
    .serverUrl(serverUrl = url)
    .protocol(protocolFactory = AppSyncWsProtocol.Factory(...))
    .reopenWhen { cause, attempt ->
        println("重连尝试: $attempt")
        delay(2.0.pow(attempt.toDouble()).toLong())
        true
    }
    .build()

订阅操作则通过 Flow 进行处理，并配置了重试逻辑：

apolloClient.subscription(UserExitedSubscription(sessionId)).toFlow()
    .map { it.data?.userExited?.exitSessionResponse }
    .retryWhen { cause, attempt -> 
          println("Flow重试: $attempt")
          delay(60000L)
          true
     }

平台差异表现

Android 平台行为

在 Android 平台上，开发者观察到以下现象：

1. 当 reopenWhen 返回 true 且网络仍处于断开状态时，会不断收到 UnknownHostException
2. 网络恢复后，WebSocket 无法自动重连
3. 通过临时方案（在网络恢复后执行一次查询）可以强制重连成功

iOS 平台行为

在 iOS 平台上，问题表现更为复杂：

1. 模拟器环境下，WebSocket 可以在网络恢复后自动重连
2. 真实设备上，reopenWhen 仅被调用一次，之后不再尝试重连
3. 错误类型为 ApolloNetworkException，不同于 Android 的 SocketException

问题根源分析

经过深入排查，发现问题主要源于以下几个方面：

1. 平台底层网络处理差异：iOS 和 Android 对网络中断的检测和报告机制不同，导致重连策略触发条件不一致
2. WebSocket 任务生命周期管理：在 iOS 设备上，WebSocket 任务在网络中断后会立即终止，而模拟器环境下有更宽松的重试机制
3. 错误传播机制：某些网络错误未能正确触发重连流程，导致重连策略中断

解决方案

最终解决方案涉及对 WebSocket 重连机制的改进：

1. 统一错误处理：对跨平台网络错误进行标准化处理
2. 增强重连检测：在网络恢复时主动触发重连检查
3. 任务生命周期管理：优化 WebSocket 任务在异常状态下的恢复逻辑

该修复已作为 Apollo Kotlin 4.0.0-beta.6 版本中实验性 WebSocketNetworkTransport 的一部分发布。

经验总结

在多平台开发中处理网络连接问题时，需要注意：

1. 不同平台对网络状态变化的处理存在差异
2. 模拟器与真实设备的行为可能不一致
3. 复杂的重连策略需要针对各平台进行充分测试
4. 结合主动检测和被动重试机制可以提高连接可靠性

通过这个案例，我们不仅解决了具体的技术问题，也为 Apollo Kotlin 项目的 WebSocket 实现提供了宝贵的实践经验。