summaryrefslogtreecommitdiff
path: root/ocaml-binary-annot/testsuite/tests/lib-threads/token2.ml
blob: 32f2c6ed37236a8571e16e7f7da9503360bf1588 (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
(* Performance test for I/O scheduling *)

let mut = Mutex.create()

let niter = ref 0

let token = ref 0

let process (n, ins, outs, nprocs) =
  let buf = String.create 1 in
  while true do
    Unix.read ins.(n) buf 0 1;
    (* Printf.printf "Thread %d got the token\n" n; *)
    if n = 0 then begin
      decr niter;
      if !niter <= 0 then exit 0
    end;
    let next = if n + 1 >= nprocs then 0 else n + 1 in
    (* Printf.printf "Thread %d sending token to thread %d\n" n next; *)
    Unix.write outs.(next) buf 0 1
  done

let main() =
  let nprocs = try int_of_string Sys.argv.(1) with _ -> 100 in
  let iter = try int_of_string Sys.argv.(2) with _ -> 1000 in
  let ins = Array.create nprocs Unix.stdin in
  let outs = Array.create nprocs Unix.stdout in
  for n = 0 to nprocs - 1 do
    let (i, o) = Unix.pipe() in ins.(n) <- i; outs.(n) <- o
  done;
  niter := iter;
  for i = 0 to nprocs - 1 do Thread.create process (i, ins, outs, nprocs) done;
  Unix.write outs.(0) "X" 0 1;
  Thread.delay 3600.

let _ = main()