OCaml中编写可中断读取线程的正确方式？Sys.set_signal子线程不中断问题

这个问题正好点出了OCaml线程与信号处理的核心限制：OCaml的用户态线程（Thread模块）不会接收信号，信号只会被主线程捕获。这就是为什么你的多线程版本没法用Ctrl-C中断——子线程里的input_line会一直阻塞，根本收不到SIGINT信号。

为什么单线程能正常工作？ #

在单线程场景下，当你按下Ctrl-C，操作系统会把SIGINT信号递送到主线程。此时OCaml会中断正在执行的阻塞系统调用（比如input_line依赖的read），触发你设置的信号处理函数，抛出SigInt异常，从而中断程序。连续按两次Ctrl-C的作用是确保即使第一次异常处理还在执行，第二次信号能直接终止程序。

多线程场景的解决方案 #

要让子线程的读取操作可中断，我们需要换一种思路：让主线程处理信号，通过共享状态通知子线程退出，同时子线程不能用完全阻塞的input_line，而是用带超时的IO操作定期检查退出标志。

下面是一个完整的实现示例：

exception SigInt

(* 共享的退出标志，用互斥锁保证线程安全 *)
let should_exit = ref false
let exit_mutex = Mutex.create ()

let set_exit () =
  Mutex.lock exit_mutex;
  should_exit := true;
  Mutex.unlock exit_mutex

let check_exit () =
  Mutex.lock exit_mutex;
  let res = !should_exit in
  Mutex.unlock exit_mutex;
  res

(* 可中断的读取函数：用Unix.select实现带超时的等待，定期检查退出标志 *)
let rec interruptible_read chan =
  if check_exit () then raise SigInt;
  (* 等待0.1秒，期间如果有输入就处理，否则继续检查退出标志 *)
  let read_fds, _, _ = Unix.select [Unix.stdin] [] [] 0.1 in
  if read_fds <> [] then
    input_line chan
  else
    interruptible_read chan

let _ =
  (* 主线程设置SIGINT处理：设置退出标志并抛出异常 *)
  Sys.set_signal Sys.sigint (Sys.Signal_handle (fun _ ->
    set_exit ();
    raise SigInt
  ));

  (* 创建子线程执行读取逻辑 *)
  let read_thread = Thread.create (fun () ->
    try
      let line = interruptible_read stdin in
      output_string stdout line;
      flush stdout
    with SigInt ->
      print_endline "THREAD_SIGINT"
  ) () in

  (* 主线程等待子线程结束，同时处理自身的SigInt异常 *)
  try
    Thread.join read_thread
  with SigInt ->
    print_endline "MAIN_SIGINT"

关键细节说明 #

1. 共享状态的线程安全：用Mutex保护should_exit变量，避免主线程和子线程同时读写导致的竞态条件。
2. 带超时的IO操作：用Unix.select替代直接调用input_line，这样子线程不会一直阻塞，每隔0.1秒就会检查一次退出标志。
3. 信号只在主线程处理：OCaml的信号处理是全局的，且只有主线程能接收信号，所以所有信号逻辑都放在主线程中。

关于OCaml 5+的Domain（原生线程） #

如果你使用OCaml 5及以上版本的Domain模块（真正的系统级线程），信号可以递送到任意Domain，但默认还是由主线程处理。你可以通过Sys.set_signal结合Domain的信号掩码来调整，但这种方式复杂度更高，兼容性也不如上面的方案，除非你需要真正的并行计算，否则还是推荐用用户态线程+共享标志的方案。

内容的提问来源于stack exchange，提问作者Clément