at the risk of duplicate state notifications, adding in a few spots o…

…f disconnect that happens during read errors or exception handling (#94)
automerge · May 14, 2024 · 9ccb152 · 9ccb152
1 parent cf9fdd5
commit 9ccb152
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Sources/AutomergeRepo/Networking/Providers/WebSocketProvider.swift b/Sources/AutomergeRepo/Networking/Providers/WebSocketProvider.swift
@@ -419,6 +419,7 @@ public final class WebSocketProvider: NetworkProvider {
             } catch {
                 // error scenario with the WebSocket connection
                 Logger.websocket.warning("WEBSOCKET: Error reading websocket: \(error.localizedDescription)")
+                statePublisher.send(.disconnected)
                 peered = false
                 msgFromWebSocket = nil
             }
@@ -443,6 +444,7 @@ public final class WebSocketProvider: NetworkProvider {
         self.peered = false
         webSocketTask?.cancel()
         webSocketTask = nil
+        statePublisher.send(.disconnected)
         Logger.websocket.warning("WEBSOCKET: receive and reconnect loop terminated")
     }