fix(graphql): fix SocketClient to close subscription on socket close

Fixes #1183
zino-hofmann · Jul 22, 2022 · 45f88f6 · 45f88f6
1 parent a19958f
commit 45f88f6
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/packages/graphql/lib/src/links/websocket_link/websocket_client.dart b/packages/graphql/lib/src/links/websocket_link/websocket_client.dart
@@ -495,7 +495,19 @@ class SocketClient {
                 .where((message) => message is SubscriptionComplete)
                 .take(1);
 
-        subscriptionComplete.listen((_) => response.close());
+        subscriptionComplete.listen(
+          (_) => response.close(),
+          onDone: () {
+            if (!config.autoReconnect) {
+              response.close();
+            }
+          },
+          onError: (_) {
+            if (!config.autoReconnect) {
+              response.close();
+            }
+          },
+        );
 
         dataErrorComplete
             .where((message) => message is SubscriptionData)