Question

我想在一个单独的线程中有一个特定的线程，队列用于任务和处理任务。应用程序将根据用户使用情况制作任务，并将其排入任务队列。然后单独的线程处理任务。即使队列为空，保持线程处于活动状态并使用它来处理排队任务也是至关重要的。

我已尝试使用TaskScheduler的{{1}}的几个实现，并将并发限制为仅一个线程，但似乎当队列变空并且任务由其他线程处理时，线程将被释放。

请您至少向我推荐一些如何实现这一目标的来源？

TL;博士尝试限制一个特定线程来处理动态添加到队列中的任务。

Edit1：

这是使用WCF和.NET framework 4.6的实验性Web应用程序。在WCF库中，我试图用一个线程处理任务来实现此行为。这个线程必须使用外部dll库初始化prolog，然后使用prolog。如果在进程中使用了其他线程，则库会抛出BlockingCollection。我做了一些研究，这很可能是因为该库中的线程管理不当。我实现了我到处都有锁的工作方式。我现在正在尝试重新实现并使其异步，因此我不会使用锁定来阻止主线程。

我不在我的电脑上，但是当我今天晚些时候回家时，我提供了一些代码。

Answer 1

你的方法似乎很好，所以你可能只是犯了一些小愚蠢的错误。

制作简单的自定义TaskScheduler实际上很容易。对于你的情况：

void Main()
{
  var cts = new CancellationTokenSource();
  var myTs = new SingleThreadTaskScheduler(cts.Token);

  myTs.Schedule(() => 
   { Print("Init start"); Thread.Sleep(1000); Print("Init done"); });
  myTs.Schedule(() => Print("Work 1"));   
  myTs.Schedule(() => Print("Work 2"));
  myTs.Schedule(() => Print("Work 3"));
  var lastOne = myTs.Schedule(() => Print("Work 4"));

  Print("Starting TS");
  myTs.Start();

  // Wait for all of them to complete...
  lastOne.GetAwaiter().GetResult();

  Thread.Sleep(1000);

  // And try to schedule another
  myTs.Schedule(() => Print("After emptied")).GetAwaiter().GetResult();

  // And shutdown; it's also pretty useful to have the 
  // TaskScheduler return a "complete task" to await
  myTs.Complete();

  Print("On main thread again");
}

void Print(string str)
{
  Console.WriteLine("{0}: {1}", Thread.CurrentThread.ManagedThreadId, str);
  Thread.Sleep(100);
}

public sealed class SingleThreadTaskScheduler : TaskScheduler
{
  [ThreadStatic]
  private static bool isExecuting;
  private readonly CancellationToken cancellationToken;

  private readonly BlockingCollection<Task> taskQueue;

  public SingleThreadTaskScheduler(CancellationToken cancellationToken)
  {
      this.cancellationToken = cancellationToken;
      this.taskQueue = new BlockingCollection<Task>();
  }

  public void Start()
  {
      new Thread(RunOnCurrentThread) { Name = "STTS Thread" }.Start();
  }

  // Just a helper for the sample code
  public Task Schedule(Action action)
  {
      return 
          Task.Factory.StartNew
              (
                  action, 
                  CancellationToken.None, 
                  TaskCreationOptions.None, 
                  this
              );
  }

  // You can have this public if you want - just make sure to hide it
  private void RunOnCurrentThread()
  {
      isExecuting = true;

      try
      {
          foreach (var task in taskQueue.GetConsumingEnumerable(cancellationToken))
          {
              TryExecuteTask(task);
          }
      }
      catch (OperationCanceledException)
      { }
      finally
      {
          isExecuting = false;
      }
  }

  // Signaling this allows the task scheduler to finish after all tasks complete
  public void Complete() { taskQueue.CompleteAdding(); }   
  protected override IEnumerable<Task> GetScheduledTasks() { return null; }

  protected override void QueueTask(Task task)
  {
      try
      {
          taskQueue.Add(task, cancellationToken);
      }
      catch (OperationCanceledException)
      { }
  }

  protected override bool TryExecuteTaskInline(Task task, bool taskWasPreviouslyQueued)
  {
      // We'd need to remove the task from queue if it was already queued. 
      // That would be too hard.
      if (taskWasPreviouslyQueued) return false;

      return isExecuting && TryExecuteTask(task);
  }
}

很容易修改它，让你完全控制任务调度程序实际执行任务的位置 - 事实上，我已经使用过的任务调度程序对此进行了调整只需将RunOnCurrentThread方法公开。

对于你的情况，你总是想要坚持一个线程，SingleThreadTaskScheduler中的方法可能更好。虽然这也有其优点：

// On a new thread
try
{
  InitializeProlog();

  try
  {
    myTs.RunOnCurrentThread();
  }
  finally
  {
    ReleaseProlog();
  }
}
catch (Exception ex)
{
  // The global handler
}

在特定线程上运行工作

1 个答案: