0
我寫了我的自定義的TaskScheduler:爲什麼在主線程中運行任務?
public class LimitedConcurrencyLevelTaskScheduler : TaskScheduler
{
private BlockingCollection<Task> _tasks = new BlockingCollection<Task>();
private List<Thread> _threads;
private bool work = true;
public LimitedConcurrencyLevelTaskScheduler(int maxConcurrencyLevel)
{
_threads = new List<Thread>();
for (int i = 0; i < maxConcurrencyLevel; i++)
{
_threads.Add(new Thread(() =>
{
while (work)
{
TryExecuteTask(_tasks.Take());
}
}) { IsBackground = true, Name = "TaskShedulerThread#" + i });
}
foreach (var thread in _threads)
{
thread.Start();
}
}
protected override void QueueTask(Task task)
{
_tasks.Add(task);
}
protected override bool TryExecuteTaskInline(Task task, bool taskWasPreviouslyQueued)
{
return !_threads.Contains(Thread.CurrentThread) && TryExecuteTask(task);
}
public override int MaximumConcurrencyLevel { get { return 1; } }
protected override IEnumerable<Task> GetScheduledTasks()
{
return _tasks.ToArray();
}
public void Dispose()
{
if (_threads != null)
{
_tasks.CompleteAdding();
work = false;
_tasks.Dispose();
_tasks = null;
_threads = null;
}
}
}
並以這種方式使用它:
static void Main(string[] args)
{
var taskScheduller = new LimitedConcurrencyLevelTaskScheduler(1);
Thread.CurrentThread.Name = "MainThread";
var taskFactory = new TaskFactory(taskScheduller);
var tasks = new Task[100];
for (int i = 0; i < 100; i++)
{
tasks[i] = taskFactory.StartNew(() => Console.WriteLine(String.Format("Call in {0}", Thread.CurrentThread.Name)));
}
Task.WaitAll(tasks);
}
PROGRAMM的輸出:
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in MainThread
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in MainThread
Call in MainThread
Call in MainThread
Call in MainThread
Call in MainThread
Call in MainThread
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
Call in TaskShedulerThread#0
...
爲什麼我的任務在主線程拼命地跑?