Kit.Core/LibExternal/System.Reactive/Linq/Observable/SkipLast.cs

116 lines
3.7 KiB
C#

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT License.
// See the LICENSE file in the project root for more information.
using System.Collections.Generic;
using System.Reactive.Concurrency;
namespace System.Reactive.Linq.ObservableImpl
{
internal static class SkipLast<TSource>
{
internal sealed class Count : Producer<TSource, Count._>
{
private readonly IObservable<TSource> _source;
private readonly int _count;
public Count(IObservable<TSource> source, int count)
{
_source = source;
_count = count;
}
protected override _ CreateSink(IObserver<TSource> observer) => new(_count, observer);
protected override void Run(_ sink) => sink.Run(_source);
internal sealed class _ : IdentitySink<TSource>
{
private readonly int _count;
private readonly Queue<TSource> _queue;
public _(int count, IObserver<TSource> observer)
: base(observer)
{
_count = count;
_queue = new Queue<TSource>();
}
public override void OnNext(TSource value)
{
_queue.Enqueue(value);
if (_queue.Count > _count)
{
ForwardOnNext(_queue.Dequeue());
}
}
}
}
internal sealed class Time : Producer<TSource, Time._>
{
private readonly IObservable<TSource> _source;
private readonly TimeSpan _duration;
private readonly IScheduler _scheduler;
public Time(IObservable<TSource> source, TimeSpan duration, IScheduler scheduler)
{
_source = source;
_duration = duration;
_scheduler = scheduler;
}
protected override _ CreateSink(IObserver<TSource> observer) => new(_duration, observer);
protected override void Run(_ sink) => sink.Run(this);
internal sealed class _ : IdentitySink<TSource>
{
private readonly TimeSpan _duration;
private readonly Queue<Reactive.TimeInterval<TSource>> _queue;
public _(TimeSpan duration, IObserver<TSource> observer)
: base(observer)
{
_duration = duration;
_queue = new Queue<Reactive.TimeInterval<TSource>>();
}
private IStopwatch? _watch;
public void Run(Time parent)
{
_watch = parent._scheduler.StartStopwatch();
SetUpstream(parent._source.SubscribeSafe(this));
}
public override void OnNext(TSource value)
{
var now = _watch!.Elapsed;
_queue.Enqueue(new Reactive.TimeInterval<TSource>(value, now));
while (_queue.Count > 0 && now - _queue.Peek().Interval >= _duration)
{
ForwardOnNext(_queue.Dequeue().Value);
}
}
public override void OnCompleted()
{
var now = _watch!.Elapsed;
while (_queue.Count > 0 && now - _queue.Peek().Interval >= _duration)
{
ForwardOnNext(_queue.Dequeue().Value);
}
ForwardOnCompleted();
}
}
}
}
}