Ожидание завершения QueueUserWorkItem

Question

Если я добавляю задания в пул потоков с помощью QueueUserWorkItem ... как мне не допустить продолжения моей программы до тех пор, пока все задания не будут завершены?

Я знаю, что мог бы добавить некоторую логику, чтобы приложение не запускалось до тех пор, пока все задания не будут завершены, но я хочу знать, есть ли что-то вроде Thread.Join() или есть ли способ извлечь каждый поток, которому назначено задание .

Answer 1

Вы можете использовать события для синхронизации. Как это:

private static ManualResetEvent resetEvent = new ManualResetEvent(false);

public static void Main()
{
    ThreadPool.QueueUserWorkItem(arg => DoWork());
    resetEvent.WaitOne();
}

public static void DoWork()
{
    Thread.Sleep(5000);
    resetEvent.Set();
}

Если вы не хотите вставлять набор событий в ваш метод, вы можете сделать что-то вроде этого:

var resetEvent = new ManualResetEvent(false);
ThreadPool.QueueUserWorkItem(
    arg => 
    {
        DoWork();
        resetEvent.Set();
    });
resetEvent.WaitOne();

Для нескольких предметов:

var events = new List<ManualResetEvent>();

foreach(var job in jobs)
{   
    var resetEvent = new ManualResetEvent(false);
    ThreadPool.QueueUserWorkItem(
        arg =>
        {
            DoWork(job);
            resetEvent.Set();
        });
    events.Add(resetEvent);
}
WaitHandle.WaitAll(events.ToArray());

Answer 2

Лучший способ сделать это - использовать класс CountdownEvent. Это довольно устоявшаяся модель, и она настолько масштабируема, насколько это возможно.

using (var finished = new CountdownEvent(1))
{
  foreach (var workitem in workitems)
  {
    var capture = workitem; // Used to capture the loop variable in the lambda expression.
    finished.AddCount(); // Indicate that there is another work item.
    ThreadPool.QueueUserWorkItem(
      (state) =>
      {
        try
        {
          ProcessWorkItem(capture);
        }
        finally
        {
          finished.Signal(); // Signal that the work item is complete.
        }
      }, null);
  }
  finished.Signal(); // Signal that queueing is complete.
  finished.Wait(); // Wait for all work items to complete.
}

Answer 3

Вы можете использовать класс .NET Barrier для достижения этой цели.

Barrier barrier = new Barrier(3); 
for(int i = 0; i < 2; i++)
{
    ThreadPool.QueueUserWorkItem(
    (state) =>
    {
       foo();
       barrier.SignalAndWait();
    }, null);
}
barrier.SignalAndWait();