Sample<TSource, TSample>
using System.Reactive.Disposables;
namespace System.Reactive.Linq.ObservableImpl
{
internal sealed class Sample<TSource, TSample> : Producer<TSource, Sample<TSource, TSample>._>
{
internal sealed class _ : IdentitySink<TSource>
{
private sealed class SampleObserver : IObserver<TSample>
{
private readonly _ _parent;
public SampleObserver(_ parent)
{
_parent = parent;
}
public void OnNext(TSample value)
{
lock (_parent._gate) {
if (_parent._hasValue) {
_parent._hasValue = false;
_parent.ForwardOnNext(_parent._value);
}
if (_parent._sourceAtEnd)
_parent.ForwardOnCompleted();
}
}
public void OnError(Exception error)
{
lock (_parent._gate) {
_parent.ForwardOnError(error);
}
}
public void OnCompleted()
{
lock (_parent._gate) {
_parent._samplerAtEnd = true;
if (_parent._hasValue) {
_parent._hasValue = false;
_parent.ForwardOnNext(_parent._value);
}
if (_parent._sourceAtEnd)
_parent.ForwardOnCompleted();
else
Disposable.TryDispose(ref _parent._samplerDisposable);
}
}
}
private readonly object _gate = new object();
private IDisposable _sourceDisposable;
private IDisposable _samplerDisposable;
private bool _hasValue;
private TSource _value;
private bool _sourceAtEnd;
private bool _samplerAtEnd;
public _(IObserver<TSource> observer)
: base(observer)
{
}
public void Run(Sample<TSource, TSample> parent)
{
Disposable.SetSingle(ref _sourceDisposable, ObservableExtensions.SubscribeSafe<TSource>(parent._source, (IObserver<TSource>)this));
Disposable.SetSingle(ref _samplerDisposable, ObservableExtensions.SubscribeSafe<TSample>(parent._sampler, (IObserver<TSample>)new SampleObserver(this)));
}
protected override void Dispose(bool disposing)
{
if (disposing) {
Disposable.TryDispose(ref _sourceDisposable);
Disposable.TryDispose(ref _samplerDisposable);
}
base.Dispose(disposing);
}
public override void OnNext(TSource value)
{
lock (_gate) {
_hasValue = true;
_value = value;
}
}
public override void OnError(Exception error)
{
lock (_gate) {
ForwardOnError(error);
}
}
public override void OnCompleted()
{
lock (_gate) {
_sourceAtEnd = true;
if (_samplerAtEnd)
ForwardOnCompleted();
else
Disposable.TryDispose(ref _sourceDisposable);
}
}
}
private readonly IObservable<TSource> _source;
private readonly IObservable<TSample> _sampler;
public Sample(IObservable<TSource> source, IObservable<TSample> sampler)
{
_source = source;
_sampler = sampler;
}
protected override _ CreateSink(IObserver<TSource> observer)
{
return new _(observer);
}
protected override void Run(_ sink)
{
sink.Run(this);
}
}
}