viewWillAppear,viewDidAppearで一度だけ処理をする #Swift

2016/02/02 修正・加筆

一度だけ処理をしたい

タイミングによっては、 viewDidLoadより、 viewWillAppear, viewDidAppearで行ったほうが良い処理もあったりしますが、
viewDidLoadとは違い、後者の2つのメソッドは、他の画面に遷移して戻ってきた時にも発火してしまうので、一度だけ処理を行いたいものも何度も呼ばれてしまう可能性があります。
(例えば、 viewWillAppearのタイミングで、はじめて遷移した一度だけサーバーから何かを通信して取得したい場合や、
はじめて遷移した一度だけ、アニメーションを行う場合。アニメーションはiOS7以降、 viewDidLoadだと正常に動かない可能性があるため。)

シチュエーションはこんな感じです。

ある画面の遷移で、
A → B → C
と遷移するときに、画面Bにおいて、
A → B (push)
と遷移するときはviewWillAppearである処理(X)をしたくて、
B ← C (pop)
と画面を戻るときは、viewWillAppearが呼ばれるけど、処理Xはしたくない。
再度、Aまで戻って、
A → B (push)
と遷移するときはviewWillAppearで処理Xをしたい、

ある ViewControllerにはじめて遷移した且つ、 viewWillAppear, viewDidAppearで一度だけ処理を行いたい場合、以下のようにするのも1つの手段ですが、毎回それのための変数を用意するのは 正直だるいです。

これを毎回書くのはだるい...

class SomeViewController: UIViewController {
    private var firstAppear: Bool = false

    override func viewWillAppear(animated: Bool) {
        super.viewWillAppear(animated)
        if !firstAppear {
            // do something...
            firstAppear = true
        }
        // ...
    }
}

なので、その実装を隠蔽して使い回しができるように、実装してみました。

FirstAppearing
(Carthage/CocoaPods対応してあります。)

実装

実装はこちらをご覧ください。

protocolとextension、Associated Objectを使っている程度の軽量ソースとなってます。
一応UIViewController(とその継承クラス)のみ使用できるようにprotocolに制約をかけています。

使用方法

以下のように、UIViewControllerに、 FirstAppearingを適応すれば、
viewWillAppearOnce, viewDidAppearOnceが使用できるようになります。
そのViewControllerが破棄されるまでの間、呼びだされても一度しか closure内に書いた処理を実行しません。
また、 @noescape属性付きなので、 [unowned self]みたいなのは不要です。

また、 fromFunctionの引数にデフォルトで __FUNCTION__を仕込むことで、呼び出し元のメソッド名が取得できるので、
これを利用して、 viewWillAppear/viewDidAppear以外のメソッドで呼び出すのを抑制しています。

FirstAppearingを適応

class SomeViewController: UIViewController, FirstAppearing {

    override func viewDidLoad() {
        super.viewDidLoad()
        viewWillAppearOnce() {
            print("Wahaha!!") // not working!!
        }
    }

    override func viewWillAppear(animated: Bool) {
        super.viewWillAppear(animated)
        viewWillAppearOnce() {
            // do something...
            print("viewWillAppear Once") // call once!
        }
    }

}

(これだけだとサンプルとしては不足しているのですが、)
SomeViewControllerから別のViewControllerに遷移(modalでもpushでも)して、
戻ってきた時に、再度 viewWillAppearが呼ばれた時は、 viewWillAppearOnce内の処理は行われません。

追記: dispatch_onceを使う場合

コメントより頂きました。
boolのフラグを立てるよりは楽になるかと思います。
これあればいいじゃんってなりそうですが、お好みでどうぞ。

class SomeViewController: UIViewController {
    private var onceTokenViewWillAppear: dispatch_once_t = 0

    override func viewWillAppear(animated: Bool) {
        super.viewWillAppear(animated)

        dispatch_once(&onceTokenViewWillAppear) {
            print(self, "viewWillAppear")
        }
    }

    private var onceTokenViewDidAppear: dispatch_once_t = 0
    override func viewDidAppear(animated: Bool) {
        super.viewDidAppear(animated)

        dispatch_once(&onceTokenViewDidAppear) {
            print(self, "viewDidAppear")
        }
    }
}