|
|
|
@@ -7,6 +7,7 @@ using System;
|
|
|
|
|
using System.Runtime.CompilerServices;
|
|
|
|
|
using System.Runtime.ExceptionServices;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
using UnityEngine.AddressableAssets;
|
|
|
|
|
using UnityEngine.ResourceManagement.AsyncOperations;
|
|
|
|
|
|
|
|
|
|
namespace Cysharp.Threading.Tasks
|
|
|
|
@@ -20,17 +21,12 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
return ToUniTask(handle).GetAwaiter();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static UniTask WithCancellation(this AsyncOperationHandle handle, CancellationToken cancellationToken)
|
|
|
|
|
public static UniTask WithCancellation(this AsyncOperationHandle handle, CancellationToken cancellationToken, bool cancelImmediately = false, bool autoReleaseWhenCanceled = false)
|
|
|
|
|
{
|
|
|
|
|
return ToUniTask(handle, cancellationToken: cancellationToken);
|
|
|
|
|
return ToUniTask(handle, cancellationToken: cancellationToken, cancelImmediately: cancelImmediately, autoReleaseWhenCanceled: autoReleaseWhenCanceled);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static UniTask WithCancellation(this AsyncOperationHandle handle, CancellationToken cancellationToken, bool cancelImmediately)
|
|
|
|
|
{
|
|
|
|
|
return ToUniTask(handle, cancellationToken: cancellationToken, cancelImmediately: cancelImmediately);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static UniTask ToUniTask(this AsyncOperationHandle handle, IProgress<float> progress = null, PlayerLoopTiming timing = PlayerLoopTiming.Update, CancellationToken cancellationToken = default(CancellationToken), bool cancelImmediately = false)
|
|
|
|
|
public static UniTask ToUniTask(this AsyncOperationHandle handle, IProgress<float> progress = null, PlayerLoopTiming timing = PlayerLoopTiming.Update, CancellationToken cancellationToken = default(CancellationToken), bool cancelImmediately = false, bool autoReleaseWhenCanceled = false)
|
|
|
|
|
{
|
|
|
|
|
if (cancellationToken.IsCancellationRequested) return UniTask.FromCanceled(cancellationToken);
|
|
|
|
|
|
|
|
|
@@ -49,7 +45,7 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
return UniTask.CompletedTask;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new UniTask(AsyncOperationHandleConfiguredSource.Create(handle, timing, progress, cancellationToken, cancelImmediately, out var token), token);
|
|
|
|
|
return new UniTask(AsyncOperationHandleConfiguredSource.Create(handle, timing, progress, cancellationToken, cancelImmediately, autoReleaseWhenCanceled, out var token), token);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public struct AsyncOperationHandleAwaiter : ICriticalNotifyCompletion
|
|
|
|
@@ -108,21 +104,23 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
TaskPool.RegisterSizeGetter(typeof(AsyncOperationHandleConfiguredSource), () => pool.Size);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
readonly Action<AsyncOperationHandle> continuationAction;
|
|
|
|
|
readonly Action<AsyncOperationHandle> completedCallback;
|
|
|
|
|
AsyncOperationHandle handle;
|
|
|
|
|
CancellationToken cancellationToken;
|
|
|
|
|
CancellationTokenRegistration cancellationTokenRegistration;
|
|
|
|
|
IProgress<float> progress;
|
|
|
|
|
bool autoReleaseWhenCanceled;
|
|
|
|
|
bool cancelImmediately;
|
|
|
|
|
bool completed;
|
|
|
|
|
|
|
|
|
|
UniTaskCompletionSourceCore<AsyncUnit> core;
|
|
|
|
|
|
|
|
|
|
AsyncOperationHandleConfiguredSource()
|
|
|
|
|
{
|
|
|
|
|
continuationAction = Continuation;
|
|
|
|
|
completedCallback = HandleCompleted;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static IUniTaskSource Create(AsyncOperationHandle handle, PlayerLoopTiming timing, IProgress<float> progress, CancellationToken cancellationToken, bool cancelImmediately, out short token)
|
|
|
|
|
public static IUniTaskSource Create(AsyncOperationHandle handle, PlayerLoopTiming timing, IProgress<float> progress, CancellationToken cancellationToken, bool cancelImmediately, bool autoReleaseWhenCanceled, out short token)
|
|
|
|
|
{
|
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
|
|
|
{
|
|
|
|
@@ -137,6 +135,8 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
result.handle = handle;
|
|
|
|
|
result.progress = progress;
|
|
|
|
|
result.cancellationToken = cancellationToken;
|
|
|
|
|
result.cancelImmediately = cancelImmediately;
|
|
|
|
|
result.autoReleaseWhenCanceled = autoReleaseWhenCanceled;
|
|
|
|
|
result.completed = false;
|
|
|
|
|
|
|
|
|
|
if (cancelImmediately && cancellationToken.CanBeCanceled)
|
|
|
|
@@ -144,6 +144,10 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
result.cancellationTokenRegistration = cancellationToken.RegisterWithoutCaptureExecutionContext(state =>
|
|
|
|
|
{
|
|
|
|
|
var promise = (AsyncOperationHandleConfiguredSource)state;
|
|
|
|
|
if (promise.autoReleaseWhenCanceled && promise.handle.IsValid())
|
|
|
|
|
{
|
|
|
|
|
Addressables.Release(promise.handle);
|
|
|
|
|
}
|
|
|
|
|
promise.core.TrySetCanceled(promise.cancellationToken);
|
|
|
|
|
}, result);
|
|
|
|
|
}
|
|
|
|
@@ -152,41 +156,56 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
|
|
|
|
|
PlayerLoopHelper.AddAction(timing, result);
|
|
|
|
|
|
|
|
|
|
handle.Completed += result.continuationAction;
|
|
|
|
|
handle.Completed += result.completedCallback;
|
|
|
|
|
|
|
|
|
|
token = result.core.Version;
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void Continuation(AsyncOperationHandle _)
|
|
|
|
|
void HandleCompleted(AsyncOperationHandle _)
|
|
|
|
|
{
|
|
|
|
|
handle.Completed -= continuationAction;
|
|
|
|
|
if (handle.IsValid())
|
|
|
|
|
{
|
|
|
|
|
handle.Completed -= completedCallback;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (completed)
|
|
|
|
|
{
|
|
|
|
|
TryReturn();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
completed = true;
|
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
|
|
|
{
|
|
|
|
|
if (autoReleaseWhenCanceled && handle.IsValid())
|
|
|
|
|
{
|
|
|
|
|
Addressables.Release(handle);
|
|
|
|
|
}
|
|
|
|
|
core.TrySetCanceled(cancellationToken);
|
|
|
|
|
}
|
|
|
|
|
else if (handle.Status == AsyncOperationStatus.Failed)
|
|
|
|
|
{
|
|
|
|
|
core.TrySetException(handle.OperationException);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
completed = true;
|
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
|
|
|
{
|
|
|
|
|
core.TrySetCanceled(cancellationToken);
|
|
|
|
|
}
|
|
|
|
|
else if (handle.Status == AsyncOperationStatus.Failed)
|
|
|
|
|
{
|
|
|
|
|
core.TrySetException(handle.OperationException);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
core.TrySetResult(AsyncUnit.Default);
|
|
|
|
|
}
|
|
|
|
|
core.TrySetResult(AsyncUnit.Default);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void GetResult(short token)
|
|
|
|
|
{
|
|
|
|
|
core.GetResult(token);
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
core.GetResult(token);
|
|
|
|
|
}
|
|
|
|
|
finally
|
|
|
|
|
{
|
|
|
|
|
if (!(cancelImmediately && cancellationToken.IsCancellationRequested))
|
|
|
|
|
{
|
|
|
|
|
TryReturn();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public UniTaskStatus GetStatus(short token)
|
|
|
|
@@ -208,13 +227,16 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
{
|
|
|
|
|
if (completed)
|
|
|
|
|
{
|
|
|
|
|
TryReturn();
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
|
|
|
{
|
|
|
|
|
completed = true;
|
|
|
|
|
if (autoReleaseWhenCanceled && handle.IsValid())
|
|
|
|
|
{
|
|
|
|
|
Addressables.Release(handle);
|
|
|
|
|
}
|
|
|
|
|
core.TrySetCanceled(cancellationToken);
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
@@ -248,17 +270,12 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
return ToUniTask(handle).GetAwaiter();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static UniTask<T> WithCancellation<T>(this AsyncOperationHandle<T> handle, CancellationToken cancellationToken)
|
|
|
|
|
public static UniTask<T> WithCancellation<T>(this AsyncOperationHandle<T> handle, CancellationToken cancellationToken, bool cancelImmediately = false, bool autoReleaseWhenCanceled = false)
|
|
|
|
|
{
|
|
|
|
|
return ToUniTask(handle, cancellationToken: cancellationToken);
|
|
|
|
|
return ToUniTask(handle, cancellationToken: cancellationToken, cancelImmediately: cancelImmediately, autoReleaseWhenCanceled: autoReleaseWhenCanceled);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static UniTask<T> WithCancellation<T>(this AsyncOperationHandle<T> handle, CancellationToken cancellationToken, bool cancelImmediately)
|
|
|
|
|
{
|
|
|
|
|
return ToUniTask(handle, cancellationToken: cancellationToken, cancelImmediately: cancelImmediately);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static UniTask<T> ToUniTask<T>(this AsyncOperationHandle<T> handle, IProgress<float> progress = null, PlayerLoopTiming timing = PlayerLoopTiming.Update, CancellationToken cancellationToken = default(CancellationToken), bool cancelImmediately = false)
|
|
|
|
|
public static UniTask<T> ToUniTask<T>(this AsyncOperationHandle<T> handle, IProgress<float> progress = null, PlayerLoopTiming timing = PlayerLoopTiming.Update, CancellationToken cancellationToken = default(CancellationToken), bool cancelImmediately = false, bool autoReleaseWhenCanceled = false)
|
|
|
|
|
{
|
|
|
|
|
if (cancellationToken.IsCancellationRequested) return UniTask.FromCanceled<T>(cancellationToken);
|
|
|
|
|
|
|
|
|
@@ -276,7 +293,7 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
return UniTask.FromResult(handle.Result);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new UniTask<T>(AsyncOperationHandleConfiguredSource<T>.Create(handle, timing, progress, cancellationToken, cancelImmediately, out var token), token);
|
|
|
|
|
return new UniTask<T>(AsyncOperationHandleConfiguredSource<T>.Create(handle, timing, progress, cancellationToken, cancelImmediately, autoReleaseWhenCanceled, out var token), token);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
sealed class AsyncOperationHandleConfiguredSource<T> : IUniTaskSource<T>, IPlayerLoopItem, ITaskPoolNode<AsyncOperationHandleConfiguredSource<T>>
|
|
|
|
@@ -290,21 +307,23 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
TaskPool.RegisterSizeGetter(typeof(AsyncOperationHandleConfiguredSource<T>), () => pool.Size);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
readonly Action<AsyncOperationHandle<T>> continuationAction;
|
|
|
|
|
readonly Action<AsyncOperationHandle<T>> completedCallback;
|
|
|
|
|
AsyncOperationHandle<T> handle;
|
|
|
|
|
CancellationToken cancellationToken;
|
|
|
|
|
CancellationTokenRegistration cancellationTokenRegistration;
|
|
|
|
|
IProgress<float> progress;
|
|
|
|
|
bool autoReleaseWhenCanceled;
|
|
|
|
|
bool cancelImmediately;
|
|
|
|
|
bool completed;
|
|
|
|
|
|
|
|
|
|
UniTaskCompletionSourceCore<T> core;
|
|
|
|
|
|
|
|
|
|
AsyncOperationHandleConfiguredSource()
|
|
|
|
|
{
|
|
|
|
|
continuationAction = Continuation;
|
|
|
|
|
completedCallback = HandleCompleted;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static IUniTaskSource<T> Create(AsyncOperationHandle<T> handle, PlayerLoopTiming timing, IProgress<float> progress, CancellationToken cancellationToken, bool cancelImmediately, out short token)
|
|
|
|
|
public static IUniTaskSource<T> Create(AsyncOperationHandle<T> handle, PlayerLoopTiming timing, IProgress<float> progress, CancellationToken cancellationToken, bool cancelImmediately, bool autoReleaseWhenCanceled, out short token)
|
|
|
|
|
{
|
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
|
|
|
{
|
|
|
|
@@ -320,12 +339,18 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
result.cancellationToken = cancellationToken;
|
|
|
|
|
result.completed = false;
|
|
|
|
|
result.progress = progress;
|
|
|
|
|
result.autoReleaseWhenCanceled = autoReleaseWhenCanceled;
|
|
|
|
|
result.cancelImmediately = cancelImmediately;
|
|
|
|
|
|
|
|
|
|
if (cancelImmediately && cancellationToken.CanBeCanceled)
|
|
|
|
|
{
|
|
|
|
|
result.cancellationTokenRegistration = cancellationToken.RegisterWithoutCaptureExecutionContext(state =>
|
|
|
|
|
{
|
|
|
|
|
var promise = (AsyncOperationHandleConfiguredSource<T>)state;
|
|
|
|
|
if (promise.autoReleaseWhenCanceled && promise.handle.IsValid())
|
|
|
|
|
{
|
|
|
|
|
Addressables.Release(promise.handle);
|
|
|
|
|
}
|
|
|
|
|
promise.core.TrySetCanceled(promise.cancellationToken);
|
|
|
|
|
}, result);
|
|
|
|
|
}
|
|
|
|
@@ -334,41 +359,53 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
|
|
|
|
|
PlayerLoopHelper.AddAction(timing, result);
|
|
|
|
|
|
|
|
|
|
handle.Completed += result.continuationAction;
|
|
|
|
|
handle.Completed += result.completedCallback;
|
|
|
|
|
|
|
|
|
|
token = result.core.Version;
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void Continuation(AsyncOperationHandle<T> argHandle)
|
|
|
|
|
void HandleCompleted(AsyncOperationHandle<T> argHandle)
|
|
|
|
|
{
|
|
|
|
|
handle.Completed -= continuationAction;
|
|
|
|
|
if (handle.IsValid())
|
|
|
|
|
{
|
|
|
|
|
handle.Completed -= completedCallback;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (completed)
|
|
|
|
|
{
|
|
|
|
|
TryReturn();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
completed = true;
|
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
|
|
|
{
|
|
|
|
|
if (autoReleaseWhenCanceled && handle.IsValid())
|
|
|
|
|
{
|
|
|
|
|
Addressables.Release(handle);
|
|
|
|
|
}
|
|
|
|
|
core.TrySetCanceled(cancellationToken);
|
|
|
|
|
}
|
|
|
|
|
else if (argHandle.Status == AsyncOperationStatus.Failed)
|
|
|
|
|
{
|
|
|
|
|
core.TrySetException(argHandle.OperationException);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
completed = true;
|
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
|
|
|
{
|
|
|
|
|
core.TrySetCanceled(cancellationToken);
|
|
|
|
|
}
|
|
|
|
|
else if (argHandle.Status == AsyncOperationStatus.Failed)
|
|
|
|
|
{
|
|
|
|
|
core.TrySetException(argHandle.OperationException);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
core.TrySetResult(argHandle.Result);
|
|
|
|
|
}
|
|
|
|
|
core.TrySetResult(argHandle.Result);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public T GetResult(short token)
|
|
|
|
|
{
|
|
|
|
|
return core.GetResult(token);
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
return core.GetResult(token);
|
|
|
|
|
}
|
|
|
|
|
finally
|
|
|
|
|
{
|
|
|
|
|
if (!(cancelImmediately && cancellationToken.IsCancellationRequested))
|
|
|
|
|
TryReturn();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void IUniTaskSource.GetResult(short token)
|
|
|
|
@@ -395,13 +432,16 @@ namespace Cysharp.Threading.Tasks
|
|
|
|
|
{
|
|
|
|
|
if (completed)
|
|
|
|
|
{
|
|
|
|
|
TryReturn();
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
|
|
|
{
|
|
|
|
|
completed = true;
|
|
|
|
|
if (autoReleaseWhenCanceled && handle.IsValid())
|
|
|
|
|
{
|
|
|
|
|
Addressables.Release(handle);
|
|
|
|
|
}
|
|
|
|
|
core.TrySetCanceled(cancellationToken);
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|