等待QueueUserWorkItem完成
本文关键字:完成 QueueUserWorkItem 等待 | 更新日期: 2023-09-27 18:00:39
如果我使用QueueUserWorkItem
将作业添加到线程池。。。在完成所有工作之前,我如何阻止我的程序继续进行?
我知道我可以添加一些逻辑来阻止应用程序运行,直到所有作业完成,但我想知道是否有类似Thread.Join()
的东西,或者是否有任何方法可以检索被分配作业的每个线程。
您可以使用事件进行同步。像这样:
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());
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.
}
您可以使用.NET的Barrier类来实现这一点。
Barrier barrier = new Barrier(3);
for(int i = 0; i < 2; i++)
{
ThreadPool.QueueUserWorkItem(
(state) =>
{
foo();
barrier.SignalAndWait();
}, null);
}
barrier.SignalAndWait();