Lightless-Sync/LightlessClient
8
1
Fork 2
Code Issues 2 Pull Requests 3 Actions Packages Projects Releases 57 Wiki Activity

:sludge:

Browse Source
This commit is contained in:
Essie
2026-01-16 19:29:24 +09:00
parent 6c7e4e6303
commit 7c281926a5
1 changed files with 10 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
LightlessSync/Utils/TaskRegistry.cs
View File

@@ -31,34 +31,26 @@ public sealed class TaskRegistry<HandleType> where HandleType : notnull
private Task GetOrStartInternal(HandleType handle, Func<Task> taskFactory) private Task GetOrStartInternal(HandleType handle, Func<Task> taskFactory)
{ {
while (true) Lazy<Task> entry = _activeTasks.GetOrAdd(handle, _ => CreateEntry(handle, taskFactory));
Task task = entry.Value;
if (task.IsCompleted)
{ {
Lazy<Task> entry = _activeTasks.GetOrAdd(handle, _ => CreateEntry(handle, taskFactory));
Task task = entry.Value;
if (!task.IsCompleted)
{
return task;
}
_activeTasks.TryRemove(new KeyValuePair<HandleType, Lazy<Task>>(handle, entry)); _activeTasks.TryRemove(new KeyValuePair<HandleType, Lazy<Task>>(handle, entry));
} }
return task;
} }
private Task<T> GetOrStartInternal<T>(HandleType handle, Func<Task<T>> taskFactory) private Task<T> GetOrStartInternal<T>(HandleType handle, Func<Task<T>> taskFactory)
{ {
while (true) Lazy<Task> entry = _activeTasks.GetOrAdd(handle, _ => CreateEntry(handle, taskFactory));
Task task = entry.Value;
if (task.IsCompleted)
{ {
Lazy<Task> entry = _activeTasks.GetOrAdd(handle, _ => CreateEntry(handle, taskFactory));
Task task = entry.Value;
if (!task.IsCompleted)
{
return (Task<T>)task;
}
_activeTasks.TryRemove(new KeyValuePair<HandleType, Lazy<Task>>(handle, entry)); _activeTasks.TryRemove(new KeyValuePair<HandleType, Lazy<Task>>(handle, entry));
} }
return (Task<T>)task;
} }
private Lazy<Task> CreateEntry(HandleType handle, Func<Task> taskFactory) private Lazy<Task> CreateEntry(HandleType handle, Func<Task> taskFactory)