using Cysharp.Threading.Tasks.Internal;
|
using System.Threading;
|
|
namespace Cysharp.Threading.Tasks.Linq
|
{
|
public static partial class UniTaskAsyncEnumerable
|
{
|
public static IUniTaskAsyncEnumerable<TValue> Return<TValue>(TValue value)
|
{
|
return new Return<TValue>(value);
|
}
|
}
|
|
internal class Return<TValue> : IUniTaskAsyncEnumerable<TValue>
|
{
|
readonly TValue value;
|
|
public Return(TValue value)
|
{
|
this.value = value;
|
}
|
|
public IUniTaskAsyncEnumerator<TValue> GetAsyncEnumerator(CancellationToken cancellationToken = default)
|
{
|
return new _Return(value, cancellationToken);
|
}
|
|
class _Return : IUniTaskAsyncEnumerator<TValue>
|
{
|
readonly TValue value;
|
CancellationToken cancellationToken;
|
|
bool called;
|
|
public _Return(TValue value, CancellationToken cancellationToken)
|
{
|
this.value = value;
|
this.cancellationToken = cancellationToken;
|
this.called = false;
|
}
|
|
public TValue Current => value;
|
|
public UniTask<bool> MoveNextAsync()
|
{
|
cancellationToken.ThrowIfCancellationRequested();
|
|
if (!called)
|
{
|
called = true;
|
return CompletedTasks.True;
|
}
|
|
return CompletedTasks.False;
|
}
|
|
public UniTask DisposeAsync()
|
{
|
return default;
|
}
|
}
|
}
|
}
|
