using System.Reactive.Disposables; namespace System.Reactive.Linq.ObservableImpl { internal class AddRef<TSource> : Producer<TSource, AddRef<TSource>._> { internal sealed class _ : Sink<TSource>, IObserver<TSource> { public _(IObserver<TSource> observer, IDisposable cancel) : base(observer, cancel) { } public void OnNext(TSource value) { _observer.OnNext(value); } public void OnError(Exception error) { _observer.OnError(error); base.Dispose(); } public void OnCompleted() { _observer.OnCompleted(); base.Dispose(); } } private readonly IObservable<TSource> _source; private readonly RefCountDisposable _refCount; public AddRef(IObservable<TSource> source, RefCountDisposable refCount) { _source = source; _refCount = refCount; } protected override _ CreateSink(IObserver<TSource> observer, IDisposable cancel) { return new _(observer, StableCompositeDisposable.Create(_refCount.GetDisposable(), cancel)); } protected override IDisposable Run(_ sink) { return ObservableExtensions.SubscribeSafe<TSource>(_source, (IObserver<TSource>)sink); } } }