Swift 3 以降の NotificationCenter の正しい使い方

  • 92
    Like
  • 1
    Comment

Swift 3 で NotificationCenter(旧 NSNotificationCenter ) がせっかく改良されたのに、そのメリットが活かされないようなコード例をよく見かけるので、正しい使い方をまとめます。

次の class を用意して、それをもとに説明していきます(struct は自身を登録できないので class で説明)。

class NotificationObserver {
    func addObserver() {
        // A: ここに NotificationCenter 通知登録処理を書く
    }
    func removeObserver() {
        // B: ここに NotificationCenter 通知解除処理を書く
    }

    /** C: 今回、通知された時に呼ばれる用のメソッド */
    @objc private func notified(notification: Notification) {
        print("呼ばれた( ´・‿・`): \(notification)")
    }
}

すでに用意されている通知を利用する場合

例えば UIApplicationDidBecomeActiveNotification(アプリがアクティブになった時に呼ばれる)のタイミングで、自身の notified(notification:) メソッドが呼ばれるようにしたい場合、Swift 2 では次のように書きます。

let center = NSNotificationCenter.defaultCenter()

// A の箇所に記述
center.addObserver(self,
                   selector: #selector(type(of: self).notified(notification:)),
                   name: UIApplicationDidBecomeActiveNotification,
                   object: nil)

Swift 3 では 'UIApplicationDidBecomeActiveNotification' has been renamed to 'NSNotification.Name.UIApplicationDidBecomeActive' というコンパイルエラーメッセージとともに以下の修正を促されるので、それに従うと次のように、コンパイルの通るコードとなります。

Screen Shot 2016-12-25 at 11.06.24.png

let center = NotificationCenter.default

center.addObserver(self,
                   selector: #selector(type(of: self).notified(notification:)),
                   name: NSNotification.Name.UIApplicationDidBecomeActive,
                   object: nil)

このままでも問題無いですが、 NSNotification.Name 部分は型推論に頼って省略できます。

center.addObserver(self,
                   selector: #selector(type(of: self).notified(notification:)),
                   name: .UIApplicationDidBecomeActive,
                   object: nil)

好みの問題もあるかもしれませんが、僕は簡潔でこちらの方が好きです(書かずに済むものは極力書かずに済ませたいと思っています)。

NSNotification.Name.UIApplicationDidBecomeActive とは何かというと、次のように定義されている、NSNotification.Name 型の値です。

extension NSNotification {
    public struct Name : RawRepresentable, Equatable, Hashable, Comparable {
        public init(_ rawValue: String)
        public init(rawValue: String)
    }
}

extension NSNotification.Name {
    public static let UIApplicationDidBecomeActive: NSNotification.Name
}

ちなみに、クロージャーで記述する場合は、notified(notification:) メソッドを用意せずに次のように書けます。

center.addObserver(forName: .UIApplicationDidBecomeActive,
                   object: nil,
                   queue: nil) { notification in
    print("呼ばれた( ´・‿・`): \(notification)")
}

あるいは、notified(notification:) メソッドを用意したまま、次のように書くこともできます。

center.addObserver(forName: .UIApplicationDidBecomeActive,
                   object: nil,
                   queue: nil, 
                   using: notified)

クロージャー版の addObserver(forName:object:queue:using:) と他の登録メソッドで、解除の仕方が異なることに注意です。 http://qiita.com/mono0926/items/754c5d2dbe431542c75e#通知の登録解除の仕方 をご覧ください。

自身で新しい通知を利用する場合

Swift 2 までは次のように書いていました。

// 文字列をベタ書きせず、このように別途定義しておいた方がベター
let myNotification = "myNotification"

center.addObserver(self,
                   selector: #selector(type(of: self).notified(notification:)),
                   name: myNotification,
                   object: nil)

これも、Xcode のマイグレーション支援に従うと、次のようにコンパイルの通るコードとなります。

center.addObserver(self,
                   selector: #selector(type(of: self).notified(notification:)),
                   name: NSNotification.Name(rawValue: myNotification),
                   object: nil)

ただ、これでは記述がまどろっこしくなっただけで、Swift 3 で良くなったメリットをまったく享受できていません。Swift 2以前はあらゆる文字列変数や文字列リテラルをそのまま name 引数に与えられたのが、Swift 3では NSNotification.Name 型を受けるように変わったのに、逐一 rawValue にて文字列を渡していたら意味がありません。

