我正在运行任务,调用从RabbitMQ读取的方法。当队列中没有任何内容时,该方法只会阻塞。所以任务有一个"运行"状态,但实际上并没有做任何事情。有没有办法优雅地结束这些任务?
访问队列的代码如下:
private void FindWork(CancellationToken ct)
{
if (ct.IsCancellationRequested)
return;
bool result = false;
bool process = false;
bool queueResult = false;
Work_Work work = null;
try
{
using (Queue workQueue = new Queue(_workQueue))
{
// Look for work on the work queue
workQueue.Open(Queue.Mode.Consume);
work = workQueue.ConsumeWithBlocking<Work_Work>();
// Do some work with the message ...
return;
任务创建如下:
private void Run()
{
while (!_stop)
{
// Remove and stopped tasks from the pool
List<int> removeThreads = new List<int>();
lock (_tasks)
{
foreach (KeyValuePair<int, Task> task in _tasks)
{
if (task.Value.Status != TaskStatus.Running)
{
task.Value.Wait();
removeThreads.Add(task.Value.Id);
}
}
foreach (int taskID in removeThreads)
_tasks.Remove(taskID);
}
CancellationToken ct = _cts.Token;
TaskFactory factory = new TaskFactory(ct, TaskCreationOptions.LongRunning, TaskContinuationOptions.LongRunning, null);
// Create new tasks if we have room in the pool
while (_tasks.Count < _runningMax)
{
Task task = factory.StartNew(() => FindWork(ct));
lock (_tasks)
_tasks.Add(task.Id, task);
}
// Take a rest so we don't run the CPU to death
Thread.Sleep(1000);
}
}
目前我已将我的任务创建代码更改为如下所示,以便我可以中止任务。我知道这不是一个好的解决方案,但我不知道还能做些什么。
while (_tasks.Count < _runningMax)
{
Task task = factory.StartNew(() =>
{
try
{
using (_cts.Token.Register(Thread.CurrentThread.Abort))
{
FindWork(ct);
}
}
catch (ThreadAbortException)
{
return;
}
}, _cts.Token);
_tasks.Add(task.Id, task);
}
答案 0 :(得分:1)
要完成这项工作,您需要更改ConsumeWithBlocking
以支持取消。我不熟悉RabbitMQ,但显然它支持cancellation on the consumer channel。
因此,不是从Thread.CurrentThread.Abort
回调中执行Token.Register
,而是通过正确的RabbitMQ API正确地取消操作。
在旁注中,您当前尝试中止的主题很可能不是被ConsumeWithBlocking
阻止的主题。
答案 1 :(得分:1)
您的方案中可以执行以下操作吗?
我没有产生多个线程并让它们在队列中等待,而是在一个无限的轮询循环中有一个线程,当一个新的工作片段进来时,它会产生一个新的线程。你可以添加一个信号量到限制您创建的线程数。检查下面的示例代码,我使用了BlockingCollection而不是RabbitMQ。
public class QueueManager
{
public BlockingCollection<Work> blockingCollection = new BlockingCollection<Work>();
private const int _maxRunningTasks = 3;
static SemaphoreSlim _sem = new SemaphoreSlim(_maxRunningTasks);
public void Queue()
{
blockingCollection.Add(new Work());
}
public void Consume()
{
while (true)
{
Work work = blockingCollection.Take();
_sem.Wait();
Task t = Task.Factory.StartNew(work.DoWork);
}
}
public class Work
{
public void DoWork()
{
Thread.Sleep(5000);
_sem.Release();
Console.WriteLine("Finished work");
}
}
}
和我的测试类
class Test
{
static void Main(string[] args)
{
Consumer c = new Consumer();
Task t = Task.Factory.StartNew(c.Consume);
c.Queue();
c.Queue();
c.Queue();
c.Queue();
c.Queue();
Thread.Sleep(1000);
Console.ReadLine();
}
}