Retry with RxSwift

RxSwiftPolling.swift

class AuthRetryToken {
    private(set) var retriesLeft = 3
    var canRetry: Bool {
        return retriesLeft > 0
    }
    
    func handleRetry() {
        guard canRetry else { return }
        retriesLeft -= 1
    }
}

extension Observable where E == UserSession {    
    func mapPollUser(with retryToken: AuthRetryToken) -> Observable<UserSession> {
        return flatMap { session -> Observable<UserSession> in
            return APIFactory.API().request(.myProfile)
                .flatMap { response -> Observable<UserSession> in
                    if response.statusCode == StatusCode.okay {
                        return Observable<Response>.just(response)
                            .mapModel(model: User.self)
                            .map { UserSession(token: session.token, user: $0) }
                    } else if retryToken.canRetry {
                        retryToken.handleRetry()
                        return self.delay(1.0, scheduler: MainScheduler.instance).mapPollUser(with: retryToken)
                    } else {
                        return Observable.error(AuthServiceError.fetchUserError)
                    }
            }
        }
    }
}

Usage.swift

createUser(with: accessToken)
  .do(onNext: { $0.persistAndMakeCurrent() })
  .mapPollUser(with: AuthRetryToken())