是否可以等待未知任务完成？

Question

使用 .NET 任务并行库，是否有一种内置方法可以等待所有正在运行的任务完成，而无需实际引用任务？

Task.WaitAny() 和 Task.WaitAll() 都需要 Task 实例的数组。我想要的是 Task.WaitAll() 之类的东西，它不带任何参数，只是等待我的应用程序生成的所有任务完成。

如果没有，我会在我的框架中构建一些东西，但如果我能提供帮助，我不想重新发明轮子。

Answer 1

我知道这不能直接回答您的问题，但您可以更改您的架构，以便将所有任务创建为一个“主要”任务的子任务。等待这个“主”任务会自动等待它的所有子任务。

你可以像这样创建一个子任务：

Task.Factory.StartNew(
    () => /* ... */,
    TaskCreationOptions.AttachedToParent
);

当涉及到子任务的主题时，您可能会发现以下链接很有用：

• http://msdn.microsoft.com/en-us/library/ff963551.aspx（请参阅“与父/子任务链接的任务”部分）
• http://msdn.microsoft.com/en-us/library/dd997417.aspx

Answer 2

由于任何 Task 都可以由任何 .NET 代码（即 BCL、某些库等）生成，因此您可能不想等待所有未完成的任务，而只等待您自己的代码具有的任务创建的。最简单的方法是创建自己的工厂，使用底层工厂跟踪所有未完成的任务并公开所需的等待函数

public class MyTaskFactory {
    private HashSet<Task> _tasks = new HashSet<Task>();

    public Task<T> StartNew<T>(Func<T> func) {
        var t = Task.Factory.StartNew(func);
        t.ContinueWith(x => {
          lock(_tasks) {
            _tasks.Remove(x);
          }
        });
        lock(_tasks) {
          _tasks.Add(t);
        }
        return t;
    }

    // ... implement other StartNew overrides ...

    public void WaitAll() {
        Task[] tasks;
        lock(_tasks) {
          tasks = _tasks.ToArray();
        }
        Task.WaitAll(tasks);
    }
}