sim1222/osukey
1
1
Fork 0
mirror of https://github.com/osukey/osukey.git synced 2025-08-05 07:33:55 +09:00
Code Issues Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
smoogipoo
2021-01-27 01:20:50 +09:00
parent 085115cba5
commit 248989b3eb
3 changed files with 51 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
osu.Game/Utils/TaskChain.cs
View File

@ -14,8 +14,8 @@ namespace osu.Game.Utils
/// </summary>
public class TaskChain
{
private readonly object currentTaskLock = new object();
private Task? currentTask;
private readonly object finalTaskLock = new object();
private Task? finalTask;
/// <summary>
/// Adds a new task to the end of this <see cref="TaskChain"/>.
@ -23,31 +23,22 @@ namespace osu.Game.Utils
/// <param name="action">The action to be executed.</param>
/// <param name="cancellationToken">The <see cref="CancellationToken"/> for this task. Does not affect further tasks in the chain.</param>
/// <returns>The awaitable <see cref="Task"/>.</returns>
public Task Add(Action action, CancellationToken cancellationToken = default)
public async Task Add(Action action, CancellationToken cancellationToken = default)
{
lock (currentTaskLock)
{
// Note: Attaching the cancellation token to the continuation could lead to re-ordering of tasks in the chain.
// Therefore, the cancellation token is not used to cancel the continuation but only the run of each task.
if (currentTask == null)
{
currentTask = Task.Run(() =>
{
cancellationToken.ThrowIfCancellationRequested();
action();
}, CancellationToken.None);
}
else
{
currentTask = currentTask.ContinueWith(_ =>
{
cancellationToken.ThrowIfCancellationRequested();
action();
}, CancellationToken.None);
}
Task? previousTask;
Task currentTask;
return currentTask;
lock (finalTaskLock)
{
previousTask = finalTask;
finalTask = currentTask = new Task(action, cancellationToken);
}
if (previousTask != null)
await previousTask;
currentTask.Start();
await currentTask;
}
}
}