fix: Websocket error broadcast for unsubscribed ID (apollographql/apo…

…llo-ios-dev#506)
apollographql · Oct 16, 2024 · 9d8aafb · 9d8aafb
1 parent bc5a0cd
commit 9d8aafb
Showing 1 changed file with 15 additions and 9 deletions.
diff --git a/Sources/ApolloWebSocket/WebSocketTransport.swift b/Sources/ApolloWebSocket/WebSocketTransport.swift
@@ -193,10 +193,21 @@ public class WebSocketTransport {
       }
 
       switch messageType {
-      case .data,
-           .next,
-           .error:
-        if let id = parseHandler.id, let responseHandler = subscribers[id] {
+      case .data, .next, .error:
+        guard let id = parseHandler.id else {
+          let websocketError = WebSocketError(
+            payload: parseHandler.payload,
+            error: parseHandler.error,
+            kind: .unprocessedMessage(text)
+          )
+          self.notifyErrorAllHandlers(websocketError)
+
+          break
+        }
+
+        // If we have a handler ID but no subscriber exists for that ID then the
+        // subscriber probably unsubscribed.
+        if let responseHandler = subscribers[id] {
           if let payload = parseHandler.payload {
             responseHandler(.success(payload))
           } else if let error = parseHandler.error {
@@ -207,11 +218,6 @@ public class WebSocketTransport {
                                                 kind: .neitherErrorNorPayloadReceived)
             responseHandler(.failure(websocketError))
           }
-        } else {
-          let websocketError = WebSocketError(payload: parseHandler.payload,
-                                              error: parseHandler.error,
-                                              kind: .unprocessedMessage(text))
-          self.notifyErrorAllHandlers(websocketError)
         }
       case .complete:
         if let id = parseHandler.id {