Window<TSource, TWindowClosing>
using System.Reactive.Concurrency;
using System.Reactive.Disposables;
using System.Reactive.Subjects;
namespace System.Reactive.Linq.ObservableImpl
{
internal static class Window<TSource, TWindowClosing>
{
internal sealed class Selector : Producer<IObservable<TSource>, Selector._>
{
internal sealed class _ : Sink<IObservable<TSource>>, IObserver<TSource>
{
private sealed class WindowClosingObserver : IObserver<TWindowClosing>
{
private readonly _ _parent;
private readonly IDisposable _self;
public WindowClosingObserver(_ parent, IDisposable self)
{
_parent = parent;
_self = self;
}
public void OnNext(TWindowClosing value)
{
_parent.CloseWindow(_self);
}
public void OnError(Exception error)
{
_parent.OnError(error);
}
public void OnCompleted()
{
_parent.CloseWindow(_self);
}
}
private readonly object _gate = new object();
private readonly AsyncLock _windowGate = new AsyncLock();
private readonly SerialDisposable _m = new SerialDisposable();
private readonly Func<IObservable<TWindowClosing>> _windowClosingSelector;
private ISubject<TSource> _window;
private RefCountDisposable _refCountDisposable;
public _(Selector parent, IObserver<IObservable<TSource>> observer, IDisposable cancel)
: base(observer, cancel)
{
_windowClosingSelector = parent._windowClosingSelector;
}
public IDisposable Run(IObservable<TSource> source)
{
_window = new Subject<TSource>();
CompositeDisposable compositeDisposable = new CompositeDisposable(2) {
_m
};
_refCountDisposable = new RefCountDisposable(compositeDisposable);
WindowObservable<TSource> value = new WindowObservable<TSource>(_window, _refCountDisposable);
_observer.OnNext(value);
compositeDisposable.Add(ObservableExtensions.SubscribeSafe<TSource>(source, (IObserver<TSource>)this));
_windowGate.Wait(CreateWindowClose);
return _refCountDisposable;
}
private void CreateWindowClose()
{
IObservable<TWindowClosing> observable = null;
try {
observable = _windowClosingSelector();
} catch (Exception error) {
lock (_gate) {
_observer.OnError(error);
base.Dispose();
}
return;
}
SingleAssignmentDisposable singleAssignmentDisposable = new SingleAssignmentDisposable();
_m.Disposable = singleAssignmentDisposable;
singleAssignmentDisposable.Disposable = ObservableExtensions.SubscribeSafe<TWindowClosing>(observable, (IObserver<TWindowClosing>)new WindowClosingObserver(this, singleAssignmentDisposable));
}
private void CloseWindow(IDisposable closingSubscription)
{
closingSubscription.Dispose();
lock (_gate) {
_window.OnCompleted();
_window = new Subject<TSource>();
WindowObservable<TSource> value = new WindowObservable<TSource>(_window, _refCountDisposable);
_observer.OnNext(value);
}
_windowGate.Wait(CreateWindowClose);
}
public void OnNext(TSource value)
{
lock (_gate) {
_window.OnNext(value);
}
}
public void OnError(Exception error)
{
lock (_gate) {
_window.OnError(error);
_observer.OnError(error);
base.Dispose();
}
}
public void OnCompleted()
{
lock (_gate) {
_window.OnCompleted();
_observer.OnCompleted();
base.Dispose();
}
}
}
private readonly IObservable<TSource> _source;
private readonly Func<IObservable<TWindowClosing>> _windowClosingSelector;
public Selector(IObservable<TSource> source, Func<IObservable<TWindowClosing>> windowClosingSelector)
{
_source = source;
_windowClosingSelector = windowClosingSelector;
}
protected override _ CreateSink(IObserver<IObservable<TSource>> observer, IDisposable cancel)
{
return new _(this, observer, cancel);
}
protected override IDisposable Run(_ sink)
{
return sink.Run(_source);
}
}
internal sealed class Boundaries : Producer<IObservable<TSource>, Boundaries._>
{
internal sealed class _ : Sink<IObservable<TSource>>, IObserver<TSource>
{
private sealed class WindowClosingObserver : IObserver<TWindowClosing>
{
private readonly _ _parent;
public WindowClosingObserver(_ parent)
{
_parent = parent;
}
public void OnNext(TWindowClosing value)
{
lock (_parent._gate) {
_parent._window.OnCompleted();
_parent._window = new Subject<TSource>();
WindowObservable<TSource> value2 = new WindowObservable<TSource>(_parent._window, _parent._refCountDisposable);
_parent._observer.OnNext(value2);
}
}
public void OnError(Exception error)
{
_parent.OnError(error);
}
public void OnCompleted()
{
_parent.OnCompleted();
}
}
private readonly object _gate = new object();
private readonly IObservable<TWindowClosing> _windowBoundaries;
private ISubject<TSource> _window;
private RefCountDisposable _refCountDisposable;
public _(Boundaries parent, IObserver<IObservable<TSource>> observer, IDisposable cancel)
: base(observer, cancel)
{
_windowBoundaries = parent._windowBoundaries;
}
public IDisposable Run(Boundaries parent)
{
_window = new Subject<TSource>();
CompositeDisposable compositeDisposable = new CompositeDisposable(2);
_refCountDisposable = new RefCountDisposable(compositeDisposable);
WindowObservable<TSource> value = new WindowObservable<TSource>(_window, _refCountDisposable);
_observer.OnNext(value);
compositeDisposable.Add(ObservableExtensions.SubscribeSafe<TSource>(parent._source, (IObserver<TSource>)this));
compositeDisposable.Add(ObservableExtensions.SubscribeSafe<TWindowClosing>(parent._windowBoundaries, (IObserver<TWindowClosing>)new WindowClosingObserver(this)));
return _refCountDisposable;
}
public void OnNext(TSource value)
{
lock (_gate) {
_window.OnNext(value);
}
}
public void OnError(Exception error)
{
lock (_gate) {
_window.OnError(error);
_observer.OnError(error);
base.Dispose();
}
}
public void OnCompleted()
{
lock (_gate) {
_window.OnCompleted();
_observer.OnCompleted();
base.Dispose();
}
}
}
private readonly IObservable<TSource> _source;
private readonly IObservable<TWindowClosing> _windowBoundaries;
public Boundaries(IObservable<TSource> source, IObservable<TWindowClosing> windowBoundaries)
{
_source = source;
_windowBoundaries = windowBoundaries;
}
protected override _ CreateSink(IObserver<IObservable<TSource>> observer, IDisposable cancel)
{
return new _(this, observer, cancel);
}
protected override IDisposable Run(_ sink)
{
return sink.Run(this);
}
}
}
}