Rate Limit with RxSwift

Question

私はHTTPクライアントでレート制限を実装するスマートな方法を探しています。 APIのレート制限がいずれかのリソースで1秒あたり5リクエストであるとします。今実装は次のようになります。

final class HTTPClient: HTTPClientProtocol { 

    func getUser() -> Observable<User> { 
     return Observable<User>.create { (observer) -> Disposable in 
      ... 
     } 
    } 

    func getProfile() -> Observable<Profile> { 
     return Observable<Profile>.create { (observer) -> Disposable in 
      ... 
     } 
    } 

    func getMessages() -> Observable<Messages> { 
     return Observable<Messages>.create { (observer) -> Disposable in 
      ... 
     } 
    } 

    func getFriends() -> Observable<Friends> { 
     return Observable<Friends>.create { (observer) -> Disposable in 
      ... 
     } 
    } 

} 

今、理想的に私は、必要に応じて、すべてのレート制限を気にすることなく、アプリケーション全体でこれらのメソッドを使用したいと思います。

1秒あたり5リクエストの例に戻る：最初の5つのリクエストはすぐに実行できます。しかしその後のすべてのリクエストは待たなければなりません。したがって、1秒のウィンドウ内では、最大で5回のリクエストを実行できます。他のすべてのリクエストは待機する必要があります。

これをRxSwiftで実行するスマートな方法はありますか？

Answer 1

カスタムスケジューラが必要です。

final class DelayScheduler: ImmediateSchedulerType { 

    init(delay: TimeInterval, queue: DispatchQueue = .main) { 
     self.queue = queue 
     dispatchDelay = delay 
    } 

    func schedule<StateType>(_ state: StateType, action: @escaping (StateType) -> Disposable) -> Disposable { 
     let cancel = SingleAssignmentDisposable() 
     lastDispatch = max(lastDispatch + dispatchDelay, .now()) 
     queue.asyncAfter(deadline: lastDispatch) { 
      guard cancel.isDisposed == false else { return } 
      cancel.setDisposable(action(state)) 
     } 
     return cancel 
    } 

    var lastDispatch: DispatchTime = .now() 
    let queue: DispatchQueue 
    let dispatchDelay: TimeInterval 
} 

その後、あなたはそのすべての観測は、このスケジューラに加入することによって、あなたのサービスを実装する：[RxSwiftとサービスクラスへの同時アクセスを制限する]（http://stackoverflow.comの

final class HTTPClient: HTTPClientProtocol { 

    func getUser() -> Observable<User> { 
     return Observable<User>.create { (observer) -> Disposable in 
      ... 
     }.subscribeOn(scheduler) 
    } 

    func getProfile() -> Observable<Profile> { 
     return Observable<Profile>.create { (observer) -> Disposable in 
      ... 
     }.subscribeOn(scheduler) 
    } 

    func getMessages() -> Observable<Messages> { 
     return Observable<Messages>.create { (observer) -> Disposable in 
      ... 
     }.subscribeOn(scheduler) 
    } 

    func getFriends() -> Observable<Friends> { 
     return Observable<Friends>.create { (observer) -> Disposable in 
      ... 
     }.subscribeOn(scheduler) 
    } 

    let scheduler = DelayScheduler(delay: 0.5) 
}