1. 目标
我希望我的自定义 WebSocket 服务器 (API) 和我的 Flutter 应用程序之间的连接能够在遇到网络问题或 WebSocket 服务器遇到问题时自动重新建立。
- 用例 1:wifi 停止并突然恢复。
- 用例 2:API 未启动,突然重启。
- 约束:我使用 Riverpod 作为状态管理库(我想保留它:))。我强调状态管理库是因为我在 StreamProvider 中创建了 WS 连接(参见 Riverpod)。
2. 没有自动重新连接的初始设置
- 我创建了一个 StreamProvider,如下所示:
final hostProvider =
StreamProvider.autoDispose.family<Host, String>((ref, ip) async* {
//SOCKET OPEN
final channel = IOWebSocketChannel.connect('ws://$ip:$port/v1/path');
ref.onDispose(() {
// SOCKET CLOSE
return channel.sink.close();
});
await for (final json in channel.stream) {
final jsonStr = jsonDecode(json as String);
yield Host.fromJson(jsonStr as Map<String, dynamic>);
}
});
- 我创建了一个小部件来使用数据:
useProvider(hostProvider(ip)).when(
data: (data) => show the result
loading: () => show progress bar
error: (error, _) => show error
);
这段代码效果很好。但是,没有自动重新连接机制。
3. 自动重新连接尝试
- 每当捕获到异常时,我都会在 try/catch 中调用函数 connectWs:
final hostProvider =
StreamProvider.autoDispose.family<Host, String>((ref, ip) async* {
// Open the connection
connectWs('ws://$ip:$port/v1/path').then((value) async* {
final channel = IOWebSocketChannel(value);
ref.onDispose(() {
return channel.sink.close();
});
await for (final json in channel.stream) {
final jsonStr = jsonDecode(json as String);
yield Host.fromJson(jsonStr as Map<String, dynamic>);
}
});
});
Future<WebSocket> connectWs(String path) async {
try {
return await WebSocket.connect(path);
} catch (e) {
print("Error! " + e.toString());
await Future.delayed(Duration(milliseconds: 2000));
return await connectWs(path);
}
}
- 我创建了一个 connectProvider 提供程序,如下所示,我在 hostProvider 中“观察”以创建一个通道。每当出现异常时,我都会使用 Riverpod 库中的刷新函数来重新创建通道:
// used in hostProvider
ref.container.refresh(connectProvider(ip))
final connectProvider =
Provider.family<Host, String>((ref, ip) {
//SOCKET OPEN
return IOWebSocketChannel.connect('ws://$ip:$port/v1/path');
});
在此先感谢您的帮助。