ReactorKit源碼閱讀

1. 前言

ReactorKit是創建在RxSwift生態下的一個狀態管理框架，說是框架，其實它的封裝很是輕量。本文是閱讀源碼的一些收穫。

2. 協議和運行時

爲了讓Swift的protocol extension具備 給實現協議的人動態添加一個屬性（依賴） 這個能力，它採用了runtime的實現方式。

例如給Reactor協議添加一個disposeBag變量：

protocol Reactor { ... }
extension Reactor {
    fileprivate var disposeBag: DisposeBag {
        get { return self.associatedObject(forKey: &disposeBagKey, default: DisposeBag()) }
    }
}
複製代碼

最終調用的是：

objc_setAssociatedObject(self, key, object, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
複製代碼

3. 誰來處理反作用

對於一個函數式框架來講，這點相當重要。好比Flux是經過ActionCreator，Redux是由Middlewares來真正處理這些對外界環境的不純函數（髒活累活）。

而ReactorKit這邊是由Reactor裏的mutate()函數來處理反作用的。

func mutate(action: Action) -> Observable<Mutation>
複製代碼

須要注意的是，因爲他的返回值是Observable，這意味着在Action轉換到Mutation的過程當中，能夠異步執行反作用來獲取結果（好比網絡請求，數據庫操做）。

4. 如何修改內部狀態(State)

Reactor的核心方法（也是最長的）就是：

func createStateStream() -> Observable<State> 
複製代碼

Reactor內聲明的大多數函數都在此方法被調用，狀態就是在這個方法內最終被改變的。

具體看一下他是怎麼作到的：

public func createStateStream() -> Observable<State> {
    let action = self._action.asObservable()
    let transformedAction = self.transform(action: action)
    let mutation = transformedAction
      .flatMap { [weak self] action -> Observable<Mutation> in
        guard let `self` = self else { return .empty() }
        return self.mutate(action: action).catchError { _ in .empty() }
      }
    let transformedMutation = self.transform(mutation: mutation)
    let state = transformedMutation
      .scan(self.initialState) { [weak self] state, mutation -> State in
        guard let `self` = self else { return state }
        return self.reduce(state: state, mutation: mutation)
      }
      .catchError { _ in .empty() }
      .startWith(self.initialState)
      .observeOn(MainScheduler.instance)
    let transformedState = self.transform(state: state)
      .do(onNext: { [weak self] state in
        self?.currentState = state
      })
      .replay(1)
    transformedState.connect().disposed(by: self.disposeBag)
    return transformedState
  }
複製代碼

• 【Action】 取Observable<Action>
• 【Action】 放到transform()函數中，給用戶時機改變其序列類型
• 【Action】 因而獲得了轉換過的Observable<Action>，記做transformedAction
• 【Action 綁定 (Action) -> Mutation】 把transformedAction綁定(flatMap)到mutate()函數上，mutate()的簽名是(Action) -> Observable<Mutation>，用戶的反作用在綁定過程當中執行。
• 【Mutation】 因而獲得新的序列，類型爲Observable<Mutation>，意味着用戶已經把數據拿到，一切就位。
• 【Mutation】 放到transform()函數中，給用戶時機改變其序列類型
• 【Mutation】 因而獲得了轉換過的Observable<Mutation>，記做transformedMutation
• 【(State, Mutation) -> State】 即執行reduce()方法，經過scan函數讓mutation能夠不斷做用到state上
• 【State】 獲得結果Observable<State>
• 【State】 放到transform()函數中，給用戶時機改變其序列類型
• 【State】 因而獲得了轉換過的Observable<State>，記做transformedState
• 【State】 順便修改內部的currentState
• 【State】 返回最終結果

這個過程結合這個圖看就更加清晰，這個函數（也是整個Reactor）的做用就是要作這麼一件事： Action->Mutation->State