1 files changed, 11 insertions, 0 deletions
diff --git a/Example/Messaging/App/iOS/AppDelegate.swift b/Example/Messaging/App/iOS/AppDelegate.swift
index 3f81784..4f8e504 100644
--- 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)")
+    }
+}