foldcc/UniTask
1
0
Fork 0
mirror of https://github.com/Cysharp/UniTask.git synced 2026-05-23 00:00:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add to the code that I forgot to write

Browse Source
This commit is contained in:
Kochoyume
2024-09-24 23:58:28 +09:00
parent dfe5ee43c2
commit 06283f0ffb
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/UniTask/Assets/Plugins/UniTask/Runtime/UniTask.WaitUntil.cs
View File

@@ -241,6 +241,10 @@ namespace Cysharp.Threading.Tasks
{ {
TryReturn(); TryReturn();
} }
else
{
TaskTracker.RemoveTracking(this);
}
} }
} }
@@ -461,6 +465,7 @@ namespace Cysharp.Threading.Tasks
result.predicate = predicate; result.predicate = predicate;
result.argument = argument; result.argument = argument;
result.cancellationToken = cancellationToken; result.cancellationToken = cancellationToken;
result.cancelImmediately = cancelImmediately;
if (cancelImmediately && cancellationToken.CanBeCanceled) if (cancelImmediately && cancellationToken.CanBeCanceled)
{ {
@@ -491,6 +496,10 @@ namespace Cysharp.Threading.Tasks
{ {
TryReturn(); TryReturn();
} }
else
{
TaskTracker.RemoveTracking(this);
}
} }
} }