ocaml/otherlibs/threads/Tests/token1.ml

37 lines
946 B
OCaml

(* 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.sleep()
let _ = main()