add example code of notification

p-x9 · Mar 5, 2023 · 05d3ff0 · 05d3ff0
1 parent 9581ebe
commit 05d3ff0
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/Example/Example/ViewController.swift b/Example/Example/ViewController.swift
@@ -27,6 +27,9 @@ class ViewController: UIViewController {
     }
 
     var orderedDictionary: OrderedDictionary<String, Any> = .init()
+
+    private let notificationCenter = NotificationCenter.default
+    private var notificationObservations = [Any]()
 
     override func viewDidLoad() {
         super.viewDidLoad()
@@ -40,11 +43,14 @@ class ViewController: UIViewController {
         super.viewWillAppear(animated)
 
         title = appContainer.activeContainer?.name
+
+        registerNotifications()
     }
 
     override func viewDidDisappear(_ animated: Bool) {
         super.viewDidDisappear(animated)
 
+        unregisterNotifications()
     }
 
     private func setupViews() {
@@ -273,6 +279,26 @@ extension ViewController: UITableViewDelegate {
     }
 }
 
+extension ViewController {
+    func registerNotifications() {
+        let willChangeObservation = notificationCenter.addObserver(forName: AppContainer.containerWillChangeNotification, object: nil, queue: .current) { _ in
+            print("container will change")
+        }
+
+        let didChangeObservation = notificationCenter.addObserver(forName: AppContainer.containerWillChangeNotification, object: nil, queue: .current) { _ in
+            print("container did change")
+        }
+
+        self.notificationObservations = [willChangeObservation, didChangeObservation]
+    }
+
+    func unregisterNotifications() {
+        notificationObservations.forEach {
+            notificationCenter.removeObserver($0)
+        }
+    }
+}
+
 
 extension ViewController {
     class View: UIView {