如何在OCaml中创建大量线程?

时间:2015-12-16 13:30:50

标签: ocaml

我在Racket小组中发现topic有关channel创建的效果。

我想写一个OCaml的版本来测试。

let post (c,x) = Event.sync (Event.send c x);;

let accept c = Event.sync (Event.receive c);;

let get_chan c = let n = accept c in print_int n;print_newline ();;

let chan_trans (old_chan, new_chan) =
  let s = accept old_chan in
  post (new_chan,(s+1));;

let rec whisper count init_val =
  let rec aux n chan =
    if n >= count then chan
    else
      let new_chan = Event.new_channel ()
      in Thread.create chan_trans (chan, new_chan);
      aux (n+1) new_chan
  in let leftest_chan = Event.new_channel ()
  in let t0 = Thread.create post (leftest_chan, init_val)
  in let rightest_chan = aux 0 leftest_chan
  in get_chan rightest_chan;;

whisper 10000 1;;

问题是,当我测试whisper 1000 1时,它按预期生成1001。但是,当我尝试测试whisper 10000 1时,会出现错误

  

致命错误:异常Sys_error(" Thread.create:资源暂时不可用")

我用这个命令编译并运行

  

ocamlc -thread unix.cma threads.cma -o prog whisper.ml&& ./ prog -I   +线程

2 个答案:

答案 0 :(得分:9)

OCaml Thread模块使用真实系统(内核)线程。线程总数受内核限制:

 cat /proc/sys/kernel/threads-max
 251422

你当然可以增加这个,

 echo 100000 > /proc/sys/kernel/threads-max

但更好的方法是将线程视为资源并相应地进行管理。

let rec whisper count init_val =
  let rec aux n t chan =
    if n >= count then chan
    else
      let new_chan = Event.new_channel () in
      let t' = Thread.create chan_trans (chan, new_chan) in
      Thread.join t;
      aux (n+1) t' new_chan in
  let leftest_chan = Event.new_channel () in
  let t = Thread.create post (leftest_chan, init_val) in
  let rightest_chan = aux 0 t leftest_chan in
  get_chan rightest_chan

在这种情况下,它将以任何大小的管道运行。例如:

$ ocamlbuild -use-ocamlfind -tag thread -pkg threads ev.native
$ time ./ev.native 
100001

real    0m1.581s

但是这种中国低语的实施非常粗糙和低效。你不应该为此使用重量级本机线程(并且都不会使用它们)。相反,您应该使用LwtAsync库中的协作轻量级线程。这将是非常有效和好的。

使用Lwt

实施

此实现紧跟着blog post的Go实现,但我认为我们可以在不使用邮箱的情况下在OCaml中更有效率和更准确地实现(但我不确定它是否符合规则)基准)。

open Lwt.Infix

let whispers n =
  let rec whisper i p =
    if i < n then
      Lwt_mvar.take p >>= fun x ->
      whisper (i+1) (Lwt_mvar.create (x+1))
    else Lwt_mvar.take p in
  whisper 0 (Lwt_mvar.create 1)

let () = print_int @@ Lwt_main.run (whispers 100000)

结果是:

$ ocamlbuild -use-ocamlfind -tag thread -pkg lwt.unix lev.native --
$ time ./lev.native 
100001
real    0m0.007s

与矿山机器上的Go实施进行比较:

$ go build whispers.go 
$ time ./whispers 
100001

real    0m0.952s

“慢”实施

上面的代码是原始Go版本的完全诚实的重新实现。但是它之所以如此之快的原因之一是OCaml和Lwt非常聪明,虽然它创建了100_000个线程和100_001个通道,但每次都没有线程产生任何线程。 whisper被称为通道已包含数据,因此线程处于就绪状态。因此,这只是一个有效的循环,可以创建线程和通道。它可以在50毫秒内创建数百万个线程。

所以这是一种惯用且正确的做事方式。但为了真实比较mimick Go行为。以下实现将 首先急切地在堆中创建100_001个通道和100_000个线程,等待从左到右的通道传输数据。然后它才会在最左边的通道中放置一个值来激发一连串的反应。这基本上可以模仿引擎盖下发生的事情。

let whispers n =
  let rec loop i p =
    if i < n then
      let p' = Lwt_mvar.create_empty () in
      let _t =
        Lwt_mvar.take p >>= fun x ->
        Lwt_mvar.put p' (x+1) in
      loop (i+1) p'
    else Lwt_mvar.take p in
  let p0 = Lwt_mvar.create_empty () in
  let t = loop 1 p0 in
  Lwt_mvar.put p0 1 >>= fun () -> t

$ time ./lev.native
100001
real    0m0.111s

所以它稍慢,实际上它比之前的实现慢了20倍(我使用了100万个线程来比较它们),但它仍然比Go快10倍。

答案 1 :(得分:3)

阅读链接帖子似乎你可能想使用lwt这是一个“OCaml的协作线程库”。结果看起来像这样:

let whisper left right =
  let%lwt n = Lwt_mvar.take right in
  Lwt_mvar.put left (n+1)

let main () =
  let n = 100_000 in
  let%lwt () = Lwt_io.printf "With %d mvars!\n" n in
  let leftmost = Lwt_mvar.create_empty () in
  let rec setup_whispers left i =
    if  i >= n
    then left
    else let right = Lwt_mvar.create_empty () in
      let () = Lwt.async (fun () -> whisper left right) in
      setup_whispers right (i+1) in
  let rightmost = setup_whispers leftmost 0 in
  let%lwt () = Lwt_mvar.put rightmost 1 in
  let%lwt res = Lwt_mvar.take leftmost in
  Lwt_io.printf "%d\n" res

let () = Lwt_main.run (main ())

然后编译并运行它

$ ocamlbuild -use-ocamlfind -pkg lwt,lwt.ppx,lwt.unix whisper.native
$ time ./whisper.native
With 100000 mvars!
100001

real    0m0.169s
user    0m0.156s
sys 0m0.008s