//
|
// WithLatestFrom.swift
|
// RxSwift
|
//
|
// Created by Yury Korolev on 10/19/15.
|
// Copyright © 2015 Krunoslav Zaher. All rights reserved.
|
//
|
|
extension ObservableType {
|
|
/**
|
Merges two observable sequences into one observable sequence by combining each element from self with the latest element from the second source, if any.
|
|
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
|
|
- parameter second: Second observable source.
|
- parameter resultSelector: Function to invoke for each element from the self combined with the latest element from the second source, if any.
|
- returns: An observable sequence containing the result of combining each element of the self with the latest element from the second source, if any, using the specified result selector function.
|
*/
|
public func withLatestFrom<Source: ObservableConvertibleType, ResultType>(_ second: Source, resultSelector: @escaping (Element, Source.Element) throws -> ResultType) -> Observable<ResultType> {
|
return WithLatestFrom(first: self.asObservable(), second: second.asObservable(), resultSelector: resultSelector)
|
}
|
|
/**
|
Merges two observable sequences into one observable sequence by using latest element from the second sequence every time when `self` emits an element.
|
|
- seealso: [combineLatest operator on reactivex.io](http://reactivex.io/documentation/operators/combinelatest.html)
|
|
- parameter second: Second observable source.
|
- returns: An observable sequence containing the result of combining each element of the self with the latest element from the second source, if any, using the specified result selector function.
|
*/
|
public func withLatestFrom<Source: ObservableConvertibleType>(_ second: Source) -> Observable<Source.Element> {
|
return WithLatestFrom(first: self.asObservable(), second: second.asObservable(), resultSelector: { $1 })
|
}
|
}
|
|
final private class WithLatestFromSink<FirstType, SecondType, Observer: ObserverType>
|
: Sink<Observer>
|
, ObserverType
|
, LockOwnerType
|
, SynchronizedOnType {
|
typealias ResultType = Observer.Element
|
typealias Parent = WithLatestFrom<FirstType, SecondType, ResultType>
|
typealias Element = FirstType
|
|
private let _parent: Parent
|
|
fileprivate var _lock = RecursiveLock()
|
fileprivate var _latest: SecondType?
|
|
init(parent: Parent, observer: Observer, cancel: Cancelable) {
|
self._parent = parent
|
|
super.init(observer: observer, cancel: cancel)
|
}
|
|
func run() -> Disposable {
|
let sndSubscription = SingleAssignmentDisposable()
|
let sndO = WithLatestFromSecond(parent: self, disposable: sndSubscription)
|
|
sndSubscription.setDisposable(self._parent._second.subscribe(sndO))
|
let fstSubscription = self._parent._first.subscribe(self)
|
|
return Disposables.create(fstSubscription, sndSubscription)
|
}
|
|
func on(_ event: Event<Element>) {
|
self.synchronizedOn(event)
|
}
|
|
func _synchronized_on(_ event: Event<Element>) {
|
switch event {
|
case let .next(value):
|
guard let latest = self._latest else { return }
|
do {
|
let res = try self._parent._resultSelector(value, latest)
|
|
self.forwardOn(.next(res))
|
} catch let e {
|
self.forwardOn(.error(e))
|
self.dispose()
|
}
|
case .completed:
|
self.forwardOn(.completed)
|
self.dispose()
|
case let .error(error):
|
self.forwardOn(.error(error))
|
self.dispose()
|
}
|
}
|
}
|
|
final private class WithLatestFromSecond<FirstType, SecondType, Observer: ObserverType>
|
: ObserverType
|
, LockOwnerType
|
, SynchronizedOnType {
|
|
typealias ResultType = Observer.Element
|
typealias Parent = WithLatestFromSink<FirstType, SecondType, Observer>
|
typealias Element = SecondType
|
|
private let _parent: Parent
|
private let _disposable: Disposable
|
|
var _lock: RecursiveLock {
|
return self._parent._lock
|
}
|
|
init(parent: Parent, disposable: Disposable) {
|
self._parent = parent
|
self._disposable = disposable
|
}
|
|
func on(_ event: Event<Element>) {
|
self.synchronizedOn(event)
|
}
|
|
func _synchronized_on(_ event: Event<Element>) {
|
switch event {
|
case let .next(value):
|
self._parent._latest = value
|
case .completed:
|
self._disposable.dispose()
|
case let .error(error):
|
self._parent.forwardOn(.error(error))
|
self._parent.dispose()
|
}
|
}
|
}
|
|
final private class WithLatestFrom<FirstType, SecondType, ResultType>: Producer<ResultType> {
|
typealias ResultSelector = (FirstType, SecondType) throws -> ResultType
|
|
fileprivate let _first: Observable<FirstType>
|
fileprivate let _second: Observable<SecondType>
|
fileprivate let _resultSelector: ResultSelector
|
|
init(first: Observable<FirstType>, second: Observable<SecondType>, resultSelector: @escaping ResultSelector) {
|
self._first = first
|
self._second = second
|
self._resultSelector = resultSelector
|
}
|
|
override func run<Observer: ObserverType>(_ observer: Observer, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where Observer.Element == ResultType {
|
let sink = WithLatestFromSink(parent: self, observer: observer, cancel: cancel)
|
let subscription = sink.run()
|
return (sink: sink, subscription: subscription)
|
}
|
}
|
