ocaml/otherlibs/threads/Tests/token1.ml

(* Performance test for mutexes and conditions *)

let mut = Mutex.create()

let niter = ref 0

let token = ref 0

let process (n, conds, nprocs) =
  while true do
    Mutex.lock mut;
    while !token <> n do 
      (* Printf.printf "Thread %d waiting (token = %d)\n" n !token; *)
      Condition.wait conds.(n) mut
    done;
    (* Printf.printf "Thread %d got token %d\n" n !token; *)
    incr token;
    if !token >= nprocs then token := 0;
    if n = 0 then begin
      decr niter;
      if !niter <= 0 then exit 0
    end;
    Condition.signal conds.(!token);
    Mutex.unlock mut
  done

let main() =
  let nprocs = int_of_string Sys.argv.(1) in
  let iter = int_of_string Sys.argv.(2) in
  let conds = Array.create nprocs (Condition.create()) in
  for i = 1 to nprocs - 1 do conds.(i) <- Condition.create() done;
  niter := iter;
  for i = 0 to nprocs - 1 do Thread.create process (i, conds, nprocs) done;
  Thread.delay 3600.

let _ = main()