次のように別途定義してある myNotificationNSNotification.Name 型とすると、良くなります。

// `rawValue:` のラベルは省略可能
let myNotification = NSNotification.Name("myNotification")

center.addObserver(self,
                   selector: #selector(type(of: self).notified(notification:)),
                   name: myNotification,
                   object: nil)

この時点で良い感じですが、さらに NSNotification.Name.UIApplicationDidBecomeActive など元々用意されているものと同様、次のように定義してみると良いでしょう。

extension NSNotification.Name {
    static let myNotification = NSNotification.Name("myNotification")
}

center.addObserver(self,
                   selector: #selector(type(of: self).notified(notification:)),
                   name: .myNotification,
                   object: nil)

Strongly Typed String Enumerationsとして、What's New in Foundation for Swift - WWDC 2016 - Videos - Apple Developer のpp.72-83にて紹介されているやり方です。

Screen Shot 2016-09-01 at 10.54.42 AM.png

さらに、次のように、struct などで区切ることで元々用意されているものと区別するなど、応用も色々考えられますが省略します。

extension NSNotification.Name {
    struct Qiita {
        static let myNotification = NSNotification.Name("qiita.myNotification")
    }
}

center.addObserver(self,
                   selector: #selector(type(of: self).notified(notification:)),
                   name: NSNotification.Name.Qiita.myNotification,
                   object: nil)

通知の発行の仕方

これは特に変わらず、上で良い感じに定義した NSNotification.Name.myNotification を使って発行します。

center.post(name: .myNotification, 
            object: nil)

次のような、よろしくないコード例が散見されて、気になっていました。

center.post(name: NSNotification.Name(rawValue: myNotification), 
            object: nil)

通知の登録解除の仕方

次のメソッドなど使って、基本的に登録と対になるように適切に解除しましょう(登録してないのに解除しても特にエラー起きないですが)。

// B の箇所に記述
// すべて解除
center.removeObserver(self)
// 特定の通知のみ解除
center.removeObserver(self,
                      name: .UIApplicationDidBecomeActive,
                      object: nil)

addObserver(forName:object:queue:using:)を使った場合の解除パターン

https://developer.apple.com/reference/foundation/notificationcenter/1411723-addobserver#return-value に載っていますが、addObserver(forName:object:queue:using:) を使う場合、通知登録解除時、戻り値の NSObjectProtocol 型のオブジェクトを保持してそれを元に removeObserver(_:) を呼ぶ必要があることに注意です。戻り値不使用であれば警告が出るのでうっかりミスはしにくいです。

次のように書きます。

class NotificationObserver {
    var token: NSObjectProtocol?

    func addObserver() {
        token = center.addObserver(forName: .myNotification,
                                   object: nil,
                                   queue: nil) { notification in
                                    print("呼ばれた( ´・‿・`): \(notification)")
        }
    }

    func removeObserver() {
        if let token = token {
            center.removeObserver(token)
            self.token = nil
        }
    }
}

1回のみ通知を受けたい場合

1回通知受けたら解除、という場合はプロパティに保持せずに次のように書けます。

var token: NSObjectProtocol!
token = center.addObserver(forName: .myNotification,
                   object: nil,
                   queue: nil) { notification in
                    print("呼ばれた( ´・‿・`): \(notification)")
                    center.removeObserver(token)
        }

以上のように、Swift 3 で NotificationCenter が改良されたのでベターな書き方をしていきましょう( ´・‿・`)

userInfo[AnyHashable : Any]? 型という問題

ただ、まだ微妙な点は残っていて、notification オブジェクトには userInfo という情報を付与できますが、それが[AnyHashable : Any]? 型なので、すごく使いにくいです。

public struct Notification : ReferenceConvertible, Equatable, Hashable {
    public var userInfo: [AnyHashable : Any]?
}

いくつかアプローチがありますが、

userInfo に頼らない

僕はこちらの方針でやっています。
NotificationCenter は 「特定の通知を検知すること」のみに使って、受け取った側でのハンドリングは受け取った側がアクセスできる情報で済ませる、という設計にしています。そもそも NotificationCenter は多用していないこともあって、これで充分済んでいます( ´・‿・`)

This post is the No.25 article of Swift Advent Calendar 2016