RxSwiftでクロージャを減らす: APIリクエストのcompletionHandlerをObservableにしてみる

RxSwiftを適用する箇所として、クロージャをObservableに切り替えるのが最近良いなと思ったので掲載。

APIクライアントからレスポンスを受けてUIを更新する操作はクロージャを使いますがこれをObservableを使って書き換えてみました。

Before

import UIKit

class ViewController: UIViewController {
    private var label: UILabel!

    override func viewDidLoad() {
        super.viewDidLoad()

        label = UILabel(frame: view.frame)
        label.numberOfLines = 10

        fetch(completion: { [weak self] data in
            guard let strongSelf = self else { return }
            strongSelf.label.text = String(data: data, encoding: .ascii)
            strongSelf.view.addSubview(strongSelf.label)
        })
    }

    private func fetch(completion: @escaping ((Data) -> Void)) {
        let request = URLRequest(url: URL(string: "https://www.google.com")!)
        URLSession.shared.dataTask(with: request, completionHandler: { data, _, error in
            guard
                error == nil,
                let data = data
            else { return }
            completion(data)
        }).resume()
    }
}

After

import UIKit
import RxSwift
import RxCocoa

class ViewController: UIViewController {
    private var label: UILabel!
    private let disposeBag = DisposeBag()

    enum MyError: Error {
        case APIClientFailed
    }

    override func viewDidLoad() {
        super.viewDidLoad()

        label = UILabel(frame: view.frame)
        label.numberOfLines = 10

        fetch().asObservable().subscribe(onNext: { [weak self] data in
            guard let strongSelf = self else { return }
            strongSelf.label.text = String(data: data, encoding: .ascii)
            strongSelf.view.addSubview(strongSelf.label)
        }).disposed(by: disposeBag)
    }

    private func fetch() -> Observable<Data> {
        return Observable<Data>.create { observer in
            let request = URLRequest(url: URL(string: "https://www.google.com")!)
            URLSession.shared.dataTask(with: request, completionHandler: { data, _, error in
                guard
                    error == nil,
                    let data = data
                else { return observer.on(.error(MyError.APIClientFailed)) }
                observer.on(.next(data))
                observer.on(.completed)
            }).resume()

            return Disposables.create()
        }
    }
}

こういう風に作っておくと、一度に2つ以上のエンドポイントにリクエストを送ってその内容をマージしたくなったり、リクエスト前後にバリデーションが必要になったりしたときにRxSwiftの能力を発揮して複雑性を増やさずにコードを書ける気がしています。