任务并行库 - 知道所有任务何时完成

Question

我使用任务并行库来启动一些任务，如下所示：

    public static void Main()
    {
        for (var i = 0; i < 10; i++)
        {
            var x = i;
            Task.Factory.StartNew(() => new WorkerClass(x).Do());
        }

        // (*) Here I'd like to wait for all tasks to finish
        Task.WaitAll(); 

        Console.WriteLine("Ready.");
        Console.ReadLine();
    }

问题是某些任务可以自己创建新任务。这就是WorkerClass 的样子：

public class WorkerClass
{
    private static readonly NLog.Logger Log = NLog.LogManager.GetCurrentClassLogger();

    private readonly int _i;

    public WorkerClass(int i)
    {
        _i = i;
    }

    public void Do()
    {
        if (_i % 3 == 0)
            Task.Factory.StartNew(() => new WorkerClass(_i + 101).Do());

        Log.Info("Started {0}", _i);
        Thread.Sleep(2000);
        Log.Info("Done {0}", _i);
    }
}

i 的每一个值是 3 的倍数，就会启动一个新任务。

我希望能够等到所有任务（包括由其他任务创建的任务）完成。

是否有干净/内置的方式来执行此操作（使用或不使用 TPL）？

Answer 1

保留对所有顶级任务的引用，然后使用WaitAll：

    var tasks = new Task[10];
    for (var i = 0; i < 10; i++)
    {
        var x = i;
        tasks[i] = Task.Factory.StartNew(() => new WorkerClass(x).Do());
    }

    Task.WaitAll( tasks );

至于子任务，只需确保将它们附加到父任务即可。这意味着在所有子任务也完成之前，父任务不会进入完成状态。

    Task.Factory.StartNew(() => { }, TaskCreationOptions.AttachedToParent);