MoyaProvider+Rx.swift 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. import Foundation
  2. import RxSwift
  3. #if !COCOAPODS
  4. import Moya
  5. #endif
  6. extension MoyaProvider: ReactiveCompatible {}
  7. public extension Reactive where Base: MoyaProviderType {
  8. /// Designated request-making method.
  9. ///
  10. /// - Parameters:
  11. /// - token: Entity, which provides specifications necessary for a `MoyaProvider`.
  12. /// - callbackQueue: Callback queue. If nil - queue from provider initializer will be used.
  13. /// - Returns: Single response object.
  14. func request(_ token: Base.Target, callbackQueue: DispatchQueue? = nil) -> Single<Response> {
  15. Single.create { [weak base] single in
  16. let cancellableToken = base?.request(token, callbackQueue: callbackQueue, progress: nil) { result in
  17. switch result {
  18. case let .success(response):
  19. single(.success(response))
  20. case let .failure(error):
  21. single(.failure(error))
  22. }
  23. }
  24. return Disposables.create {
  25. cancellableToken?.cancel()
  26. }
  27. }
  28. }
  29. /// Designated request-making method with progress.
  30. func requestWithProgress(_ token: Base.Target, callbackQueue: DispatchQueue? = nil) -> Observable<ProgressResponse> {
  31. let progressBlock = AnyObserver<ProgressResponse>.onNext
  32. let response: Observable<ProgressResponse> = Observable.create { [weak base] observer in
  33. let cancellableToken = base?.request(token, callbackQueue: callbackQueue, progress: progressBlock(observer)) { result in
  34. switch result {
  35. case .success:
  36. observer.onCompleted()
  37. case let .failure(error):
  38. observer.onError(error)
  39. }
  40. }
  41. return Disposables.create {
  42. cancellableToken?.cancel()
  43. }
  44. }
  45. // Accumulate all progress and combine them when the result comes
  46. return response.scan(ProgressResponse()) { last, progress in
  47. let progressObject = progress.progressObject ?? last.progressObject
  48. let response = progress.response ?? last.response
  49. return ProgressResponse(progress: progressObject, response: response)
  50. }
  51. }
  52. }