pengtaihao
/
UniTask
mirror of https://github.com/Cysharp/UniTask
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Implement WhenAny and WhenAll overloads through T4 templates

pull/322/head
Artem Perepelitsa 2021-10-14 22:25:15 +03:00
parent eb78c76e91
commit 318464916b
2 changed files with 115 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
src/UniTask/Assets/Plugins/UniTask/Runtime/UniTask.WhenAll.Generated.tt
View File

@ -118,5 +118,64 @@ namespace Cysharp.Threading.Tasks
}
}
<# } #>
<# for(var i = 2; i <= 15; i++ ) {
var range = Enumerable.Range(1, i);
var args = string.Join(", ", range.Select(x => $"UniTask task{x}"));
var targs = string.Join(", ", range.Select(x => $"task{x}"));
#>
public static UniTask WhenAll(<#= args #>)
{
return new UniTask(new WhenAllPromise(<#= targs #>), 0);
}
<# } #>
sealed partial class WhenAllPromise : IUniTaskSource
{
<# for(var i = 2; i <= 15; i++ ) {
var range = Enumerable.Range(1, i);
var args = string.Join(", ", range.Select(x => $"UniTask task{x}"));
var startTasks = string.Join("\n", range.Select(x => $"{new string(' ', 16)}StartTask(task{x});"));
#>
public WhenAllPromise(<#= args #>)
{
TaskTracker.TrackActiveTask(this, 3);
this.tasksLength = <#= i #>;
this.completeCount = 0;
<#= startTasks #>
}
<# } #>
private void StartTask(UniTask task)
{
UniTask.Awaiter awaiter;
try
{
awaiter = task.GetAwaiter();
}
catch (Exception ex)
{
core.TrySetException(ex);
return;
}
if (awaiter.IsCompleted)
{
TryInvokeContinuation(this, awaiter);
}
else
{
awaiter.SourceOnCompleted(state =>
{
using (var t = (StateTuple<WhenAllPromise, UniTask.Awaiter>)state)
{
TryInvokeContinuation(t.Item1, t.Item2);
}
}, StateTuple.Create(this, awaiter));
}
}
}
}
}

56
src/UniTask/Assets/Plugins/UniTask/Runtime/UniTask.WhenAny.Generated.tt
View File

@ -113,5 +113,61 @@ namespace Cysharp.Threading.Tasks
}
<# } #>
<# for(var i = 2; i <= 15; i++ ) {
var range = Enumerable.Range(1, i);
var args = string.Join(", ", range.Select(x => $"UniTask task{x}"));
var targs = string.Join(", ", range.Select(x => $"task{x}"));
#>
public static UniTask<int> WhenAny(<#= args #>)
{
return new UniTask<int>(new WhenAnyPromise(<#= targs #>), 0);
}
<# } #>
sealed partial class WhenAnyPromise : IUniTaskSource<int>
{
<# for(var i = 2; i <= 15; i++ ) {
var range = Enumerable.Range(1, i);
var args = string.Join(", ", range.Select(x => $"UniTask task{x}"));
var startTasks = string.Join("\n", range.Select(x => $"{new string(' ', 16)}StartTask(task{x}, {x-1});"));
#>
public WhenAnyPromise(<#= args #>)
{
TaskTracker.TrackActiveTask(this, 3);
<#= startTasks #>
}
<# } #>
void StartTask(UniTask task, int index)
{
Awaiter awaiter;
try
{
awaiter = task.GetAwaiter();
}
catch (Exception ex)
{
core.TrySetException(ex);
return; // consume others.
}
if (awaiter.IsCompleted)
{
TryInvokeContinuation(this, awaiter, index);
}
else
{
awaiter.SourceOnCompleted(state =>
{
using (var t = (StateTuple<WhenAnyPromise, Awaiter, int>)state)
{
TryInvokeContinuation(t.Item1, t.Item2, t.Item3);
}
}, StateTuple.Create(this, awaiter, index));
}
}
}
}
}