blob: b2e24117674a1aba332faa0b9f9989d42c8321d9 (
plain)
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
|
(* TEST
{
bytecode;
}{
native;
}
*)
open Effect
open Effect.Deep
let num_domains = 2
type _ Effect.t += Fork : (unit -> unit) -> unit Effect.t
let fork f = perform (Fork f)
let with_mutex m f =
Mutex.lock m;
Fun.protect ~finally:(fun () -> Mutex.unlock m) f
let rec work n () = if n = 0 then () else fork (work (n - 1))
let run =
let run_q : (unit, unit) continuation Queue.t = Queue.create () in
let run_m = Mutex.create () in
let enqueue k = with_mutex run_m (fun () -> Queue.push k run_q) in
let dequeue () =
with_mutex run_m (fun () -> Queue.take_opt run_q)
|> Option.iter (fun k -> continue k ())
in
let rec spawn f =
(* Effect handler => instantiates fiber *)
match_with f ()
{
retc = (fun () -> dequeue ());
exnc =
(fun e ->
print_string (Printexc.to_string e);
dequeue ());
effc =
(fun (type a) (e : a Effect.t) ->
match e with
| Fork f ->
Some
(fun (k : (a, unit) continuation) ->
enqueue k;
spawn f)
| _ -> None);
}
in
let domains =
Array.init num_domains (fun _ ->
Domain.spawn (fun () -> spawn (work 100000)))
in
Array.iter Domain.join domains;
print_endline "OK"
|
