如何创建自定义消息泵

本文关键字：自定义消息 创建 何创建 | 更新日期: 2023-09-27 18:00:15

我正在做一个小练习，需要创建类似于消息泵的东西。我有一个工作队列，我希望工作完全在一个线程上完成，而任何线程都可以将工作添加到要完成的队列中。

Queue<WorkToDo> queue;

线程使用等待句柄来告诉泵有工作要做。

WaitHandle signal;

只要有工作要做，泵就会循环，然后等待信号再次启动。

while(ApplicationIsRunning){
    while(queue.HasWork){
        DoWork(queue.NextWorkItem)
    }
    signal.Reset();
    signal.WaitOne();
}

每隔一个线程都可以向队列中添加工作，并发出等待句柄的信号。。。

public void AddWork(WorkToDo work){
    queue.Add(work);
    signal.Set();
}

问题是，如果添加工作的速度足够快，可能会出现工作可以留在队列中的情况，因为在队列检查工作和WaitHandle重置之间，另一个线程可以将工作添加到队列中。

如果不在WaitHandle周围放置代价高昂的互斥锁，我该如何减轻这种情况？

如何创建自定义消息泵

您可以使用BlockingCollection<T>来简化队列的实现，因为它将为您处理同步：

public class MessagePump
{
    private BlockingCollection<Action> actions = new BlockingCollection<Action>();
    public void Run() //you may want to restrict this so that only one caller from one thread is running messages
    {
        foreach (var action in actions.GetConsumingEnumerable())
            action();
    }
    public void AddWork(Action action)
    {
        actions.Add(action);
    }
    public void Stop()
    {
        actions.CompleteAdding();
    }
}

你不应该做一个完整的互斥，但你可以放一个锁语句吗

public void AddWork(WorkToDo work)
{
  queue.Add(work);
  lock(lockingObject)
  {
    signal.Set();
  }
}

对锁定对象使用你想要的任何东西，大多数人会说使用信号本身是个坏主意。

响应下面@500-Server内部错误的评论，你可以在做这项工作之前重置信号。以下内容应保护事物：

while(ApplicationIsRunning)
{
  while(queue.HasWork)
  {
    WorkItem wi;
    lock(lockingObject)
    {
       wi = queue.NextWorkItem;
       if(!queue.HasWork)
       {
          signal.Reset();
       }
    }
    DoWork(wi)
  }
  signal.WaitOne();
}

这样，如果您有更多的工作，内部队列就会继续运行。如果没有，它会掉到signal.WaitOne()上，只有在没有更多工作排队的情况下，我们才会重置。

这里唯一的缺点是，如果在执行DoWork时有工作，我们可能会连续多次重置。

您可以使用WaitOne（TimeSpan）方法，这样您就有了一个混合信号/轮询循环。基本上指定的等待时间跨度最多为1秒。这将导致一项任务在比赛中最多保持一秒钟（或您指定的任何轮询时间），或者直到另一项任务添加到您的队列中。

相关文章：