Benchmarking module for LWT streams

bench.ml

module TestStream = Nstream

let n_stream = 1_000    (* # of items to pre-load stream.   *)
let n_threads = 1_000   (* # of threads reading the stream. *)

let s =
  let s, p = TestStream.create () in
  begin
    for ii = 1 to n_stream do
      p (Some ii)
    done;
    s
  end

(* The 'i' argument is the thread "ID". It's included in case we
   need to print thread info. *)

let rec thr i s = function
  | 0 -> Lwt.return_unit
  | n -> let%lwt _ = TestStream.next s in
         thr i s (n - 1)

let rec spawn = function
  | 1 -> [thr 1 s n_stream]
  | n -> (* This expression is split into two because we need to
            guarantee the stream is cloned before we recurse. *)
         let t = thr n (TestStream.clone s) n_stream in
         t :: spawn (n - 1)

let () =
  begin
    Lwt_main.run (Lwt.join (spawn n_threads));
    Gc.print_stat stdout
  end