Throttle<TSource, TThrottle>
using System.Reactive.Disposables;
namespace System.Reactive.Linq.ObservableImpl
{
internal sealed class Throttle<TSource, TThrottle> : Producer<TSource, Throttle<TSource, TThrottle>._>
{
internal sealed class _ : IdentitySink<TSource>
{
private sealed class ThrottleObserver : SafeObserver<TThrottle>
{
private readonly _ _parent;
private readonly TSource _value;
private readonly ulong _currentid;
public ThrottleObserver(_ parent, TSource value, ulong currentid)
{
_parent = parent;
_value = value;
_currentid = currentid;
}
public override void OnNext(TThrottle value)
{
lock (_parent._gate) {
if (_parent._hasValue && _parent._id == _currentid)
_parent.ForwardOnNext(_value);
_parent._hasValue = false;
Dispose();
}
}
public override void OnError(Exception error)
{
lock (_parent._gate) {
_parent.ForwardOnError(error);
}
}
public override void OnCompleted()
{
lock (_parent._gate) {
if (_parent._hasValue && _parent._id == _currentid)
_parent.ForwardOnNext(_value);
_parent._hasValue = false;
Dispose();
}
}
}
private readonly Func<TSource, IObservable<TThrottle>> _throttleSelector;
private readonly object _gate = new object();
private TSource _value;
private bool _hasValue;
private IDisposable _serialCancelable;
private ulong _id;
public _(Throttle<TSource, TThrottle> parent, IObserver<TSource> observer)
: base(observer)
{
_throttleSelector = parent._throttleSelector;
}
protected override void Dispose(bool disposing)
{
if (disposing)
Disposable.TryDispose(ref _serialCancelable);
base.Dispose(disposing);
}
public override void OnNext(TSource value)
{
IObservable<TThrottle> observable = null;
try {
observable = _throttleSelector(value);
} catch (Exception error) {
lock (_gate) {
ForwardOnError(error);
}
return;
}
ulong id = default(ulong);
lock (_gate) {
_hasValue = true;
_value = value;
_id++;
id = _id;
}
Disposable.TrySetSerial(ref _serialCancelable, null);
ThrottleObserver throttleObserver = new ThrottleObserver(this, value, id);
throttleObserver.SetResource(ObservableExtensions.SubscribeSafe<TThrottle>(observable, (IObserver<TThrottle>)throttleObserver));
Disposable.TrySetSerial(ref _serialCancelable, throttleObserver);
}
public override void OnError(Exception error)
{
Disposable.TryDispose(ref _serialCancelable);
lock (_gate) {
ForwardOnError(error);
_hasValue = false;
_id++;
}
}
public override void OnCompleted()
{
Disposable.TryDispose(ref _serialCancelable);
lock (_gate) {
if (_hasValue)
ForwardOnNext(_value);
ForwardOnCompleted();
_hasValue = false;
_id++;
}
}
}
private readonly IObservable<TSource> _source;
private readonly Func<TSource, IObservable<TThrottle>> _throttleSelector;
public Throttle(IObservable<TSource> source, Func<TSource, IObservable<TThrottle>> throttleSelector)
{
_source = source;
_throttleSelector = throttleSelector;
}
protected override _ CreateSink(IObserver<TSource> observer)
{
return new _(this, observer);
}
protected override void Run(_ sink)
{
sink.Run(_source);
}
}
}