Notification, 코드를 개선해보자

들어가며

Notification은 값, 데이터 전달할 때 정말 유용한 녀석이다. 어떻게 보면 사용 방법도 되게 단순한데 써야 할 코드가 곳곳에 퍼져있고, 많이 쓰다보면 관리가 어려워진다. 적절할 때 쓰면 참 좋은 친구... (특히 계층이 복잡할 때 쓰면 꿀이다 🍯) 오늘 그래서 단순하게 장점 설명하려고 글을 쓰게 된 것은 아니고, 불편했던 점을 개선해 본 것을 기록하고자 한다.

(아 그리고 글 쓰다 보면 말투가 정말 오락가락 한데 이해해주시길 ... 글 좀 더 써보고 한 가지로 정착해야겠다.)

---

불편했던 점

불편했던 점은 Notification.Name 안에 들어가는 문자열을 post 하는 쪽과 addObserving 하는 쪽을 정확히 맞춰주어야 한다는 점이다. 휴먼에러가 발생할 가능성이 높고, 코드가 곳곳에 퍼져있으면 정말 찾아보기 어렵다. 그리고 생각보다 코드 라인 수가 좀 많아지고, post · addObserver · removeObserver · 그리고 액션을 실행하는 함수(값을 받아서 사용할 수도 있음) 이렇게 작성해야 하는 파트가 참 많다.

참 좋은데 귀찮다는 것이다. 그래서 처음에는 Property Wrapper로 어찌 해보려다가 굳이 그렇게까지 복잡하게 Wrapping을 안해도 될 것 같았다. 찾아보니 나름 라인 수를 줄이면서 편리하게 관리할 수 있는 방법을 찾아서 기록하고자 한다.

---

예시

간단하게 역방향으로 값을 전달하는 예시이다.

Notification을 이용해서 역방향 값 전달하기

코드 개선

extension Notification.Name {
    
    static let test = Notification.Name("test")
}

이렇게 Notification.Name의 Extension으로 Key값을 관리하는 건 많이 알고 있는 사실이다. 리터럴한 값을 관리하기는 좋지만 여전히 작성해야 하는코드가 많다. 귀찮다..😂

extension Notification.Name {
    
    func post(object: Any? = nil, userInfo: [AnyHashable : Any]? = nil) {
        print("✈️ 〰〰〰〰〰〰〰 post (신호 쐈어요!!)")
        NotificationCenter.default.post(name: self, object: object, userInfo: userInfo)
    }

    @discardableResult
    func addObserver(object: Any? = nil, queue: OperationQueue? = nil, using: @escaping (Notification) -> Void) -> NSObjectProtocol {
        print("👀 〰〰〰〰〰〰〰 addObserver (옵저버 등록!!)")
        return NotificationCenter.default.addObserver(forName: self, object: object, queue: queue, using: using)
    }
    
    func removeObserver(observer: Any, object: Any? = nil) {
        print("⛔️ 〰〰〰〰〰〰〰 removeObserver (옵저버 등록 해제!!)")
        return NotificationCenter.default.removeObserver(observer, name: self, object: object)
    }
}

그래서 이렇게 메서드를 추가로 만들어준다. 키 값에 따라서 Notification.Name 값을 post, addObserver에 그대로 이용하자는 것이다.

1️⃣ Post

// 전

NotificationCenter.default.post(
    name: Notification.Name("test"),
    object: nil,
    userInfo: ["message": messageTextField.text!]
)

// 후

Notification.Name.test.post(object: nil, userInfo: ["message": messageTextField.text!])

나는 후자가 더 깔끔하고 사용하기 편하다고 느꼈다.

2️⃣ addObserver

// 전

NotificationCenter.default.addObserver(
    self,
    selector: #selector(testObserver),
    name: Notification.Name("test"),
    object: nil
)

@objc func testObserver(_ notification: NSNotification) {
    if let message = notification.userInfo?["message"] as? String {
        self.overviewLabel.text = message
    }
}

// 후

Notification.Name.test.addObserver { [weak self] noti in
    print("====================================")
    print("📩 〰〰〰〰〰〰〰 구독(메시지) 알림 왔어요잉")
    print("====================================")
    
    guard let self = self else { return }
    self.overviewLabel.text = noti.userInfo?["message"] as? String
}

print문을 포함하고 보더라도 후자가 훨씬 간단해보인다. 클로저를 통해 이어서 바로 동작을 수행하게 된다. 개선을 해보면서 편리함을 느꼈다. 물론 중요한 건 코드를 얼마나 깔끔하게 쓰냐보다 적재적소에 올바른 코드를 쓸 수 있냐가 아닐까 싶다.

혹시 잘못된 내용이나 이상한 점이 있다면 언제든지 답글을 달아주세요!