Added cleanup after disconnect and call delegate after cleanup.

centrifugal · Feb 16, 2018 · fc2d32f · fc2d32f
1 parent a3a7a67
commit fc2d32f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/CentrifugeiOS/Classes/CentrifugeClientImpl.swift b/CentrifugeiOS/Classes/CentrifugeClientImpl.swift
@@ -147,6 +147,7 @@ class CentrifugeClientImpl: NSObject, CentrifugeClient, WebSocketDelegate {
      */
     func defaultProcessHandler(messages: [CentrifugeServerMessage]?, error: Error?) {
         if let error = error {
+            resetState()
             delegate?.client(self, didDisconnectWithError: error)
             return
         }
@@ -216,9 +217,9 @@ class CentrifugeClientImpl: NSObject, CentrifugeClient, WebSocketDelegate {
 
         // Client events
         case .disconnect:
-            delegate?.client(self, didDisconnectWithError: NSError.errorWithMessage(message: message))
             resetState()
             ws.disconnect()
+            delegate?.client(self, didDisconnectWithError: NSError.errorWithMessage(message: message))
         case .refresh:
             delegate?.client(self, didReceiveRefreshMessage: message)
         default: