81 lines
2.6 KiB
C#
81 lines
2.6 KiB
C#
using ApplicationException = System.ApplicationException;
|
||
using TaskExtensions = ConsoleApp2.Helpers.TaskExtensions;
|
||
|
||
namespace ConsoleApp2.Services;
|
||
|
||
/// <summary>
|
||
/// 快速批量创建和等待任务
|
||
/// </summary>
|
||
public class TaskManager
|
||
{
|
||
private int _runningTaskCount;
|
||
|
||
public int RunningTaskCount => _runningTaskCount;
|
||
public int MaxTaskCount { get; }
|
||
|
||
public event Action<Exception>? OnException;
|
||
public event Action? OnTaskCompleteSuccessfully;
|
||
|
||
public TaskManager(int maxTaskCount)
|
||
{
|
||
MaxTaskCount = maxTaskCount;
|
||
}
|
||
|
||
public async ValueTask<Task> CreateTaskAsync(Func<Task> func, CancellationToken cancellationToken = default)
|
||
{
|
||
await TaskExtensions.WaitUntil(() => _runningTaskCount < MaxTaskCount, 25, cancellationToken);
|
||
return RunTask(func, cancellationToken);
|
||
}
|
||
|
||
public async ValueTask<Task> CreateTaskAsync(Func<object?, Task> func, object? arg, CancellationToken ct = default)
|
||
{
|
||
await TaskExtensions.WaitUntil(() => _runningTaskCount < MaxTaskCount, 25, ct);
|
||
return RunTaskNoClosure(func, arg, ct);
|
||
}
|
||
|
||
private Task RunTask(Func<Task> func, CancellationToken cancellationToken = default)
|
||
{
|
||
var task = Task.Run(async () =>
|
||
{
|
||
try
|
||
{
|
||
await func();
|
||
OnTaskCompleteSuccessfully?.Invoke();
|
||
}
|
||
catch(Exception ex)
|
||
{
|
||
OnException?.Invoke(ex);
|
||
}
|
||
finally
|
||
{
|
||
Interlocked.Decrement(ref _runningTaskCount);
|
||
}
|
||
}, cancellationToken);
|
||
Interlocked.Increment(ref _runningTaskCount);
|
||
return task;
|
||
}
|
||
|
||
private Task RunTaskNoClosure(Func<object?, Task> func, object? arg, CancellationToken cancellationToken = default)
|
||
{
|
||
var task = Task.Factory.StartNew(async obj => // 性能考虑，这个lambda中不要捕获任何外部变量!
|
||
{
|
||
if (obj is not Tuple<Func<object?, Task>, object?> tuple)
|
||
throw new ApplicationException("这个异常不该出现");
|
||
try
|
||
{
|
||
await tuple.Item1(tuple.Item2);
|
||
OnTaskCompleteSuccessfully?.Invoke();
|
||
}
|
||
catch(Exception ex)
|
||
{
|
||
OnException?.Invoke(ex);
|
||
}
|
||
finally
|
||
{
|
||
Interlocked.Decrement(ref _runningTaskCount);
|
||
}
|
||
}, Tuple.Create(func, arg), cancellationToken).Unwrap();
|
||
Interlocked.Increment(ref _runningTaskCount);
|
||
return task;
|
||
}
|
||
} |