79 lines
2.6 KiB
C#
79 lines
2.6 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.Reactive.Disposables;
|
|
using System.Threading;
|
|
|
|
namespace System.Reactive.Linq.ObservableImpl
|
|
{
|
|
internal sealed class Finally<TSource> : Producer<TSource, Finally<TSource>._>
|
|
{
|
|
private readonly IObservable<TSource> _source;
|
|
private readonly Action _finallyAction;
|
|
|
|
public Finally(IObservable<TSource> source, Action finallyAction)
|
|
{
|
|
_source = source;
|
|
_finallyAction = finallyAction;
|
|
}
|
|
|
|
protected override _ CreateSink(IObserver<TSource> observer) => new(_finallyAction, observer);
|
|
|
|
protected override void Run(_ sink) => sink.Run(_source);
|
|
|
|
internal sealed class _ : IdentitySink<TSource>
|
|
{
|
|
private readonly Action _finallyAction;
|
|
private IDisposable? _sourceDisposable;
|
|
|
|
public _(Action finallyAction, IObserver<TSource> observer)
|
|
: base(observer)
|
|
{
|
|
_finallyAction = finallyAction;
|
|
}
|
|
|
|
public override void Run(IObservable<TSource> source)
|
|
{
|
|
var d = source.SubscribeSafe(this);
|
|
|
|
if (Interlocked.CompareExchange(ref _sourceDisposable, d, null) == BooleanDisposable.True)
|
|
{
|
|
// The Dispose(bool) methode was already called before the
|
|
// subscription could be assign, hence the subscription
|
|
// needs to be diposed here and the action needs to be invoked.
|
|
try
|
|
{
|
|
d.Dispose();
|
|
}
|
|
finally
|
|
{
|
|
_finallyAction();
|
|
}
|
|
}
|
|
}
|
|
|
|
protected override void Dispose(bool disposing)
|
|
{
|
|
base.Dispose(disposing);
|
|
|
|
if (disposing)
|
|
{
|
|
var d = Interlocked.Exchange(ref _sourceDisposable, BooleanDisposable.True);
|
|
if (d != BooleanDisposable.True && d != null)
|
|
{
|
|
try
|
|
{
|
|
d.Dispose();
|
|
}
|
|
finally
|
|
{
|
|
_finallyAction();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|