@@ -15,34 +15,34 @@ public protocol ApolloWebSocketClient: WebSocketClient {
1515}
1616
1717public protocol WebSocketTransportDelegate : class {
18- func webSocketTransportDidConnect( _ webSocketTransport: WebSocketTransport )
19- func webSocketTransportDidReconnect( _ webSocketTransport: WebSocketTransport )
20- func webSocketTransport( _ webSocketTransport: WebSocketTransport , didDisconnectWithError error: Error ? )
18+ func webSocketTransportDidConnect( _ webSocketTransport: WebSocketTransport )
19+ func webSocketTransportDidReconnect( _ webSocketTransport: WebSocketTransport )
20+ func webSocketTransport( _ webSocketTransport: WebSocketTransport , didDisconnectWithError error: Error ? )
2121}
2222
2323public extension WebSocketTransportDelegate {
24- func webSocketTransportDidConnect( _ webSocketTransport: WebSocketTransport ) { }
25- func webSocketTransportDidReconnect( _ webSocketTransport: WebSocketTransport ) { }
26- func webSocketTransport( _ webSocketTransport: WebSocketTransport , didDisconnectWithError error: Error ? ) { }
24+ func webSocketTransportDidConnect( _ webSocketTransport: WebSocketTransport ) { }
25+ func webSocketTransportDidReconnect( _ webSocketTransport: WebSocketTransport ) { }
26+ func webSocketTransport( _ webSocketTransport: WebSocketTransport , didDisconnectWithError error: Error ? ) { }
2727}
2828
2929/// A network transport that uses web sockets requests to send GraphQL subscription operations to a server, and that uses the Starscream implementation of web sockets.
3030public class WebSocketTransport : NetworkTransport , WebSocketDelegate {
3131 public static var provider : ApolloWebSocketClient . Type = ApolloWebSocket . self
3232 public weak var delegate : WebSocketTransportDelegate ?
33-
33+
3434 var reconnect = false
3535 var websocket : ApolloWebSocketClient
3636 var error : Error ? = nil
3737 let serializationFormat = JSONSerializationFormat . self
38-
38+
3939 private final let protocols = [ " graphql-ws " ]
40-
40+
4141 private var acked = false
4242
4343 private var queue : [ Int : String ] = [ : ]
4444 private var connectingPayload : GraphQLMap ?
45-
45+
4646 private var subscribers = [ String : ( JSONObject ? , Error ? ) -> Void ] ( )
4747 private var subscriptions : [ String : String ] = [ : ]
4848
@@ -63,8 +63,8 @@ public class WebSocketTransport: NetworkTransport, WebSocketDelegate {
6363
6464 public func send< Operation> ( operation: Operation , completionHandler: @escaping ( _ response: GraphQLResponse < Operation > ? , _ error: Error ? ) -> Void ) -> Cancellable {
6565 if let error = self . error {
66- completionHandler ( nil , error)
67- return EmptyCancellable ( )
66+ completionHandler ( nil , error)
67+ return EmptyCancellable ( )
6868 }
6969
7070 return WebSocketTask ( self , operation) { ( body, error) in
@@ -150,14 +150,14 @@ public class WebSocketTransport: NetworkTransport, WebSocketDelegate {
150150 self . error = nil
151151 initServer ( )
152152 if reconnected {
153- self . delegate? . webSocketTransportDidReconnect ( self )
153+ self . delegate? . webSocketTransportDidReconnect ( self )
154154 // re-send the subscriptions whenever we are re-connected
155155 // for the first connect, any subscriptions are already in queue
156156 for (_, msg) in self . subscriptions {
157157 write ( msg)
158158 }
159159 } else {
160- self . delegate? . webSocketTransportDidConnect ( self )
160+ self . delegate? . webSocketTransportDidConnect ( self )
161161 }
162162
163163 reconnected = true
0 commit comments