Files
2025-08-13 09:26:42 +08:00

177 lines
3.3 KiB
C#

using System;
using UniRx.InternalUtil;
namespace UniRx
{
public sealed class Subject<T> : IDisposable, IObserver<T>, ISubject<T>, ISubject<T, T>, IObservable<T>, IOptimizedObservable<T>
{
private class Subscription : IDisposable
{
private readonly object gate = new object();
private Subject<T> parent;
private IObserver<T> unsubscribeTarget;
public Subscription(Subject<T> parent, IObserver<T> unsubscribeTarget)
{
this.parent = parent;
this.unsubscribeTarget = unsubscribeTarget;
}
public void Dispose()
{
lock (gate)
{
if (parent == null)
{
return;
}
lock (parent.observerLock)
{
ListObserver<T> listObserver = parent.outObserver as ListObserver<T>;
if (listObserver != null)
{
parent.outObserver = listObserver.Remove(unsubscribeTarget);
}
else
{
parent.outObserver = new EmptyObserver<T>();
}
unsubscribeTarget = null;
parent = null;
}
}
}
}
private object observerLock = new object();
private bool isStopped;
private bool isDisposed;
private Exception lastError;
private IObserver<T> outObserver = new EmptyObserver<T>();
public bool HasObservers
{
get
{
return !(outObserver is EmptyObserver<T>) && !isStopped && !isDisposed;
}
}
public void OnCompleted()
{
IObserver<T> observer;
lock (observerLock)
{
ThrowIfDisposed();
if (isStopped)
{
return;
}
observer = outObserver;
outObserver = new EmptyObserver<T>();
isStopped = true;
}
observer.OnCompleted();
}
public void OnError(Exception error)
{
if (error == null)
{
throw new ArgumentNullException("error");
}
IObserver<T> observer;
lock (observerLock)
{
ThrowIfDisposed();
if (isStopped)
{
return;
}
observer = outObserver;
outObserver = new EmptyObserver<T>();
isStopped = true;
lastError = error;
}
observer.OnError(error);
}
public void OnNext(T value)
{
outObserver.OnNext(value);
}
public IDisposable Subscribe(IObserver<T> observer)
{
if (observer == null)
{
throw new ArgumentNullException("observer");
}
Exception ex = null;
lock (observerLock)
{
ThrowIfDisposed();
if (!isStopped)
{
ListObserver<T> listObserver = outObserver as ListObserver<T>;
if (listObserver != null)
{
outObserver = listObserver.Add(observer);
}
else
{
IObserver<T> observer2 = outObserver;
if (observer2 is EmptyObserver<T>)
{
outObserver = observer;
}
else
{
outObserver = new ListObserver<T>(new ImmutableList<IObserver<T>>(new IObserver<T>[2] { observer2, observer }));
}
}
return new Subscription(this, observer);
}
ex = lastError;
}
if (ex != null)
{
observer.OnError(ex);
}
else
{
observer.OnCompleted();
}
return Disposable.Empty;
}
public void Dispose()
{
lock (observerLock)
{
isDisposed = true;
outObserver = new DisposedObserver<T>();
}
}
private void ThrowIfDisposed()
{
if (isDisposed)
{
throw new ObjectDisposedException(string.Empty);
}
}
public bool IsRequiredSubscribeOnCurrentThread()
{
return false;
}
}
}