WithLatestFrom<TFirst, TSecond, TResult>
sealed class WithLatestFrom<TFirst, TSecond, TResult> : Producer<TResult, _<TFirst, TSecond, TResult>>
using System.Reactive.Disposables;
namespace System.Reactive.Linq.ObservableImpl
{
internal sealed class WithLatestFrom<TFirst, TSecond, TResult> : Producer<TResult, WithLatestFrom<TFirst, TSecond, TResult>._>
{
internal sealed class _ : IdentitySink<TResult>
{
private sealed class FirstObserver : IObserver<TFirst>
{
private readonly _ _parent;
public FirstObserver(_ parent)
{
_parent = parent;
}
public void OnCompleted()
{
lock (_parent._gate) {
_parent.ForwardOnCompleted();
}
}
public void OnError(Exception error)
{
lock (_parent._gate) {
_parent.ForwardOnError(error);
}
}
public void OnNext(TFirst value)
{
if (_parent._hasLatest) {
TSecond latest = default(TSecond);
lock (_parent._latestGate) {
latest = _parent._latest;
}
TResult val = default(TResult);
try {
val = _parent._resultSelector(value, latest);
} catch (Exception error) {
lock (_parent._gate) {
_parent.ForwardOnError(error);
}
return;
}
lock (_parent._gate) {
_parent.ForwardOnNext(val);
}
}
}
}
private sealed class SecondObserver : IObserver<TSecond>
{
private readonly _ _parent;
public SecondObserver(_ parent)
{
_parent = parent;
}
public void OnCompleted()
{
Disposable.TryDispose(ref _parent._secondDisposable);
}
public void OnError(Exception error)
{
lock (_parent._gate) {
_parent.ForwardOnError(error);
}
}
public void OnNext(TSecond value)
{
lock (_parent._latestGate) {
_parent._latest = value;
}
if (!_parent._hasLatest)
_parent._hasLatest = true;
}
}
private readonly Func<TFirst, TSecond, TResult> _resultSelector;
private object _gate;
private volatile bool _hasLatest;
private TSecond _latest;
private object _latestGate;
private IDisposable _secondDisposable;
public _(Func<TFirst, TSecond, TResult> resultSelector, IObserver<TResult> observer)
: base(observer)
{
_resultSelector = resultSelector;
}
public void Run(IObservable<TFirst> first, IObservable<TSecond> second)
{
_gate = new object();
_latestGate = new object();
FirstObserver observer = new FirstObserver(this);
SecondObserver observer2 = new SecondObserver(this);
Disposable.SetSingle(ref _secondDisposable, ObservableExtensions.SubscribeSafe<TSecond>(second, (IObserver<TSecond>)observer2));
SetUpstream(ObservableExtensions.SubscribeSafe<TFirst>(first, (IObserver<TFirst>)observer));
}
protected override void Dispose(bool disposing)
{
if (disposing)
Disposable.TryDispose(ref _secondDisposable);
base.Dispose(disposing);
}
}
private readonly IObservable<TFirst> _first;
private readonly IObservable<TSecond> _second;
private readonly Func<TFirst, TSecond, TResult> _resultSelector;
public WithLatestFrom(IObservable<TFirst> first, IObservable<TSecond> second, Func<TFirst, TSecond, TResult> resultSelector)
{
_first = first;
_second = second;
_resultSelector = resultSelector;
}
protected override _ CreateSink(IObserver<TResult> observer)
{
return new _(_resultSelector, observer);
}
protected override void Run(_ sink)
{
sink.Run(_first, _second);
}
}
}