如何知道线程池的所有线程是否已经完成了它的任务？

Question

我有这个应用程序，它将递归给定目录中的所有文件夹并查找 PDF。如果找到 PDF 文件，应用程序将使用 ITextSharp 计算其页数。我通过使用一个线程递归地扫描所有文件夹中的 pdf 来做到这一点，然后如果找到 PDF，它将被排队到线程池中。代码如下所示：

//spawn a thread to handle the processing of pdf on each folder.
                var th = new Thread(() =>
                {
                    pdfDirectories = Directory.GetDirectories(pdfPath);
                    processDir(pdfDirectories);
                });
                th.Start();



 private void processDir(string[] dirs)
        {
            foreach (var dir in dirs)
            {
                pdfFiles = Directory.GetFiles(dir, "*.pdf");
                processFiles(pdfFiles);

                string[] newdir = Directory.GetDirectories(dir);
                processDir(newdir);
            }
        }



private void processFiles(string[] files)
        {
            foreach (var pdf in files)
            {
                ThreadPoolHelper.QueueUserWorkItem(
                    new { path = pdf },
                    (data) => { processPDF(data.path); }
                    );
            }
        }

我的问题是，我怎么知道线程池的线程已经完成了所有排队项的处理，所以我可以告诉用户应用程序已经完成了它的预期任务？

Answer 1

通常我会通过一个计数器变量来做这样的事情。 对于您在ThreadPool 中排队的每个工作项，将一个添加到计数器变量。 然后当它被处理时，你会减少计数器变量。

确保您通过Interlocked 类上的方法进行递增和递减，因为这将确保以线程安全的方式完成。

一旦计数器达到零，您可以使用 ManualResetEvent 标记任务已完成

如果您可以访问 .NET 4，那么您可以使用新的 CountdownEvent 类来做类似的事情。

Answer 2

1) 如何知道是否所有线程都结束了？

您必须让线程自己签入/签出，方法是将您的代码括起来：

Interlocked.Increment(ref jobCounter);
// your code
Interlocked.Decrement(ref jobCounter);

如果这让您的匿名委托过于混乱，那么只需使用包装方法。您可能还必须添加异常处理。

联锁方法仍然避免了等待它变为 0 的问题，使用 Sleep() 的循环是一个弱但在这种情况下可行的解决方案。

2) 您正在递归 Tree walker 中启动线程。请注意，您可能创建了太多它们，这会影响性能。