CurrentThreadScheduler.cs
2.51 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
using System;
using System.Collections;
using System.Collections.Generic;
namespace DepthFirstScheduler
{
public static partial class Scheduler
{
private static IScheduler currentThread;
public static IScheduler CurrentThread
{
get { return currentThread ?? (currentThread = new CurrentThreadScheduler()); }
}
public class CurrentThreadScheduler : IScheduler
{
[ThreadStatic]
private static Queue<TaskChain> queue;
private static Queue<TaskChain> GetQueue()
{
return queue;
}
private static void SetQueue(Queue<TaskChain> newQueue)
{
queue = newQueue;
}
public void Enqueue(TaskChain item)
{
var q = GetQueue();
if (q == null)
{
q = new Queue<TaskChain>(5);
q.Enqueue(item);
SetQueue(q);
try
{
Trampoline.Run(q);
}
finally
{
SetQueue(null);
}
}
else
{
q.Enqueue(item);
}
}
#region IDisposable Support
private bool disposedValue = false;
protected virtual void Dispose(bool disposing)
{
if (!disposedValue)
{
if (disposing)
{
var queue = GetQueue();
if (queue != null) queue.Clear();
SetQueue(null);
}
disposedValue = true;
}
}
public void Dispose()
{
Dispose(true);
}
#endregion
}
static class Trampoline
{
public static void Run(Queue<TaskChain> queue)
{
while (queue.Count > 0)
{
var chain = queue.Dequeue();
while (true)
{
var status = chain.Next();
if (status != ExecutionStatus.Continue)
{
break;
}
}
}
}
}
}
}