如何用OCaml Unix模块等待多子进程？现有实现是否存在未定义行为？

一、当前实现的问题 #

1. 进程爆炸的致命逻辑错误 #

你的代码存在核心逻辑漏洞：子进程执行完f h后并未立即退出，而是会继续执行后续的loop t代码，导致子进程也会创建新的子进程，最终产生指数级数量的进程，完全偏离了「每个子进程处理一个任务」的设计初衷。

比如处理[1;2;3;4]时：

• 父进程创建子进程P1处理1，随后父进程继续创建P2处理2；
• 但P1执行完f 1后，会继续执行loop [2;3;4]，创建自己的子进程P1-2处理2，以此类推，最终进程数量会远超预期。

2. 关于Unix.wait的契约问题 #

你依赖「调用n次Unix.wait就能等待且仅等待n个子进程」的行为不属于未定义行为：

• Unix.wait的契约是：阻塞等待任意一个子进程终止，返回该子进程的PID和终止状态；
• 如果父进程确实有且仅有n个未被等待的子进程，调用n次wait确实能等待所有子进程完成。

但这个实现不够健壮：如果父进程在调用par_iter之前已经存在未被回收的子进程，wait可能会先等待那些无关的子进程，导致当前任务的子进程没有被正确等待。

二、正确的par_iter实现 #

我们需要修正子进程的退出逻辑，同时可以采用更健壮的方式等待所有子进程：

方案1：记录子进程PID并精准等待 #

open Unix

let par_iter (items: 'i list) ~(f: 'i -> unit): unit =
  let rec fork_and_collect_pids = function
    | [] -> []
    | h :: t ->
      match fork () with
      | 0 ->
          f h;
          exit 0  (* 子进程执行完任务立即退出，避免继续执行后续代码 *)
      | pid -> pid :: fork_and_collect_pids t
  in
  let child_pids = fork_and_collect_pids items in
  (* 父进程逐个等待每个子进程 *)
  List.iter child_pids ~f:(fun pid -> ignore @@ waitpid [] pid)

方案2：循环等待直到所有子进程被回收 #

如果不需要关注单个子进程的状态，也可以用循环wait的方式，直到没有子进程可等待：

open Unix

let par_iter (items: 'i list) ~(f: 'i -> unit): unit =
  let rec fork_children = function
    | [] -> ()
    | h :: t ->
      match fork () with
      | 0 ->
          f h;
          exit 0
      | _ -> fork_children t
  in
  fork_children items;
  (* 父进程循环wait，直到所有子进程都被回收 *)
  let rec wait_all () =
    try
      ignore @@ wait ();
      wait_all ()
    with
    | Unix_error (ECHILD, _, _) -> ()  (* 没有更多子进程时结束 *)
  in
  wait_all ()

实现说明 #

• 子进程在执行完任务f h后立即调用exit 0，彻底避免了子进程继续创建新进程的问题；
• 方案1通过记录每个子进程的PID，使用waitpid精准等待每个子进程，适合需要跟踪特定子进程状态的场景；
• 方案2通过循环wait直到触发ECHILD错误（表示没有未被回收的子进程），实现更简洁，适合不需要单独跟踪子进程的场景。

内容的提问来源于stack exchange，提问作者Max Heiber