Log direct-channel state changes to console (#287)

firebase · Sep 19, 2017 · 66f3aed · 66f3aed
1 parent 9447e72
commit 66f3aed
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/Example/Messaging/App/iOS/AppDelegate.swift b/Example/Messaging/App/iOS/AppDelegate.swift
@@ -51,6 +51,8 @@ class AppDelegate: UIResponder, UIApplicationDelegate {
     FirebaseApp.configure()
     Messaging.messaging().delegate = self
     Messaging.messaging().shouldEstablishDirectChannel = true
+    // Just for logging to the console when we establish/tear down our socket connection.
+    listenForDirectChannelStateChanges();
 
     NotificationsController.configure()
 
@@ -126,3 +128,12 @@ extension AppDelegate: MessagingDelegate {
   }
 }
 
+extension AppDelegate {
+    func listenForDirectChannelStateChanges() {
+        NotificationCenter.default.addObserver(self, selector: #selector(onMessagingDirectChannelStateChanged(_:)), name: .MessagingConnectionStateChanged, object: nil)
+    }
+
+    func onMessagingDirectChannelStateChanged(_ notification: Notification) {
+        print("FCM Direct Channel Established: \(Messaging.messaging().isDirectChannelEstablished)")
+    }
+}