Subclassing Provider and refreshing session automatically（子類化Provider尾膊，session的自動刷新）

基于 Artsy's implementation. 使用RxSwift

class OnlineProvider: RxMoyaProvider<MyService> {

  // First of all, we need to override designated initializer
  override init(endpointClosure: MoyaProvider<MyService>.EndpointClosure = MoyaProvider.defaultEndpointMapping,
      requestClosure: MoyaProvider<MyService>.RequestClosure = MoyaProvider.defaultRequestMapping,
      stubClosure: MoyaProvider<MyService>.StubClosure = MoyaProvider.neverStub,
      manager: Manager = Alamofire.SessionManager.default,
      plugins: [PluginType] = []) {

      super.init(endpointClosure: endpointClosure, requestClosure: requestClosure, stubClosure: stubClosure, manager: manager, plugins: plugins)
  }

  // Request to fetch and store new XApp token if the current token is missing or expired.
  func XAppTokenRequest() -> Observable<String?> {

      var appToken = UserInfo.shared.accessToken

      // If we have a valid token, just return it
      if appToken.isValidAndNotExpired {
          return Observable.just(appToken.token)
      }

      // Do not attempt to refresh a session if we don't have valid credentials
      guard let userId = UserInfo.shared.userId, refreshToken = UserInfo.shared.accessToken.refreshToken else {
          return Observable.just(nil)
      }

      // Create actual refresh request
      let newTokenRequest = super.request(MyService.refreshSession(userId: userId, refreshToken: refreshToken))
          .filterSuccessfulStatusCodes()
          .mapJSON()
          .map { element -> (token: String?, refreshToken: String?, expiryTime: Double?) in
              guard let dictionary = element as? NSDictionary else { return (token: nil, refreshToken: nil, expiryTime: nil) }

              return (token: dictionary["auth_token"] as? String, refreshToken: dictionary["refresh_token"] as? String, expiryTime: dictionary["session_time_valid"] as? Double)
          }
          .doOn { event in
              guard case .next(let element) = event else { return }

              UserInfo.shared.accessToken.token = element.0
              UserInfo.shared.accessToken.refreshToken = element.1
              UserInfo.shared.accessToken.setExpirySecondsLeft(element.2)
          }
          .map { (token, refreshToken, expiry) -> String? in
              return token
          }
          .catchError { e -> Observable<String?> in
              guard let error = e as? MoyaError else { throw e }
              guard case .statusCode(let response) = error else { throw e }

              // If we have 401 error - delete all credentials and handle logout
              if response.statusCode == 401 {
                  UserInfo.shared.invalidate()
                  Router.shared.popToLoginScreen()
              }
              throw error
          }

      return newTokenRequest
  }

  // Override request to inject XAppTokenRequest if needed
  override func request(token: MyService) -> Observable<Moya.Response> {
      let actualRequest = super.request(token)

      return self.XAppTokenRequest().flatMap { _ in
          actualRequest
      }
  }
}

創(chuàng)建自定義provider像往常一樣以同樣的方式:

let MyServiceProvider = OnlineProvider()

同樣 你也可以傳遞參數(shù)

let MyServiceProvider = OnlineProvider(endpointClosure: endpointClosure, plugins: [NetworkLogger()])

總結 這小節(jié)的核心:

1. 自定義Provider
2. 實現(xiàn)一個具體的session自動刷新的案例