fix logic

aws-amplify · Nov 14, 2023 · b503056 · b503056
1 parent f980283
commit b503056
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/...onProvider/AppsyncRealtimeConnectionAsync/RealtimeConnectionProviderAsync+Websocket.swift b/...onProvider/AppsyncRealtimeConnectionAsync/RealtimeConnectionProviderAsync+Websocket.swift
@@ -31,6 +31,9 @@ extension RealtimeConnectionProviderAsync: AppSyncWebsocketDelegate {
                 return
             }
             #if os(watchOS)
+            AppSyncLogger.debug(
+                "[RealtimeConnectionProvider] on watchOS, received disconnect."
+            )
             self.updateCallback(event: .error(ConnectionProviderError.connection(
                 "This API uses low-level networking (websockets). Running on watchOS only works for specific circumstances.",
                 error)))

diff --git a/AppSyncRealTimeClient/Websocket/Starscream/StarscreamAdapter.swift b/AppSyncRealTimeClient/Websocket/Starscream/StarscreamAdapter.swift
@@ -68,7 +68,8 @@ public class StarscreamAdapter: AppSyncWebsocketProvider {
                         AppSyncLogger.debug(
                             "[StarscreamAdapter] Manually send disconnect."
                         )
-                        self.delegate?.websocketDidDisconnect(provider: self, error: nil)
+                        let error = ConnectionProviderError.connection("WatchOS Error", nil)
+                        delegate?.websocketDidDisconnect(provider: self, error: error)
                     } else {
                         AppSyncLogger.debug(
                             "[StarscreamAdapter] watchOS subscriptions are connected within 3s."