wenzelm@28165
|
1 |
(* Title: Pure/Concurrent/task_queue.ML
|
wenzelm@28165
|
2 |
Author: Makarius
|
wenzelm@28165
|
3 |
|
wenzelm@28165
|
4 |
Ordered queue of grouped tasks.
|
wenzelm@28165
|
5 |
*)
|
wenzelm@28165
|
6 |
|
wenzelm@28165
|
7 |
signature TASK_QUEUE =
|
wenzelm@28165
|
8 |
sig
|
wenzelm@29340
|
9 |
type task
|
wenzelm@29122
|
10 |
val new_task: int -> task
|
wenzelm@29122
|
11 |
val pri_of_task: task -> int
|
wenzelm@28196
|
12 |
val str_of_task: task -> string
|
wenzelm@29340
|
13 |
type group
|
wenzelm@32052
|
14 |
val group_id: group -> int
|
wenzelm@29340
|
15 |
val eq_group: group * group -> bool
|
wenzelm@32121
|
16 |
val new_group: group option -> group
|
wenzelm@32121
|
17 |
val group_status: group -> exn list
|
wenzelm@28179
|
18 |
val str_of_group: group -> string
|
wenzelm@28165
|
19 |
type queue
|
wenzelm@28165
|
20 |
val empty: queue
|
wenzelm@28204
|
21 |
val is_empty: queue -> bool
|
wenzelm@32052
|
22 |
val status: queue -> {ready: int, pending: int, running: int}
|
wenzelm@32218
|
23 |
val enqueue: group -> task list -> int -> (bool -> bool) -> queue -> (task * bool) * queue
|
wenzelm@29369
|
24 |
val extend: task -> (bool -> bool) -> queue -> queue option
|
wenzelm@29369
|
25 |
val dequeue: queue -> (task * group * (bool -> bool) list) option * queue
|
wenzelm@32071
|
26 |
val dequeue_towards: task list -> queue ->
|
wenzelm@32071
|
27 |
(((task * group * (bool -> bool) list) * task list) option * queue)
|
wenzelm@28190
|
28 |
val interrupt: queue -> task -> unit
|
wenzelm@28190
|
29 |
val interrupt_external: queue -> string -> unit
|
wenzelm@32121
|
30 |
val is_canceled: group -> bool
|
wenzelm@32119
|
31 |
val cancel_group: group -> exn -> unit
|
wenzelm@29340
|
32 |
val cancel: queue -> group -> bool
|
wenzelm@29340
|
33 |
val cancel_all: queue -> group list
|
wenzelm@32218
|
34 |
val finish: task -> queue -> bool * queue
|
wenzelm@28165
|
35 |
end;
|
wenzelm@28165
|
36 |
|
wenzelm@29340
|
37 |
structure Task_Queue:> TASK_QUEUE =
|
wenzelm@28165
|
38 |
struct
|
wenzelm@28165
|
39 |
|
wenzelm@29122
|
40 |
(* tasks *)
|
wenzelm@28165
|
41 |
|
wenzelm@29122
|
42 |
datatype task = Task of int * serial;
|
wenzelm@29122
|
43 |
fun new_task pri = Task (pri, serial ());
|
wenzelm@29003
|
44 |
|
wenzelm@29122
|
45 |
fun pri_of_task (Task (pri, _)) = pri;
|
wenzelm@29122
|
46 |
fun str_of_task (Task (_, i)) = string_of_int i;
|
wenzelm@28165
|
47 |
|
wenzelm@29122
|
48 |
fun task_ord (Task t1, Task t2) = prod_ord (rev_order o int_ord) int_ord (t1, t2);
|
wenzelm@31972
|
49 |
structure Task_Graph = Graph(type key = task val ord = task_ord);
|
wenzelm@29122
|
50 |
|
wenzelm@29122
|
51 |
|
wenzelm@32121
|
52 |
(* nested groups *)
|
wenzelm@29003
|
53 |
|
wenzelm@32121
|
54 |
datatype group = Group of
|
wenzelm@32121
|
55 |
{parent: group option,
|
wenzelm@32121
|
56 |
id: serial,
|
wenzelm@32121
|
57 |
status: exn list ref};
|
wenzelm@32052
|
58 |
|
wenzelm@32121
|
59 |
fun make_group (parent, id, status) = Group {parent = parent, id = id, status = status};
|
wenzelm@28551
|
60 |
|
wenzelm@32121
|
61 |
fun new_group parent = make_group (parent, serial (), ref []);
|
wenzelm@29122
|
62 |
|
wenzelm@32121
|
63 |
fun group_id (Group {id, ...}) = id;
|
wenzelm@32121
|
64 |
fun eq_group (group1, group2) = group_id group1 = group_id group2;
|
wenzelm@28165
|
65 |
|
wenzelm@32121
|
66 |
fun group_ancestry (Group {parent, id, ...}) =
|
wenzelm@32121
|
67 |
id :: (case parent of NONE => [] | SOME group => group_ancestry group);
|
wenzelm@32121
|
68 |
|
wenzelm@32121
|
69 |
|
wenzelm@32121
|
70 |
fun cancel_group (Group {status, ...}) exn = CRITICAL (fn () =>
|
wenzelm@32121
|
71 |
(case exn of
|
wenzelm@32121
|
72 |
Exn.Interrupt => if null (! status) then status := [exn] else ()
|
wenzelm@32121
|
73 |
| _ => change status (cons exn)));
|
wenzelm@32121
|
74 |
|
wenzelm@32121
|
75 |
fun group_status (Group {parent, status, ...}) = (*non-critical*)
|
wenzelm@32121
|
76 |
! status @ (case parent of NONE => [] | SOME group => group_status group);
|
wenzelm@32121
|
77 |
|
wenzelm@32121
|
78 |
fun is_canceled (Group {parent, status, ...}) = (*non-critical*)
|
wenzelm@32121
|
79 |
not (null (! status)) orelse (case parent of NONE => false | SOME group => is_canceled group);
|
wenzelm@32121
|
80 |
|
wenzelm@32121
|
81 |
fun str_of_group group =
|
wenzelm@32121
|
82 |
(is_canceled group ? enclose "(" ")") (string_of_int (group_id group));
|
wenzelm@28179
|
83 |
|
wenzelm@28165
|
84 |
|
wenzelm@28176
|
85 |
(* jobs *)
|
wenzelm@28165
|
86 |
|
wenzelm@28165
|
87 |
datatype job =
|
wenzelm@29369
|
88 |
Job of (bool -> bool) list |
|
wenzelm@28165
|
89 |
Running of Thread.thread;
|
wenzelm@28165
|
90 |
|
wenzelm@29122
|
91 |
type jobs = (group * job) Task_Graph.T;
|
wenzelm@28176
|
92 |
|
wenzelm@29122
|
93 |
fun get_group (jobs: jobs) task = #1 (Task_Graph.get_node jobs task);
|
wenzelm@29122
|
94 |
fun get_job (jobs: jobs) task = #2 (Task_Graph.get_node jobs task);
|
wenzelm@29369
|
95 |
fun set_job task job (jobs: jobs) = Task_Graph.map_node task (fn (group, _) => (group, job)) jobs;
|
wenzelm@28202
|
96 |
|
wenzelm@28176
|
97 |
|
wenzelm@28176
|
98 |
(* queue of grouped jobs *)
|
wenzelm@28176
|
99 |
|
wenzelm@31617
|
100 |
datatype result = Unknown | Result of task | No_Result;
|
wenzelm@31617
|
101 |
|
wenzelm@28165
|
102 |
datatype queue = Queue of
|
wenzelm@28184
|
103 |
{groups: task list Inttab.table, (*groups with presently active members*)
|
wenzelm@31617
|
104 |
jobs: jobs, (*job dependency graph*)
|
wenzelm@31617
|
105 |
cache: result}; (*last dequeue result*)
|
wenzelm@28165
|
106 |
|
wenzelm@31617
|
107 |
fun make_queue groups jobs cache = Queue {groups = groups, jobs = jobs, cache = cache};
|
wenzelm@28204
|
108 |
|
wenzelm@31617
|
109 |
val empty = make_queue Inttab.empty Task_Graph.empty No_Result;
|
wenzelm@29122
|
110 |
fun is_empty (Queue {jobs, ...}) = Task_Graph.is_empty jobs;
|
wenzelm@28165
|
111 |
|
wenzelm@28165
|
112 |
|
wenzelm@32121
|
113 |
(* queue status *)
|
wenzelm@32052
|
114 |
|
wenzelm@32052
|
115 |
fun status (Queue {jobs, ...}) =
|
wenzelm@32052
|
116 |
let
|
wenzelm@32052
|
117 |
val (x, y, z) =
|
wenzelm@32052
|
118 |
Task_Graph.fold (fn (task, ((_, job), (deps, _))) => fn (x, y, z) =>
|
wenzelm@32052
|
119 |
(case job of
|
wenzelm@32052
|
120 |
Job _ => if null deps then (x + 1, y, z) else (x, y + 1, z)
|
wenzelm@32052
|
121 |
| Running _ => (x, y, z + 1)))
|
wenzelm@32052
|
122 |
jobs (0, 0, 0);
|
wenzelm@32052
|
123 |
in {ready = x, pending = y, running = z} end;
|
wenzelm@32052
|
124 |
|
wenzelm@32052
|
125 |
|
wenzelm@32121
|
126 |
(* cancel -- peers and sub-groups *)
|
wenzelm@32121
|
127 |
|
wenzelm@32121
|
128 |
fun cancel (Queue {groups, jobs, ...}) group =
|
wenzelm@32121
|
129 |
let
|
wenzelm@32121
|
130 |
val _ = cancel_group group Exn.Interrupt;
|
wenzelm@32121
|
131 |
val tasks = Inttab.lookup_list groups (group_id group);
|
wenzelm@32121
|
132 |
val running =
|
wenzelm@32121
|
133 |
fold (get_job jobs #> (fn Running t => insert Thread.equal t | _ => I)) tasks [];
|
wenzelm@32121
|
134 |
val _ = List.app SimpleThread.interrupt running;
|
wenzelm@32121
|
135 |
in null running end;
|
wenzelm@32121
|
136 |
|
wenzelm@32121
|
137 |
fun cancel_all (Queue {jobs, ...}) =
|
wenzelm@32121
|
138 |
let
|
wenzelm@32121
|
139 |
fun cancel_job (group, job) (groups, running) =
|
wenzelm@32121
|
140 |
(cancel_group group Exn.Interrupt;
|
wenzelm@32121
|
141 |
(case job of Running t => (insert eq_group group groups, insert Thread.equal t running)
|
wenzelm@32121
|
142 |
| _ => (groups, running)));
|
wenzelm@32121
|
143 |
val (groups, running) = Task_Graph.fold (cancel_job o #1 o #2) jobs ([], []);
|
wenzelm@32121
|
144 |
val _ = List.app SimpleThread.interrupt running;
|
wenzelm@32121
|
145 |
in groups end;
|
wenzelm@32121
|
146 |
|
wenzelm@32121
|
147 |
|
wenzelm@28185
|
148 |
(* enqueue *)
|
wenzelm@28165
|
149 |
|
wenzelm@32190
|
150 |
fun add_job task dep (jobs: jobs) =
|
wenzelm@32190
|
151 |
Task_Graph.add_edge (dep, task) jobs handle Task_Graph.UNDEF _ => jobs;
|
wenzelm@32190
|
152 |
|
wenzelm@32190
|
153 |
fun get_deps (jobs: jobs) task =
|
wenzelm@32190
|
154 |
Task_Graph.imm_preds jobs task handle Task_Graph.UNDEF _ => [];
|
wenzelm@32190
|
155 |
|
wenzelm@32121
|
156 |
fun enqueue group deps pri job (Queue {groups, jobs, cache}) =
|
wenzelm@28165
|
157 |
let
|
wenzelm@29122
|
158 |
val task = new_task pri;
|
wenzelm@32121
|
159 |
val groups' = groups
|
wenzelm@32121
|
160 |
|> fold (fn gid => Inttab.cons_list (gid, task)) (group_ancestry group);
|
wenzelm@28185
|
161 |
val jobs' = jobs
|
wenzelm@32121
|
162 |
|> Task_Graph.new_node (task, (group, Job [job]))
|
wenzelm@32190
|
163 |
|> fold (add_job task) deps
|
wenzelm@32190
|
164 |
|> fold (fold (add_job task) o get_deps jobs) deps;
|
wenzelm@32218
|
165 |
val minimal = null (get_deps jobs' task);
|
wenzelm@31638
|
166 |
val cache' =
|
wenzelm@31638
|
167 |
(case cache of
|
wenzelm@31638
|
168 |
Result last =>
|
wenzelm@31638
|
169 |
if task_ord (last, task) = LESS
|
wenzelm@31638
|
170 |
then cache else Unknown
|
wenzelm@31638
|
171 |
| _ => Unknown);
|
wenzelm@32218
|
172 |
in ((task, minimal), make_queue groups' jobs' cache') end;
|
wenzelm@28165
|
173 |
|
wenzelm@31617
|
174 |
fun extend task job (Queue {groups, jobs, cache}) =
|
wenzelm@29369
|
175 |
(case try (get_job jobs) task of
|
wenzelm@31617
|
176 |
SOME (Job list) => SOME (make_queue groups (set_job task (Job (job :: list)) jobs) cache)
|
wenzelm@29369
|
177 |
| _ => NONE);
|
wenzelm@29369
|
178 |
|
wenzelm@28185
|
179 |
|
wenzelm@28185
|
180 |
(* dequeue *)
|
wenzelm@28185
|
181 |
|
wenzelm@31617
|
182 |
fun dequeue (queue as Queue {groups, jobs, cache}) =
|
wenzelm@29122
|
183 |
let
|
wenzelm@29369
|
184 |
fun ready (task, ((group, Job list), ([], _))) = SOME (task, group, rev list)
|
wenzelm@29122
|
185 |
| ready _ = NONE;
|
wenzelm@31617
|
186 |
fun deq boundary =
|
wenzelm@31617
|
187 |
(case Task_Graph.get_first boundary ready jobs of
|
wenzelm@31617
|
188 |
NONE => (NONE, make_queue groups jobs No_Result)
|
wenzelm@31617
|
189 |
| SOME (result as (task, _, _)) =>
|
wenzelm@31617
|
190 |
let
|
wenzelm@31617
|
191 |
val jobs' = set_job task (Running (Thread.self ())) jobs;
|
wenzelm@31617
|
192 |
val cache' = Result task;
|
wenzelm@31617
|
193 |
in (SOME result, make_queue groups jobs' cache') end);
|
wenzelm@29122
|
194 |
in
|
wenzelm@31617
|
195 |
(case cache of
|
wenzelm@31617
|
196 |
Unknown => deq NONE
|
wenzelm@31617
|
197 |
| Result last => deq (SOME last)
|
wenzelm@31617
|
198 |
| No_Result => (NONE, queue))
|
wenzelm@28384
|
199 |
end;
|
wenzelm@28202
|
200 |
|
wenzelm@28176
|
201 |
|
wenzelm@32071
|
202 |
(* dequeue_towards -- adhoc dependencies *)
|
wenzelm@32071
|
203 |
|
wenzelm@32071
|
204 |
fun dequeue_towards deps (queue as Queue {groups, jobs, ...}) =
|
wenzelm@32071
|
205 |
let
|
wenzelm@32071
|
206 |
fun ready task =
|
wenzelm@32071
|
207 |
(case Task_Graph.get_node jobs task of
|
wenzelm@32071
|
208 |
(group, Job list) =>
|
wenzelm@32121
|
209 |
if null (Task_Graph.imm_preds jobs task)
|
wenzelm@32121
|
210 |
then SOME (task, group, rev list)
|
wenzelm@32071
|
211 |
else NONE
|
wenzelm@32071
|
212 |
| _ => NONE);
|
wenzelm@32071
|
213 |
val tasks = filter (can (Task_Graph.get_node jobs)) deps;
|
wenzelm@32192
|
214 |
fun result (res as (task, _, _)) =
|
wenzelm@32192
|
215 |
let
|
wenzelm@32192
|
216 |
val jobs' = set_job task (Running (Thread.self ())) jobs;
|
wenzelm@32192
|
217 |
val cache' = Unknown;
|
wenzelm@32192
|
218 |
in (SOME (res, tasks), make_queue groups jobs' cache') end;
|
wenzelm@32071
|
219 |
in
|
wenzelm@32113
|
220 |
(case get_first ready tasks of
|
wenzelm@32192
|
221 |
SOME res => result res
|
wenzelm@32192
|
222 |
| NONE =>
|
wenzelm@32192
|
223 |
(case get_first (get_first ready o Task_Graph.imm_preds jobs) tasks of
|
wenzelm@32192
|
224 |
SOME res => result res
|
wenzelm@32192
|
225 |
| NONE => (NONE, queue)))
|
wenzelm@32071
|
226 |
end;
|
wenzelm@32071
|
227 |
|
wenzelm@32071
|
228 |
|
wenzelm@28190
|
229 |
(* sporadic interrupts *)
|
wenzelm@28190
|
230 |
|
wenzelm@28190
|
231 |
fun interrupt (Queue {jobs, ...}) task =
|
wenzelm@32121
|
232 |
(case try (get_job jobs) task of
|
wenzelm@32121
|
233 |
SOME (Running thread) => SimpleThread.interrupt thread
|
wenzelm@32121
|
234 |
| _ => ());
|
wenzelm@28190
|
235 |
|
wenzelm@29122
|
236 |
fun interrupt_external (queue as Queue {jobs, ...}) str =
|
wenzelm@29122
|
237 |
(case Int.fromString str of
|
wenzelm@29122
|
238 |
SOME i =>
|
wenzelm@31617
|
239 |
(case Task_Graph.get_first NONE
|
wenzelm@29122
|
240 |
(fn (task as Task (_, j), _) => if i = j then SOME task else NONE) jobs
|
wenzelm@29122
|
241 |
of SOME task => interrupt queue task | NONE => ())
|
wenzelm@29122
|
242 |
| NONE => ());
|
wenzelm@28190
|
243 |
|
wenzelm@28190
|
244 |
|
wenzelm@29340
|
245 |
(* termination *)
|
wenzelm@28176
|
246 |
|
wenzelm@31638
|
247 |
fun finish task (Queue {groups, jobs, cache}) =
|
wenzelm@28176
|
248 |
let
|
wenzelm@32121
|
249 |
val group = get_group jobs task;
|
wenzelm@32121
|
250 |
val groups' = groups
|
wenzelm@32121
|
251 |
|> fold (fn gid => Inttab.remove_list (op =) (gid, task)) (group_ancestry group);
|
wenzelm@29122
|
252 |
val jobs' = Task_Graph.del_node task jobs;
|
wenzelm@32218
|
253 |
val maximal = null (Task_Graph.imm_succs jobs task);
|
wenzelm@32218
|
254 |
val cache' = if maximal then cache else Unknown;
|
wenzelm@32218
|
255 |
in (maximal, make_queue groups' jobs' cache') end;
|
wenzelm@28165
|
256 |
|
wenzelm@28165
|
257 |
end;
|