using System.Reactive.Concurrency; using System.Reactive.Disposables; namespace System.Reactive.Linq.ObservableImpl { internal sealed class TakeUntil<TSource> : Producer<TSource, TakeUntil<TSource>._> { internal sealed class _ : IdentitySink<TSource> { private IDisposable _timerDisposable; private int _wip; private Exception _error; public _(IObserver<TSource> observer) : base(observer) { } public void Run(TakeUntil<TSource> parent) { Disposable.SetSingle(ref _timerDisposable, Scheduler.ScheduleAction<_>(parent._scheduler, this, parent._endTime, (Action<_>)delegate(_ state) { state.Tick(); })); Run(parent._source); } protected override void Dispose(bool disposing) { if (disposing) Disposable.TryDispose(ref _timerDisposable); base.Dispose(disposing); } private void Tick() { OnCompleted(); } public override void OnNext(TSource value) { HalfSerializer.ForwardOnNext<TSource>((ISink<TSource>)this, value, ref _wip, ref _error); } public override void OnError(Exception error) { HalfSerializer.ForwardOnError<TSource>((ISink<TSource>)this, error, ref _wip, ref _error); } public override void OnCompleted() { HalfSerializer.ForwardOnCompleted<TSource>((ISink<TSource>)this, ref _wip, ref _error); } } private readonly IObservable<TSource> _source; private readonly DateTimeOffset _endTime; internal readonly IScheduler _scheduler; public TakeUntil(IObservable<TSource> source, DateTimeOffset endTime, IScheduler scheduler) { _source = source; _endTime = endTime; _scheduler = scheduler; } public IObservable<TSource> Combine(DateTimeOffset endTime) { if (_endTime <= endTime) return this; return new TakeUntil<TSource>(_source, endTime, _scheduler); } protected override _ CreateSink(IObserver<TSource> observer) { return new _(observer); } protected override void Run(_ sink) { sink.Run(this); } } }