UniRx 에셋 추가
This commit is contained in:
@@ -0,0 +1,77 @@
|
||||
using System;
|
||||
using UniRx.Operators;
|
||||
|
||||
namespace UniRx.Operators
|
||||
{
|
||||
internal class RefCountObservable<T> : OperatorObservableBase<T>
|
||||
{
|
||||
readonly IConnectableObservable<T> source;
|
||||
readonly object gate = new object();
|
||||
int refCount = 0;
|
||||
IDisposable connection;
|
||||
|
||||
public RefCountObservable(IConnectableObservable<T> source)
|
||||
: base(source.IsRequiredSubscribeOnCurrentThread())
|
||||
{
|
||||
this.source = source;
|
||||
}
|
||||
|
||||
protected override IDisposable SubscribeCore(IObserver<T> observer, IDisposable cancel)
|
||||
{
|
||||
return new RefCount(this, observer, cancel).Run();
|
||||
}
|
||||
|
||||
class RefCount : OperatorObserverBase<T, T>
|
||||
{
|
||||
readonly RefCountObservable<T> parent;
|
||||
|
||||
public RefCount(RefCountObservable<T> parent, IObserver<T> observer, IDisposable cancel) : base(observer, cancel)
|
||||
{
|
||||
this.parent = parent;
|
||||
}
|
||||
|
||||
public IDisposable Run()
|
||||
{
|
||||
var subcription = parent.source.Subscribe(this);
|
||||
|
||||
lock (parent.gate)
|
||||
{
|
||||
if (++parent.refCount == 1)
|
||||
{
|
||||
parent.connection = parent.source.Connect();
|
||||
}
|
||||
}
|
||||
|
||||
return Disposable.Create(() =>
|
||||
{
|
||||
subcription.Dispose();
|
||||
|
||||
lock (parent.gate)
|
||||
{
|
||||
if (--parent.refCount == 0)
|
||||
{
|
||||
parent.connection.Dispose();
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
public override void OnNext(T value)
|
||||
{
|
||||
base.observer.OnNext(value);
|
||||
}
|
||||
|
||||
public override void OnError(Exception error)
|
||||
{
|
||||
try { observer.OnError(error); }
|
||||
finally { Dispose(); }
|
||||
}
|
||||
|
||||
public override void OnCompleted()
|
||||
{
|
||||
try { observer.OnCompleted(); }
|
||||
finally { Dispose(); }
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user