7

我有一个异步方法

private async Task DoSomething(CancellationToken token)

任务列表

private List<Task> workers = new List<Task>();

我必须创建运行该方法的 N 个线程

public void CreateThreads(int n)
{
    tokenSource = new CancellationTokenSource();
    token = tokenSource.Token;
    for (int i = 0; i < n; i++)
    {
        workers.Add(DoSomething(token));
    }
}

但问题是那些必须在给定时间运行

public async Task StartAllWorkers()
{
    if (workers.Count > 0)
    {
        try
        {
            while (workers.Count > 0)
            {
                Task finishedWorker = await Task.WhenAny(workers.ToArray());
                workers.Remove(finishedWorker);
                finishedWorker.Dispose();
            }
            if (workers.Count == 0)
            {
                tokenSource = null;
            }
        }
        catch (OperationCanceledException)
        {
            throw;
        }
    }
}

但实际上它们在我调用 CreateThreads 方法时运行（在 StartAllWorkers 之前）。我搜索了像我这样的关键字和问题，但找不到任何关于阻止任务运行的信息。我尝试了很多不同的方法，但任何可以完全解决我的问题的方法。例如，将代码从DoSomethinga 移到 aworkers.Add(new Task(async () => { }, token));会运行StartAllWorkers()，但线程永远不会真正启动。

还有另一种调用方法tokenSource.Cancel()。

4

2 回答 2

13

您可以TaskCompletionSource<T>将异步方法用作一次性“信号”。

所以你会像这样创建它：

private TaskCompletionSource<object> _tcs;
public void CreateThreads(int n)
{
    _tcs = new TaskCompletionSource<object>();
    tokenSource = new CancellationTokenSource();
    token = tokenSource.Token;
    for (int i = 0; i < n; i++)
    {
        workers.Add(DoSomething(_tcs.Task, token));
    }
}

然后当你准备好开始任务时，只需完成“开始”信号任务：

public Task StartAllWorkers()
{
    _tcs.TrySetCompleted(null);
    return Task.WhenAll(workers);
}

（StartAllWorkers上面的方法与您的原始方法的语义略有不同：您的原始方法会在第一个任务取消后立即抛出取消异常；这个将等到所有方法完成然后抛出取消异常）

然后你DoSomething只需要尊重“开始信号”：

private static async Task DoSomething(Task start, CancellationToken token)
{
    await start;
    ... // rest of your code
}
于 2013-10-27T14:22:43.160 回答
2

这个想法怎么样：

与其持有任务列表，不如持有 TaskReference 项目列表：

public class TaskReference
{
    private readonly Func<Task> _func;

    public TaskReference(Func<Task> func)
    {
        _func = func;
    }

    public async Task RunAsync()
    {
        await _func();
    }
}

添加到列表的工作方式如下：

taskList.Add(new TaskReference(() => DoSomething(myToken)));

并像这样执行：

await Task.WhenAll(taskList.Select(o => o.RunAsync()));
于 2015-03-03T14:55:23.783 回答